@@ -91,14 +91,16 @@ func TestApplyRunRetentionOnProject_WorkflowWithRetention(t *testing.T) {
9191 Initiator : & sdk.V2Initiator {
9292 UserID : lambdauser .ID ,
9393 },
94- RunNumber : 0 ,
95- WorkflowRef : "" ,
94+ RunNumber : 0 ,
95+ Contexts : sdk.WorkflowRunContext {
96+ Git : sdk.GitContext {Ref : "" },
97+ },
9698 }
9799 // Create run on master - We must keep only run_number 19 and 20
98100 for i := 0 ; i < 20 ; i ++ {
99101 wr .RunNumber = int64 (runNumber )
100102 runNumber ++
101- wr .WorkflowRef = "refs/heads/master"
103+ wr .Contexts . Git . Ref = "refs/heads/master"
102104
103105 require .NoError (t , workflow_v2 .InsertRun (ctx , db , & wr ))
104106
@@ -113,15 +115,15 @@ func TestApplyRunRetentionOnProject_WorkflowWithRetention(t *testing.T) {
113115 for i := 0 ; i < 20 ; i ++ {
114116 wr .RunNumber = int64 (runNumber )
115117 runNumber ++
116- wr .WorkflowRef = "refs/heads/dev/my/feat"
118+ wr .Contexts . Git . Ref = "refs/heads/dev/my/feat"
117119 require .NoError (t , workflow_v2 .InsertRun (ctx , db , & wr ))
118120 }
119121
120122 // Create run on fix branch. We must keep ony run_number 60
121123 for i := 0 ; i < 20 ; i ++ {
122124 wr .RunNumber = int64 (runNumber )
123125 runNumber ++
124- wr .WorkflowRef = "refs/heads/fix"
126+ wr .Contexts . Git . Ref = "refs/heads/fix"
125127 require .NoError (t , workflow_v2 .InsertRun (ctx , db , & wr ))
126128 }
127129
@@ -201,14 +203,18 @@ func TestApplyRunRetentionOnProject_FallbackProjectDefaultRule(t *testing.T) {
201203 Initiator : & sdk.V2Initiator {
202204 UserID : lambdauser .ID ,
203205 },
204- RunNumber : 0 ,
205- WorkflowRef : "" ,
206+ RunNumber : 0 ,
207+ Contexts : sdk.WorkflowRunContext {
208+ Git : sdk.GitContext {
209+ Ref : "" ,
210+ },
211+ },
206212 }
207213 // Create run on master - We must keep only run_number 19 and 20
208214 for i := 0 ; i < 20 ; i ++ {
209215 wr .RunNumber = int64 (runNumber )
210216 runNumber ++
211- wr .WorkflowRef = "refs/heads/master"
217+ wr .Contexts . Git . Ref = "refs/heads/master"
212218 require .NoError (t , workflow_v2 .InsertRun (ctx , db , & wr ))
213219
214220 }
0 commit comments