|
22 | 22 |
|
23 | 23 | from update_intent import update_intent |
24 | 24 |
|
25 | | -PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT") |
| 25 | +PROJECT_ID: str = os.getenv("GOOGLE_CLOUD_PROJECT") |
26 | 26 | pytest.INTENT_ID = None |
27 | 27 |
|
28 | 28 |
|
29 | | -def create_intent(project_id): |
30 | | - intents_client = IntentsClient() |
| 29 | +def create_intent(project_id: str) -> str: |
| 30 | + intents_client: IntentsClient = IntentsClient() |
31 | 31 |
|
32 | | - parent = AgentsClient.agent_path(project_id) |
| 32 | + parent: str = AgentsClient.agent_path(project_id) |
33 | 33 |
|
34 | | - intent = Intent() |
| 34 | + intent: Intent = Intent() |
35 | 35 |
|
36 | | - intent.display_name = "fake_intent" |
| 36 | + intent.display_name = f"fake_intent_{uuid.uuid4()}" |
37 | 37 |
|
38 | | - intents = intents_client.create_intent(request={"parent": parent, "intent": intent}) |
| 38 | + intents: Intent = intents_client.create_intent( |
| 39 | + request={"parent": parent, "intent": intent} |
| 40 | + ) |
39 | 41 |
|
40 | 42 | return intents.name.split("/")[4] |
41 | 43 |
|
42 | 44 |
|
43 | 45 | @pytest.fixture(scope="function", autouse=True) |
44 | | -def setup_teardown(): |
| 46 | +def setup_teardown() -> None: |
45 | 47 | pytest.INTENT_ID = create_intent(project_id=PROJECT_ID) |
46 | 48 | print("Created Intent in setUp") |
47 | 49 |
|
48 | 50 |
|
49 | | -def test_update_intent(): |
| 51 | +def test_update_intent() -> None: |
50 | 52 |
|
51 | | - fake_intent = "fake_intent_{}".format(uuid.uuid4()) |
| 53 | + # A new display name with an updated suffix |
| 54 | + new_display_name: str = f"fake_intent_{uuid.uuid4()}" |
52 | 55 |
|
53 | | - actualResponse = update_intent(PROJECT_ID, pytest.INTENT_ID, fake_intent) |
54 | | - expectedResponse = fake_intent |
| 56 | + actual_response: Intent = update_intent( |
| 57 | + PROJECT_ID, pytest.INTENT_ID, new_display_name |
| 58 | + ) |
| 59 | + expected_response: str = new_display_name |
55 | 60 |
|
56 | | - intents_client = IntentsClient() |
| 61 | + intents_client: IntentsClient = IntentsClient() |
57 | 62 |
|
58 | | - intents_client.delete_intent(name=actualResponse.name) |
| 63 | + intents_client.delete_intent(name=actual_response.name) |
59 | 64 |
|
60 | | - assert actualResponse.display_name == expectedResponse |
| 65 | + assert actual_response.display_name == expected_response |
0 commit comments