From 30935ee3e1f54c95f72543879319b27a9489f403 Mon Sep 17 00:00:00 2001
From: nnegrey <nnegrey@google.com>
Date: Thu, 16 Jan 2020 13:50:21 -0700
Subject: [PATCH] dialogflow: add and increase timeout to document creation for
 tests

---
 .../main/java/com/example/dialogflow/DocumentManagement.java   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/dialogflow/cloud-client/src/main/java/com/example/dialogflow/DocumentManagement.java b/dialogflow/cloud-client/src/main/java/com/example/dialogflow/DocumentManagement.java
index e0992d08ee9..00782964f27 100644
--- a/dialogflow/cloud-client/src/main/java/com/example/dialogflow/DocumentManagement.java
+++ b/dialogflow/cloud-client/src/main/java/com/example/dialogflow/DocumentManagement.java
@@ -25,6 +25,7 @@
 import com.google.common.collect.Lists;
 
 import java.util.List;
+import java.util.concurrent.TimeUnit;
 
 public class DocumentManagement {
   //  [START dialogflow_list_document]
@@ -86,7 +87,7 @@ public static Document createDocument(
               .build();
       OperationFuture<Document, KnowledgeOperationMetadata> response =
           documentsClient.createDocumentAsync(createDocumentRequest);
-      Document createdDocument = response.get();
+      Document createdDocument = response.get(120, TimeUnit.SECONDS);
       System.out.format("Created Document:\n");
       System.out.format(" - Display Name: %s\n", createdDocument.getDisplayName());
       System.out.format(" - Knowledge ID: %s\n", createdDocument.getName());