@@ -3,6 +3,7 @@ package workflow_test
33import (
44 "context"
55 "fmt"
6+ "github.com/stretchr/testify/assert"
67 "testing"
78
89 "github.com/ovh/cds/engine/api/bootstrap"
@@ -131,13 +132,18 @@ func TestPurgeWorkflowRun(t *testing.T) {
131132 test .NoError (t , err )
132133
133134 for i := 0 ; i < 5 ; i ++ {
134- _ , _ , errWr := workflow .ManualRun (context .TODO (), db , cache , proj , w1 , & sdk.WorkflowNodeRunManual {
135- User : * u ,
136- Payload : map [string ]string {
137- "git.branch" : "master" ,
138- "git.author" : "test" ,
135+ wr , errWR := workflow .CreateRun (db , w1 , nil , u )
136+ assert .NoError (t , errWR )
137+ wr .Workflow = * w1
138+ _ , errWr := workflow .StartWorkflowRun (context .TODO (), db , cache , proj , wr , & sdk.WorkflowRunPostHandlerOption {
139+ Manual : & sdk.WorkflowNodeRunManual {
140+ User : * u ,
141+ Payload : map [string ]string {
142+ "git.branch" : "master" ,
143+ "git.author" : "test" ,
144+ },
139145 },
140- }, nil )
146+ }, u , nil )
141147 test .NoError (t , errWr )
142148 }
143149
@@ -222,15 +228,19 @@ func TestPurgeWorkflowRunWithRunningStatus(t *testing.T) {
222228 test .NoError (t , err )
223229
224230 for i := 0 ; i < 5 ; i ++ {
225- wfr , _ , errWr := workflow .ManualRun (context .TODO (), db , cache , proj , w1 , & sdk.WorkflowNodeRunManual {
226- User : * u ,
227- Payload : map [string ]string {
228- "git.branch" : "master" ,
229- "git.author" : "test" ,
231+ wfr , errWR := workflow .CreateRun (db , w1 , nil , u )
232+ assert .NoError (t , errWR )
233+ wfr .Workflow = * w1
234+ _ , errWr := workflow .StartWorkflowRun (context .TODO (), db , cache , proj , wfr , & sdk.WorkflowRunPostHandlerOption {
235+ Manual : & sdk.WorkflowNodeRunManual {
236+ User : * u ,
237+ Payload : map [string ]string {
238+ "git.branch" : "master" ,
239+ "git.author" : "test" ,
240+ },
230241 },
231- }, nil )
242+ }, u , nil )
232243 test .NoError (t , errWr )
233-
234244 wfr .Status = sdk .StatusBuilding .String ()
235245 test .NoError (t , workflow .UpdateWorkflowRunStatus (db , wfr ))
236246 }
@@ -315,23 +325,33 @@ func TestPurgeWorkflowRunWithOneSuccessWorkflowRun(t *testing.T) {
315325 })
316326 test .NoError (t , err )
317327
318- _ , _ , errWr := workflow .ManualRun (context .TODO (), db , cache , proj , w1 , & sdk.WorkflowNodeRunManual {
319- User : * u ,
320- Payload : map [string ]string {
321- "git.branch" : "master" ,
322- "git.author" : "test" ,
323- },
324- }, nil )
325- test .NoError (t , errWr )
326-
327- for i := 0 ; i < 5 ; i ++ {
328- wfr , _ , errWr := workflow .ManualRun (context .TODO (), db , cache , proj , w1 , & sdk.WorkflowNodeRunManual {
328+ wr , errWR := workflow .CreateRun (db , w1 , nil , u )
329+ assert .NoError (t , errWR )
330+ wr .Workflow = * w1
331+ _ , errWr := workflow .StartWorkflowRun (context .TODO (), db , cache , proj , wr , & sdk.WorkflowRunPostHandlerOption {
332+ Manual : & sdk.WorkflowNodeRunManual {
329333 User : * u ,
330334 Payload : map [string ]string {
331335 "git.branch" : "master" ,
332336 "git.author" : "test" ,
333337 },
334- }, nil )
338+ },
339+ }, u , nil )
340+ test .NoError (t , errWr )
341+
342+ for i := 0 ; i < 5 ; i ++ {
343+ wfr , errWR := workflow .CreateRun (db , w1 , nil , u )
344+ assert .NoError (t , errWR )
345+ wfr .Workflow = * w1
346+ _ , errWr := workflow .StartWorkflowRun (context .TODO (), db , cache , proj , wfr , & sdk.WorkflowRunPostHandlerOption {
347+ Manual : & sdk.WorkflowNodeRunManual {
348+ User : * u ,
349+ Payload : map [string ]string {
350+ "git.branch" : "master" ,
351+ "git.author" : "test" ,
352+ },
353+ },
354+ }, u , nil )
335355 test .NoError (t , errWr )
336356
337357 wfr .Status = sdk .StatusFail .String ()
@@ -423,13 +443,18 @@ func TestPurgeWorkflowRunWithNoSuccessWorkflowRun(t *testing.T) {
423443 test .NoError (t , err )
424444
425445 for i := 0 ; i < 5 ; i ++ {
426- wfr , _ , errWr := workflow .ManualRun (context .TODO (), db , cache , proj , w1 , & sdk.WorkflowNodeRunManual {
427- User : * u ,
428- Payload : map [string ]string {
429- "git.branch" : "master" ,
430- "git.author" : "test" ,
446+ wfr , errWR := workflow .CreateRun (db , w1 , nil , u )
447+ assert .NoError (t , errWR )
448+ wfr .Workflow = * w1
449+ _ , errWr := workflow .StartWorkflowRun (context .TODO (), db , cache , proj , wfr , & sdk.WorkflowRunPostHandlerOption {
450+ Manual : & sdk.WorkflowNodeRunManual {
451+ User : * u ,
452+ Payload : map [string ]string {
453+ "git.branch" : "master" ,
454+ "git.author" : "test" ,
455+ },
431456 },
432- }, nil )
457+ }, u , nil )
433458 test .NoError (t , errWr )
434459
435460 wfr .Status = sdk .StatusFail .String ()
@@ -515,13 +540,18 @@ func TestPurgeWorkflowRunWithoutTags(t *testing.T) {
515540
516541 branches := []string {"master" , "master" , "master" , "develop" , "develop" , "testBr" , "testBr" , "testBr" , "testBr" , "test4" }
517542 for i := 0 ; i < 10 ; i ++ {
518- _ , _ , errWr := workflow .ManualRun (context .TODO (), db , cache , proj , w1 , & sdk.WorkflowNodeRunManual {
519- User : * u ,
520- Payload : map [string ]string {
521- "git.branch" : branches [i ],
522- "git.author" : "test" ,
543+ wr , errWR := workflow .CreateRun (db , w1 , nil , u )
544+ assert .NoError (t , errWR )
545+ wr .Workflow = * w1
546+ _ , errWr := workflow .StartWorkflowRun (context .TODO (), db , cache , proj , wr , & sdk.WorkflowRunPostHandlerOption {
547+ Manual : & sdk.WorkflowNodeRunManual {
548+ User : * u ,
549+ Payload : map [string ]string {
550+ "git.branch" : branches [i ],
551+ "git.author" : "test" ,
552+ },
523553 },
524- }, nil )
554+ }, u , nil )
525555 test .NoError (t , errWr )
526556 }
527557
@@ -604,13 +634,18 @@ func TestPurgeWorkflowRunWithoutTagsBiggerHistoryLength(t *testing.T) {
604634
605635 branches := []string {"master" , "master" , "master" , "develop" , "develop" , "testBr" , "testBr" , "testBr" , "testBr" , "test4" }
606636 for i := 0 ; i < 10 ; i ++ {
607- _ , _ , errWr := workflow .ManualRun (context .TODO (), db , cache , proj , w1 , & sdk.WorkflowNodeRunManual {
608- User : * u ,
609- Payload : map [string ]string {
610- "git.branch" : branches [i ],
611- "git.author" : "test" ,
637+ wr , errWR := workflow .CreateRun (db , w1 , nil , u )
638+ assert .NoError (t , errWR )
639+ wr .Workflow = * w1
640+ _ , errWr := workflow .StartWorkflowRun (context .TODO (), db , cache , proj , wr , & sdk.WorkflowRunPostHandlerOption {
641+ Manual : & sdk.WorkflowNodeRunManual {
642+ User : * u ,
643+ Payload : map [string ]string {
644+ "git.branch" : branches [i ],
645+ "git.author" : "test" ,
646+ },
612647 },
613- }, nil )
648+ }, u , nil )
614649 test .NoError (t , errWr )
615650 }
616651
0 commit comments