chore: added extra flow for traingflow test (#209)

munkhuushmgl · web-flow · commit ae769cb33cef · 2021-05-07T20:02:04.000Z
Fixes #203 ☕️ Problem is IT has only one flow that is being used in numerous tests. Because of that trainFlowTest is flaky, so it might be good idea to have seperate flow only for trainFlowTest
diff --git a/google-cloud-dialogflow-cx/src/test/java/com/google/cloud/dialogflow/cx/v3beta1/it/ITSystemTest.java b/google-cloud-dialogflow-cx/src/test/java/com/google/cloud/dialogflow/cx/v3beta1/it/ITSystemTest.java
@@ -103,6 +103,7 @@ public class ITSystemTest {
   private static String agentName;
   private static String entityTypesName;
   private static String flowName;
+  private static String trainFlowName;
   private static String intentsName;
   private static String pageName;
   private static String transitionRouteGroupName;
@@ -173,6 +174,17 @@ public static void setUp() throws IOException, ExecutionException, InterruptedEx
     Flow flowResponse = flowsClient.createFlow(createFlowRequest);
     flowName = flowResponse.getName();
 
+    Flow trainFlow =
+        Flow.newBuilder()
+            .setNluSettings(NLUSETTINGS)
+            .setDisplayName(DISPLAY_NAME)
+            .setDescription(DESCRIPTION)
+            .build();
+    CreateFlowRequest createTrainFlowRequest =
+        CreateFlowRequest.newBuilder().setParent(agentName).setFlow(trainFlow).build();
+    Flow trainFlowResponse = flowsClient.createFlow(createTrainFlowRequest);
+    trainFlowName = trainFlowResponse.getName();
+
     /* create intents */
     intentsClient = IntentsClient.create();
     Intent intent = Intent.newBuilder().setDisplayName(DISPLAY_NAME).setPriority(1).build();
@@ -245,6 +257,10 @@ public static void tearDown() {
     /* delete flows */
     DeleteFlowRequest deleteFlowRequest = DeleteFlowRequest.newBuilder().setName(flowName).build();
     flowsClient.deleteFlow(deleteFlowRequest);
+
+    DeleteFlowRequest deleteTrainFlowRequest =
+        DeleteFlowRequest.newBuilder().setName(trainFlowName).build();
+    flowsClient.deleteFlow(deleteTrainFlowRequest);
     flowsClient.close();
 
     /* delete entity types */
@@ -381,7 +397,7 @@ public void listFlowsTest() {
   @Test
   public void trainFlowTest() throws ExecutionException, InterruptedException {
     Empty expectedResponse = Empty.newBuilder().build();
-    Empty response = flowsClient.trainFlowAsync(flowName).get();
+    Empty response = flowsClient.trainFlowAsync(trainFlowName).get();
     assertEquals(expectedResponse, response);
   }
 
