8
8
import org .junit .Test ;
9
9
10
10
import com .google .cloud .workflows .v1 .WorkflowName ;
11
+
12
+ import java .io .IOException ;
13
+
11
14
import static com .example .workflows .WorkflowsQuickstart .workflowsQuickstart ;
12
15
import static org .junit .Assert .*;
13
16
@@ -31,55 +34,49 @@ public static void beforeClass() {
31
34
}
32
35
33
36
@ Test
34
- public void testQuickstart () {
35
- System .out .println ("testing!!" );
36
- System .out .println ("testing!!" );
37
- System .out .println ("testing!!" );
38
- System .out .println ("testing!!" );
37
+ public void testQuickstart () throws IOException , InterruptedException {
39
38
System .out .println ("testing!!" );
39
+
40
40
// Deploy the workflow
41
41
deployWorkflow (projectId , LOCATION_ID , WORKFLOW_ID );
42
42
43
43
// Run the workflow we deployed
44
- // String res = workflowsQuickstart(projectId, LOCATION_ID, WORKFLOW_ID);
45
- //
46
- // // A very basic assertion that we have some result.
47
- // assertNotNull("Result should not be null", res);
48
- // assertNotEquals("Result should not be empty", res, "");
44
+ String res = workflowsQuickstart (projectId , LOCATION_ID , WORKFLOW_ID );
45
+
46
+ // A very basic assertion that we have some result.
47
+ assertNotNull ("Result should not be null" , res );
48
+ assertNotEquals ("Result should not be empty" , res , "" );
49
49
}
50
50
51
- private boolean deployWorkflow (String projectId , String location , String workflowId ) {
51
+ private boolean deployWorkflow (String projectId , String location , String workflowId ) throws IOException , InterruptedException {
52
52
// Create a new workflow if it doesn't exist
53
53
if (!workflowExists (projectId , location , workflowId )) {
54
54
System .out .println ("START DEPLOY" );
55
- try (WorkflowsClient workflowsClient = WorkflowsClient .create ()) {
56
- // Deploy workflow
57
- Workflow workflow = Workflow .newBuilder ()
58
- .setName (workflowId )
59
- .setSourceContents (WORKFLOW_SOURCE )
60
- .build ();
61
- workflowsClient .createWorkflowAsync (location , workflow , workflowId );
62
-
63
- // Wait until workflow is active
64
- Workflow deployedWorkflow = null ;
65
-
66
- System .out .println ("DEPLOY START" );
67
- // Wait for the deployment to finish
68
- do {
69
- System .out .println ("SLEEP" );
70
- deployedWorkflow = workflowsClient .getWorkflow (WorkflowName .newBuilder ()
71
- .setProject (projectId )
72
- .setLocation (location )
73
- .setWorkflow (workflowId )
74
- .build ());
75
- Thread .sleep (2_000 );
76
- } while (deployedWorkflow == null || deployedWorkflow .getState ().equals (Workflow .State .ACTIVE ));
77
-
78
- // Return true if the workflow is now active
79
- return deployedWorkflow .getState () != Workflow .State .ACTIVE ;
80
- } catch (Exception e ) {
81
- System .out .printf ("Error deploying workflow: %s\n " , e );
82
- }
55
+ WorkflowsClient workflowsClient = WorkflowsClient .create ();
56
+ // Deploy workflow
57
+ Workflow workflow = Workflow .newBuilder ()
58
+ .setName (workflowId )
59
+ .setSourceContents (WORKFLOW_SOURCE )
60
+ .build ();
61
+ workflowsClient .createWorkflowAsync (location , workflow , workflowId );
62
+
63
+ // Wait until workflow is active
64
+ Workflow deployedWorkflow = null ;
65
+
66
+ System .out .println ("DEPLOY START" );
67
+ // Wait for the deployment to finish
68
+ do {
69
+ System .out .println ("SLEEP" );
70
+ deployedWorkflow = workflowsClient .getWorkflow (WorkflowName .newBuilder ()
71
+ .setProject (projectId )
72
+ .setLocation (location )
73
+ .setWorkflow (workflowId )
74
+ .build ());
75
+ Thread .sleep (2_000 );
76
+ } while (deployedWorkflow == null || deployedWorkflow .getState ().equals (Workflow .State .ACTIVE ));
77
+
78
+ // Return true if the workflow is now active
79
+ return deployedWorkflow .getState () != Workflow .State .ACTIVE ;
83
80
}
84
81
return false ;
85
82
}
0 commit comments