From 3e80235277a0df12644d743f6853ba45263a1239 Mon Sep 17 00:00:00 2001
From: Gal Zahavi <38544478+galz10@users.noreply.github.com>
Date: Fri, 13 Aug 2021 11:44:22 -0700
Subject: [PATCH] docs(samples): add update intent sample (#142)

* add update intent sample

* fixed lint

* added agent creation

* fixed lint

* added randomized agent name

* Failing test fix

* Fixed failing test

* Change intent request

* removed self from test

* lint fix

* Update samples/snippets/update_intent_test.py

Co-authored-by: Leah E. Cole <6719667+leahecole@users.noreply.github.com>

Co-authored-by: Leah E. Cole <6719667+leahecole@users.noreply.github.com>
---
 samples/snippets/update_intent.py      | 29 +++++++++
 samples/snippets/update_intent_test.py | 82 ++++++++++++++++++++++++++
 2 files changed, 111 insertions(+)
 create mode 100644 samples/snippets/update_intent.py
 create mode 100644 samples/snippets/update_intent_test.py

diff --git a/samples/snippets/update_intent.py b/samples/snippets/update_intent.py
new file mode 100644
index 00000000..87e89d85
--- /dev/null
+++ b/samples/snippets/update_intent.py
@@ -0,0 +1,29 @@
+# Copyright 2021, 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.
+
+from google.cloud.dialogflowcx_v3.services.intents import IntentsClient
+from google.protobuf import field_mask_pb2
+
+
+def update_intent(project_id, agent_id, intent_id, location, displayName):
+
+    intents_client = IntentsClient()
+
+    intent_name = intents_client.intent_path(project_id, location, agent_id, intent_id)
+
+    intent = intents_client.get_intent(request={"name": intent_name})
+
+    intent.display_name = displayName
+    update_mask = field_mask_pb2.FieldMask(paths=["display_name"])
+    response = intents_client.update_intent(intent=intent, update_mask=update_mask)
+    return response
diff --git a/samples/snippets/update_intent_test.py b/samples/snippets/update_intent_test.py
new file mode 100644
index 00000000..58fafa60
--- /dev/null
+++ b/samples/snippets/update_intent_test.py
@@ -0,0 +1,82 @@
+# Copyright 2021, 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 uuid
+
+from google.cloud.dialogflowcx_v3.services.agents.client import AgentsClient
+from google.cloud.dialogflowcx_v3.services.intents.client import IntentsClient
+from google.cloud.dialogflowcx_v3.types.agent import Agent, DeleteAgentRequest
+from google.cloud.dialogflowcx_v3.types.intent import CreateIntentRequest, Intent
+
+import pytest
+
+from update_intent import update_intent
+
+PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
+pytest.INTENT_ID = None
+pytest.AGENT_ID = None
+pytest.PARENT = None
+
+
+def create_agent(project_id, display_name):
+    parent = "projects/" + project_id + "/locations/global"
+
+    agents_client = AgentsClient()
+
+    agent = Agent(
+        display_name=display_name,
+        default_language_code="en",
+        time_zone="America/Los_Angeles",
+    )
+
+    response = agents_client.create_agent(request={"agent": agent, "parent": parent})
+
+    return response
+
+
+def delete_agent(name):
+    agents_client = AgentsClient()
+    agent = DeleteAgentRequest(name=name)
+    agents_client.delete_agent(request=agent)
+
+
+@pytest.fixture(scope="function", autouse=True)
+def setup_teardown():
+    agentName = "temp_agent_" + str(uuid.uuid4())
+    pytest.PARENT = create_agent(PROJECT_ID, agentName).name
+    pytest.AGENT_ID = pytest.PARENT.split("/")[5]
+    print("Created Agent in setUp")
+    intentClient = IntentsClient()
+    intent = Intent()
+
+    intent.display_name = "fake_intent"
+
+    req = CreateIntentRequest()
+    req.parent = pytest.PARENT
+    req.intent = intent
+
+    pytest.INTENT_ID = intentClient.create_intent(request=req).name.split("/")[7]
+
+    yield
+
+    delete_agent(pytest.PARENT)
+
+
+def test_fieldmaskTest():
+    fake_intent = f"fake_intent_{uuid.uuid4()}"
+    actualResponse = update_intent(
+        PROJECT_ID, pytest.AGENT_ID, pytest.INTENT_ID, "global", fake_intent
+    )
+
+    assert actualResponse.display_name == fake_intent