@@ -36,12 +36,12 @@ func TestNoReconciliationActionsIfNoPresets(t *testing.T) {
36
36
37
37
clock := quartz .NewMock (t )
38
38
ctx := testutil .Context (t , testutil .WaitLong )
39
- db , pubsub := dbtestutil .NewDB (t )
39
+ db , ps := dbtestutil .NewDB (t )
40
40
cfg := codersdk.PrebuildsConfig {
41
41
ReconciliationInterval : serpent .Duration (testutil .WaitLong ),
42
42
}
43
43
logger := testutil .Logger (t )
44
- controller := prebuilds .NewStoreReconciler (db , pubsub , cfg , logger , quartz .NewMock (t ))
44
+ controller := prebuilds .NewStoreReconciler (db , ps , cfg , logger , quartz .NewMock (t ))
45
45
46
46
// given a template version with no presets
47
47
org := dbgen .Organization (t , db , database.Organization {})
@@ -81,12 +81,12 @@ func TestNoReconciliationActionsIfNoPrebuilds(t *testing.T) {
81
81
82
82
clock := quartz .NewMock (t )
83
83
ctx := testutil .Context (t , testutil .WaitLong )
84
- db , pubsub := dbtestutil .NewDB (t )
84
+ db , ps := dbtestutil .NewDB (t )
85
85
cfg := codersdk.PrebuildsConfig {
86
86
ReconciliationInterval : serpent .Duration (testutil .WaitLong ),
87
87
}
88
88
logger := testutil .Logger (t )
89
- controller := prebuilds .NewStoreReconciler (db , pubsub , cfg , logger , quartz .NewMock (t ))
89
+ controller := prebuilds .NewStoreReconciler (db , ps , cfg , logger , quartz .NewMock (t ))
90
90
91
91
// given there are presets, but no prebuilds
92
92
org := dbgen .Organization (t , db , database.Organization {})
@@ -300,8 +300,8 @@ func TestPrebuildReconciliation(t *testing.T) {
300
300
})
301
301
orgID , templateID := setupTestDBTemplate (t , db , ownerID )
302
302
templateVersionID := setupTestDBTemplateVersion (
303
- t ,
304
303
ctx ,
304
+ t ,
305
305
clock ,
306
306
db ,
307
307
pubsub ,
@@ -310,8 +310,8 @@ func TestPrebuildReconciliation(t *testing.T) {
310
310
templateID ,
311
311
)
312
312
preset := setupTestDBPreset (
313
- t ,
314
313
ctx ,
314
+ t ,
315
315
db ,
316
316
pubsub ,
317
317
templateVersionID ,
@@ -335,7 +335,7 @@ func TestPrebuildReconciliation(t *testing.T) {
335
335
if ! templateVersionActive {
336
336
// Create a new template version and mark it as active
337
337
// This marks the template version that we care about as inactive
338
- setupTestDBTemplateVersion (t , ctx , clock , db , pubsub , orgID , ownerID , templateID )
338
+ setupTestDBTemplateVersion (ctx , t , clock , db , pubsub , orgID , ownerID , templateID )
339
339
}
340
340
341
341
// Run the reconciliation multiple times to ensure idempotency
@@ -408,9 +408,9 @@ func TestFailedBuildBackoff(t *testing.T) {
408
408
ID : userID ,
409
409
})
410
410
orgID , templateID := setupTestDBTemplate (t , db , userID )
411
- templateVersionID := setupTestDBTemplateVersion (t , ctx , clock , db , ps , orgID , userID , templateID )
411
+ templateVersionID := setupTestDBTemplateVersion (ctx , t , clock , db , ps , orgID , userID , templateID )
412
412
413
- preset := setupTestDBPreset (t , ctx , db , ps , templateVersionID , desiredInstances , "test" )
413
+ preset := setupTestDBPreset (ctx , t , db , ps , templateVersionID , desiredInstances , "test" )
414
414
for range desiredInstances {
415
415
_ = setupTestDBPrebuild (t , ctx , clock , db , ps , database .WorkspaceTransitionStart , database .ProvisionerJobStatusFailed , orgID , preset , templateID , templateVersionID )
416
416
}
@@ -513,8 +513,8 @@ const (
513
513
)
514
514
515
515
func setupTestDBTemplateVersion (
516
- t * testing.T ,
517
516
ctx context.Context ,
517
+ t * testing.T ,
518
518
clock quartz.Clock ,
519
519
db database.Store ,
520
520
ps pubsub.Pubsub ,
@@ -544,8 +544,8 @@ func setupTestDBTemplateVersion(
544
544
}
545
545
546
546
func setupTestDBPreset (
547
- t * testing.T ,
548
547
ctx context.Context ,
548
+ t * testing.T ,
549
549
db database.Store ,
550
550
ps pubsub.Pubsub ,
551
551
templateVersionID uuid.UUID ,
@@ -585,12 +585,12 @@ func setupTestDBPrebuild(
585
585
templateVersionID uuid.UUID ,
586
586
) database.WorkspaceTable {
587
587
t .Helper ()
588
- return setupTestDBWorkspace (t , ctx , clock , db , ps , transition , prebuildStatus , orgID , preset , templateID , templateVersionID , prebuilds .OwnerID , prebuilds .OwnerID )
588
+ return setupTestDBWorkspace (ctx , t , clock , db , ps , transition , prebuildStatus , orgID , preset , templateID , templateVersionID , prebuilds .OwnerID , prebuilds .OwnerID )
589
589
}
590
590
591
591
func setupTestDBWorkspace (
592
- t * testing.T ,
593
592
ctx context.Context ,
593
+ t * testing.T ,
594
594
clock quartz.Clock ,
595
595
db database.Store ,
596
596
ps pubsub.Pubsub ,
0 commit comments