@@ -242,7 +242,7 @@ func (s *MethodTestSuite) TestAPIKey() {
242
242
Returns (slice .New (keyA ,keyB ))
243
243
}))
244
244
s .Run ("GetAPIKeysLastUsedAfter" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
245
- now := time .Now ()
245
+ now := dbtime .Now ()
246
246
a := database.APIKey {LastUsed :now .Add (time .Hour )}
247
247
b := database.APIKey {LastUsed :now .Add (time .Hour )}
248
248
dbm .EXPECT ().GetAPIKeysLastUsedAfter (gomock .Any (),gomock .Any ()).Return ([]database.APIKey {a ,b },nil ).AnyTimes ()
@@ -258,7 +258,7 @@ func (s *MethodTestSuite) TestAPIKey() {
258
258
s .Run ("UpdateAPIKeyByID" ,s .Mocked (func (dbm * dbmock.MockStore ,faker * gofakeit.Faker ,check * expects ) {
259
259
u := testutil .Fake (s .T (),faker , database.User {})
260
260
a := testutil .Fake (s .T (),faker , database.APIKey {UserID :u .ID ,IPAddress :defaultIPAddress ()})
261
- arg := database.UpdateAPIKeyByIDParams {ID :a .ID ,IPAddress :defaultIPAddress (),LastUsed :time .Now (),ExpiresAt :time .Now ().Add (time .Hour )}
261
+ arg := database.UpdateAPIKeyByIDParams {ID :a .ID ,IPAddress :defaultIPAddress (),LastUsed :dbtime .Now (),ExpiresAt :dbtime .Now ().Add (time .Hour )}
262
262
dbm .EXPECT ().GetAPIKeyByID (gomock .Any (),a .ID ).Return (a ,nil ).AnyTimes ()
263
263
dbm .EXPECT ().UpdateAPIKeyByID (gomock .Any (),arg ).Return (nil ).AnyTimes ()
264
264
check .Args (arg ).Asserts (a ,policy .ActionUpdate ).Returns ()
@@ -675,7 +675,7 @@ func (s *MethodTestSuite) TestLicense() {
675
675
s .Run ("GetUnexpiredLicenses" ,s .Mocked (func (db * dbmock.MockStore ,faker * gofakeit.Faker ,check * expects ) {
676
676
l := database.License {
677
677
ID :1 ,
678
- Exp :time .Now ().Add (time .Hour * 24 * 30 ),
678
+ Exp :dbtime .Now ().Add (time .Hour * 24 * 30 ),
679
679
UUID :uuid .New (),
680
680
}
681
681
db .EXPECT ().GetUnexpiredLicenses (gomock .Any ()).
@@ -1089,7 +1089,7 @@ func (s *MethodTestSuite) TestTemplate() {
1089
1089
check .Args (arg ).Asserts (t1 ,policy .ActionRead ).Returns (slice .New (a ,b ))
1090
1090
}))
1091
1091
s .Run ("GetTemplateVersionsCreatedAfter" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
1092
- now := time .Now ()
1092
+ now := dbtime .Now ()
1093
1093
dbm .EXPECT ().GetTemplateVersionsCreatedAfter (gomock .Any (),now .Add (- time .Hour )).Return ([]database.TemplateVersion {},nil ).AnyTimes ()
1094
1094
check .Args (now .Add (- time .Hour )).Asserts (rbac .ResourceTemplate .All (),policy .ActionRead )
1095
1095
}))
@@ -1609,7 +1609,7 @@ func (s *MethodTestSuite) TestUser() {
1609
1609
)
1610
1610
}))
1611
1611
s .Run ("GetUserStatusCounts" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
1612
- arg := database.GetUserStatusCountsParams {StartTime :time .Now ().Add (- time .Hour * 24 * 30 ),EndTime :time .Now (),Interval :int32 ((time .Hour * 24 ).Seconds ())}
1612
+ arg := database.GetUserStatusCountsParams {StartTime :dbtime .Now ().Add (- time .Hour * 24 * 30 ),EndTime :dbtime .Now (),Interval :int32 ((time .Hour * 24 ).Seconds ())}
1613
1613
dbm .EXPECT ().GetUserStatusCounts (gomock .Any (),arg ).Return ([]database.GetUserStatusCountsRow {},nil ).AnyTimes ()
1614
1614
check .Args (arg ).Asserts (rbac .ResourceUser ,policy .ActionRead )
1615
1615
}))
@@ -2640,7 +2640,7 @@ func (s *MethodTestSuite) TestCryptoKeys() {
2640
2640
}))
2641
2641
s .Run ("UpdateCryptoKeyDeletesAt" ,s .Mocked (func (dbm * dbmock.MockStore ,faker * gofakeit.Faker ,check * expects ) {
2642
2642
key := testutil .Fake (s .T (),faker , database.CryptoKey {Feature :database .CryptoKeyFeatureWorkspaceAppsAPIKey ,Sequence :4 })
2643
- arg := database.UpdateCryptoKeyDeletesAtParams {Feature :key .Feature ,Sequence :key .Sequence ,DeletesAt : sql.NullTime {Time :time .Now (),Valid :true }}
2643
+ arg := database.UpdateCryptoKeyDeletesAtParams {Feature :key .Feature ,Sequence :key .Sequence ,DeletesAt : sql.NullTime {Time :dbtime .Now (),Valid :true }}
2644
2644
dbm .EXPECT ().UpdateCryptoKeyDeletesAt (gomock .Any (),arg ).Return (key ,nil ).AnyTimes ()
2645
2645
check .Args (arg ).Asserts (rbac .ResourceCryptoKey ,policy .ActionUpdate )
2646
2646
}))
@@ -2666,7 +2666,7 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2666
2666
check .Args (ids ).Asserts (rbac .ResourceSystem ,policy .ActionRead )
2667
2667
}))
2668
2668
s .Run ("GetWorkspaceAppStatusesByAppIDs" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
2669
- ids := []uuid.UUID { uuid . New ()}
2669
+ var ids []uuid.UUID
2670
2670
dbm .EXPECT ().GetWorkspaceAppStatusesByAppIDs (gomock .Any (),ids ).Return ([]database.WorkspaceAppStatus {},nil ).AnyTimes ()
2671
2671
check .Args (ids ).Asserts (rbac .ResourceSystem ,policy .ActionRead )
2672
2672
}))
@@ -2725,12 +2725,12 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2725
2725
check .Args (arg ).Asserts (rbac .ResourceSystem ,policy .ActionUpdate )
2726
2726
}))
2727
2727
s .Run ("DeleteReplicasUpdatedBefore" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
2728
- t := time .Now ().Add (time .Hour )
2728
+ t := dbtime .Now ().Add (time .Hour )
2729
2729
dbm .EXPECT ().DeleteReplicasUpdatedBefore (gomock .Any (),t ).Return (nil ).AnyTimes ()
2730
2730
check .Args (t ).Asserts (rbac .ResourceSystem ,policy .ActionDelete )
2731
2731
}))
2732
2732
s .Run ("GetReplicasUpdatedAfter" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
2733
- t := time .Now ().Add (- time .Hour )
2733
+ t := dbtime .Now ().Add (- time .Hour )
2734
2734
dbm .EXPECT ().GetReplicasUpdatedAfter (gomock .Any (),t ).Return ([]database.Replica {},nil ).AnyTimes ()
2735
2735
check .Args (t ).Asserts (rbac .ResourceSystem ,policy .ActionRead )
2736
2736
}))
@@ -2763,27 +2763,27 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2763
2763
check .Args ().Asserts (rbac .ResourceSystem ,policy .ActionRead )
2764
2764
}))
2765
2765
s .Run ("GetWorkspaceBuildsCreatedAfter" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
2766
- ts := time .Now ()
2766
+ ts := dbtime .Now ()
2767
2767
dbm .EXPECT ().GetWorkspaceBuildsCreatedAfter (gomock .Any (),ts ).Return ([]database.WorkspaceBuild {},nil ).AnyTimes ()
2768
2768
check .Args (ts ).Asserts (rbac .ResourceSystem ,policy .ActionRead )
2769
2769
}))
2770
2770
s .Run ("GetWorkspaceAgentsCreatedAfter" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
2771
- ts := time .Now ()
2771
+ ts := dbtime .Now ()
2772
2772
dbm .EXPECT ().GetWorkspaceAgentsCreatedAfter (gomock .Any (),ts ).Return ([]database.WorkspaceAgent {},nil ).AnyTimes ()
2773
2773
check .Args (ts ).Asserts (rbac .ResourceSystem ,policy .ActionRead )
2774
2774
}))
2775
2775
s .Run ("GetWorkspaceAppsCreatedAfter" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
2776
- ts := time .Now ()
2776
+ ts := dbtime .Now ()
2777
2777
dbm .EXPECT ().GetWorkspaceAppsCreatedAfter (gomock .Any (),ts ).Return ([]database.WorkspaceApp {},nil ).AnyTimes ()
2778
2778
check .Args (ts ).Asserts (rbac .ResourceSystem ,policy .ActionRead )
2779
2779
}))
2780
2780
s .Run ("GetWorkspaceResourcesCreatedAfter" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
2781
- ts := time .Now ()
2781
+ ts := dbtime .Now ()
2782
2782
dbm .EXPECT ().GetWorkspaceResourcesCreatedAfter (gomock .Any (),ts ).Return ([]database.WorkspaceResource {},nil ).AnyTimes ()
2783
2783
check .Args (ts ).Asserts (rbac .ResourceSystem ,policy .ActionRead )
2784
2784
}))
2785
2785
s .Run ("GetWorkspaceResourceMetadataCreatedAfter" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
2786
- ts := time .Now ()
2786
+ ts := dbtime .Now ()
2787
2787
dbm .EXPECT ().GetWorkspaceResourceMetadataCreatedAfter (gomock .Any (),ts ).Return ([]database.WorkspaceResourceMetadatum {},nil ).AnyTimes ()
2788
2788
check .Args (ts ).Asserts (rbac .ResourceSystem ,policy .ActionRead )
2789
2789
}))
@@ -2792,7 +2792,7 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2792
2792
check .Args ().Asserts (rbac .ResourceSystem ,policy .ActionDelete )
2793
2793
}))
2794
2794
s .Run ("GetProvisionerJobsCreatedAfter" ,s .Mocked (func (dbm * dbmock.MockStore ,_ * gofakeit.Faker ,check * expects ) {
2795
- ts := time .Now ()
2795
+ ts := dbtime .Now ()
2796
2796
dbm .EXPECT ().GetProvisionerJobsCreatedAfter (gomock .Any (),ts ).Return ([]database.ProvisionerJob {},nil ).AnyTimes ()
2797
2797
check .Args (ts ).Asserts (rbac .ResourceProvisionerJobs ,policy .ActionRead )
2798
2798
}))
@@ -2901,7 +2901,7 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2901
2901
check .Args (arg ).Asserts (rbac .ResourceSystem ,policy .ActionUpdate ).Returns ()
2902
2902
}))
2903
2903
s .Run ("AcquireProvisionerJob" ,s .Mocked (func (dbm * dbmock.MockStore ,faker * gofakeit.Faker ,check * expects ) {
2904
- arg := database.AcquireProvisionerJobParams {StartedAt : sql.NullTime {Valid :true ,Time :time .Now ()},OrganizationID :uuid .New (),Types : []database.ProvisionerType {database .ProvisionerTypeEcho },ProvisionerTags :json .RawMessage ("{}" )}
2904
+ arg := database.AcquireProvisionerJobParams {StartedAt : sql.NullTime {Valid :true ,Time :dbtime .Now ()},OrganizationID :uuid .New (),Types : []database.ProvisionerType {database .ProvisionerTypeEcho },ProvisionerTags :json .RawMessage ("{}" )}
2905
2905
dbm .EXPECT ().AcquireProvisionerJob (gomock .Any (),arg ).Return (testutil .Fake (s .T (),faker , database.ProvisionerJob {}),nil ).AnyTimes ()
2906
2906
check .Args (arg ).Asserts (rbac .ResourceProvisionerJobs ,policy .ActionUpdate )
2907
2907
}))
@@ -2919,7 +2919,7 @@ func (s *MethodTestSuite) TestSystemFunctions() {
2919
2919
}))
2920
2920
s .Run ("UpdateProvisionerJobByID" ,s .Mocked (func (dbm * dbmock.MockStore ,faker * gofakeit.Faker ,check * expects ) {
2921
2921
j := testutil .Fake (s .T (),faker , database.ProvisionerJob {})
2922
- arg := database.UpdateProvisionerJobByIDParams {ID :j .ID ,UpdatedAt :time .Now ()}
2922
+ arg := database.UpdateProvisionerJobByIDParams {ID :j .ID ,UpdatedAt :dbtime .Now ()}
2923
2923
dbm .EXPECT ().UpdateProvisionerJobByID (gomock .Any (),arg ).Return (nil ).AnyTimes ()
2924
2924
check .Args (arg ).Asserts (rbac .ResourceProvisionerJobs ,policy .ActionUpdate )
2925
2925
}))
@@ -3785,7 +3785,7 @@ func (s *MethodTestSuite) TestOAuth2ProviderAppSecrets() {
3785
3785
secrets := []database.OAuth2ProviderAppSecret {
3786
3786
dbgen .OAuth2ProviderAppSecret (s .T (),db , database.OAuth2ProviderAppSecret {
3787
3787
AppID :app1 .ID ,
3788
- CreatedAt :time .Now ().Add (- time .Hour ),// For ordering.
3788
+ CreatedAt :dbtime .Now ().Add (- time .Hour ),// For ordering.
3789
3789
SecretPrefix : []byte ("1" ),
3790
3790
}),
3791
3791
dbgen .OAuth2ProviderAppSecret (s .T (),db , database.OAuth2ProviderAppSecret {