From 2f163262a5e70958eb9152dc0e52c316c524f48d Mon Sep 17 00:00:00 2001
From: Noah Negrey <nnegrey@users.noreply.github.com>
Date: Fri, 15 Nov 2019 15:15:24 -0700
Subject: [PATCH] Add Set Endpoint Samples
 [(#2497)](https://github.com/GoogleCloudPlatform/python-docs-samples/issues/2497)

* Add Set Endpoint Samples

* Add additional test result option

* Sample Request update

* Add filter_
---
 automl/beta/requirements.txt     |  1 +
 automl/beta/set_endpoint.py      | 39 ++++++++++++++++++++++++++++++++
 automl/beta/set_endpoint_test.py | 26 +++++++++++++++++++++
 3 files changed, 66 insertions(+)
 create mode 100644 automl/beta/requirements.txt
 create mode 100644 automl/beta/set_endpoint.py
 create mode 100644 automl/beta/set_endpoint_test.py

diff --git a/automl/beta/requirements.txt b/automl/beta/requirements.txt
new file mode 100644
index 000000000000..7d202213a88a
--- /dev/null
+++ b/automl/beta/requirements.txt
@@ -0,0 +1 @@
+google-cloud-automl==0.7.0
diff --git a/automl/beta/set_endpoint.py b/automl/beta/set_endpoint.py
new file mode 100644
index 000000000000..436e427ec827
--- /dev/null
+++ b/automl/beta/set_endpoint.py
@@ -0,0 +1,39 @@
+# Copyright 2019 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+def set_endpoint(project_id):
+    """Change your endpoint"""
+    # [START automl_set_endpoint]
+    from google.cloud import automl_v1beta1 as automl
+
+    # You must first create a dataset, using the `eu` endpoint, before you can
+    # call other operations such as: list, get, import, delete, etc.
+    client_options = {'api_endpoint': 'eu-automl.googleapis.com:443'}
+
+    # Instantiates a client
+    client = automl.AutoMlClient(client_options=client_options)
+
+    # A resource that represents Google Cloud Platform location.
+    # project_id = 'YOUR_PROJECT_ID'
+    project_location = client.location_path(project_id, 'eu')
+    # [END automl_set_endpoint]
+
+    # List all the datasets available
+    # Note: Create a dataset in `eu`, before calling `list_datasets`.
+    response = client.list_datasets(
+        project_location, filter_='')
+
+    for dataset in response:
+        print(dataset)
diff --git a/automl/beta/set_endpoint_test.py b/automl/beta/set_endpoint_test.py
new file mode 100644
index 000000000000..88a0164c66a2
--- /dev/null
+++ b/automl/beta/set_endpoint_test.py
@@ -0,0 +1,26 @@
+# Copyright 2019 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+import set_endpoint
+
+PROJECT_ID = os.environ['GCLOUD_PROJECT']
+
+
+def test_set_endpoint(capsys):
+    set_endpoint.set_endpoint(PROJECT_ID)
+
+    out, _ = capsys.readouterr()
+    # Look for the display name
+    assert 'do_not_delete_me' in out