[dialogflow] fix: marking a test with flaky (#3671)

fixes #3490

Author: Takashi Matsuo

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,

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))

dialogflow/cloud-client/requirements-test.txt

@@ -1 +1,2 @@
 pytest==5.3.2
+flaky==3.6.1