[dialogflow] fix: marking a test with flaky

fixes #3490
GoogleCloudPlatform · May 2, 2020 · 10cd0a1 · 10cd0a1
1 parent af91571
commit 10cd0a1
Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/dialogflow/cloud-client/create_document_test.py b/dialogflow/cloud-client/create_document_test.py
@@ -47,6 +47,7 @@ def setup_teardown():
     client.delete_knowledge_base(knowledge_base_path, force=True)
 
 
+@pytest.mark.flaky(max_runs=3, min_passes=1)
 def test_create_document(capsys):
     document_management.create_document(
         PROJECT_ID, pytest.KNOWLEDGE_BASE_ID, DOCUMENT_DISPLAY_NAME,

diff --git a/dialogflow/cloud-client/document_management.py b/dialogflow/cloud-client/document_management.py
@@ -60,7 +60,7 @@ def create_document(project_id, knowledge_base_id, display_name, mime_type,
 
     response = client.create_document(knowledge_base_path, document)
     print('Waiting for results...')
-    document = response.result(timeout=90)
+    document = response.result(timeout=120)
     print('Created Document:')
     print(' - Display Name: {}'.format(document.display_name))
     print(' - Knowledge ID: {}'.format(document.name))

diff --git a/dialogflow/cloud-client/requirements-test.txt b/dialogflow/cloud-client/requirements-test.txt
@@ -1 +1,2 @@
 pytest==5.3.2
+flaky==3.6.1