@@ -4727,248 +4727,112 @@ func (s *MethodTestSuite) TestNotifications() {
4727
4727
}
4728
4728
4729
4729
func (s * MethodTestSuite ) TestPrebuilds () {
4730
- s .Run ("GetPresetByWorkspaceBuildID" , s .Subtest (func (db database.Store , check * expects ) {
4731
- org := dbgen .Organization (s .T (), db , database.Organization {})
4732
- user := dbgen .User (s .T (), db , database.User {})
4733
- template := dbgen .Template (s .T (), db , database.Template {
4734
- CreatedBy : user .ID ,
4735
- OrganizationID : org .ID ,
4736
- })
4737
- templateVersion := dbgen .TemplateVersion (s .T (), db , database.TemplateVersion {
4738
- TemplateID : uuid.NullUUID {UUID : template .ID , Valid : true },
4739
- OrganizationID : org .ID ,
4740
- CreatedBy : user .ID ,
4741
- })
4742
- preset , err := db .InsertPreset (context .Background (), database.InsertPresetParams {
4743
- TemplateVersionID : templateVersion .ID ,
4744
- Name : "test" ,
4745
- })
4746
- require .NoError (s .T (), err )
4747
- workspace := dbgen .Workspace (s .T (), db , database.WorkspaceTable {
4748
- OrganizationID : org .ID ,
4749
- OwnerID : user .ID ,
4750
- TemplateID : template .ID ,
4751
- })
4752
- job := dbgen .ProvisionerJob (s .T (), db , nil , database.ProvisionerJob {
4753
- OrganizationID : org .ID ,
4754
- })
4755
- workspaceBuild := dbgen .WorkspaceBuild (s .T (), db , database.WorkspaceBuild {
4756
- WorkspaceID : workspace .ID ,
4757
- TemplateVersionID : templateVersion .ID ,
4758
- TemplateVersionPresetID : uuid.NullUUID {UUID : preset .ID , Valid : true },
4759
- InitiatorID : user .ID ,
4760
- JobID : job .ID ,
4761
- })
4762
- _ , err = db .GetPresetByWorkspaceBuildID (context .Background (), workspaceBuild .ID )
4763
- require .NoError (s .T (), err )
4764
- check .Args (workspaceBuild .ID ).Asserts (rbac .ResourceTemplate , policy .ActionRead )
4765
- }))
4766
- s .Run ("GetPresetParametersByTemplateVersionID" , s .Subtest (func (db database.Store , check * expects ) {
4767
- ctx := context .Background ()
4768
- org := dbgen .Organization (s .T (), db , database.Organization {})
4769
- user := dbgen .User (s .T (), db , database.User {})
4770
- template := dbgen .Template (s .T (), db , database.Template {
4771
- CreatedBy : user .ID ,
4772
- OrganizationID : org .ID ,
4773
- })
4774
- templateVersion := dbgen .TemplateVersion (s .T (), db , database.TemplateVersion {
4775
- TemplateID : uuid.NullUUID {UUID : template .ID , Valid : true },
4776
- OrganizationID : org .ID ,
4777
- CreatedBy : user .ID ,
4778
- })
4779
- preset , err := db .InsertPreset (ctx , database.InsertPresetParams {
4780
- TemplateVersionID : templateVersion .ID ,
4781
- Name : "test" ,
4782
- })
4783
- require .NoError (s .T (), err )
4784
- insertedParameters , err := db .InsertPresetParameters (ctx , database.InsertPresetParametersParams {
4785
- TemplateVersionPresetID : preset .ID ,
4786
- Names : []string {"test" },
4787
- Values : []string {"test" },
4788
- })
4789
- require .NoError (s .T (), err )
4790
- check .
4791
- Args (templateVersion .ID ).
4792
- Asserts (template .RBACObject (), policy .ActionRead ).
4793
- Returns (insertedParameters )
4794
- }))
4795
- s .Run ("GetPresetParametersByPresetID" , s .Subtest (func (db database.Store , check * expects ) {
4796
- ctx := context .Background ()
4797
- org := dbgen .Organization (s .T (), db , database.Organization {})
4798
- user := dbgen .User (s .T (), db , database.User {})
4799
- template := dbgen .Template (s .T (), db , database.Template {
4800
- CreatedBy : user .ID ,
4801
- OrganizationID : org .ID ,
4802
- })
4803
- templateVersion := dbgen .TemplateVersion (s .T (), db , database.TemplateVersion {
4804
- TemplateID : uuid.NullUUID {UUID : template .ID , Valid : true },
4805
- OrganizationID : org .ID ,
4806
- CreatedBy : user .ID ,
4807
- })
4808
- preset , err := db .InsertPreset (ctx , database.InsertPresetParams {
4809
- TemplateVersionID : templateVersion .ID ,
4810
- Name : "test" ,
4811
- })
4812
- require .NoError (s .T (), err )
4813
- insertedParameters , err := db .InsertPresetParameters (ctx , database.InsertPresetParametersParams {
4814
- TemplateVersionPresetID : preset .ID ,
4815
- Names : []string {"test" },
4816
- Values : []string {"test" },
4817
- })
4818
- require .NoError (s .T (), err )
4819
- check .
4820
- Args (preset .ID ).
4821
- Asserts (template .RBACObject (), policy .ActionRead ).
4822
- Returns (insertedParameters )
4823
- }))
4824
- s .Run ("GetActivePresetPrebuildSchedules" , s .Subtest (func (db database.Store , check * expects ) {
4825
- check .Args ().
4826
- Asserts (rbac .ResourceTemplate .All (), policy .ActionRead ).
4827
- Returns ([]database.TemplateVersionPresetPrebuildSchedule {})
4828
- }))
4829
- s .Run ("GetPresetsByTemplateVersionID" , s .Subtest (func (db database.Store , check * expects ) {
4830
- ctx := context .Background ()
4831
- org := dbgen .Organization (s .T (), db , database.Organization {})
4832
- user := dbgen .User (s .T (), db , database.User {})
4833
- template := dbgen .Template (s .T (), db , database.Template {
4834
- CreatedBy : user .ID ,
4835
- OrganizationID : org .ID ,
4836
- })
4837
- templateVersion := dbgen .TemplateVersion (s .T (), db , database.TemplateVersion {
4838
- TemplateID : uuid.NullUUID {UUID : template .ID , Valid : true },
4839
- OrganizationID : org .ID ,
4840
- CreatedBy : user .ID ,
4841
- })
4842
-
4843
- _ , err := db .InsertPreset (ctx , database.InsertPresetParams {
4844
- TemplateVersionID : templateVersion .ID ,
4845
- Name : "test" ,
4846
- })
4847
- require .NoError (s .T (), err )
4848
-
4849
- presets , err := db .GetPresetsByTemplateVersionID (ctx , templateVersion .ID )
4850
- require .NoError (s .T (), err )
4851
-
4852
- check .Args (templateVersion .ID ).Asserts (template .RBACObject (), policy .ActionRead ).Returns (presets )
4853
- }))
4854
- s .Run ("ClaimPrebuiltWorkspace" , s .Subtest (func (db database.Store , check * expects ) {
4855
- org := dbgen .Organization (s .T (), db , database.Organization {})
4856
- user := dbgen .User (s .T (), db , database.User {})
4857
- template := dbgen .Template (s .T (), db , database.Template {
4858
- OrganizationID : org .ID ,
4859
- CreatedBy : user .ID ,
4860
- })
4861
- templateVersion := dbgen .TemplateVersion (s .T (), db , database.TemplateVersion {
4862
- TemplateID : uuid.NullUUID {
4863
- UUID : template .ID ,
4864
- Valid : true ,
4865
- },
4866
- OrganizationID : org .ID ,
4867
- CreatedBy : user .ID ,
4868
- })
4869
- preset := dbgen .Preset (s .T (), db , database.InsertPresetParams {
4870
- TemplateVersionID : templateVersion .ID ,
4871
- })
4872
- check .Args (database.ClaimPrebuiltWorkspaceParams {
4873
- NewUserID : user .ID ,
4874
- NewName : "" ,
4875
- PresetID : preset .ID ,
4876
- }).Asserts (
4877
- rbac .ResourceWorkspace .WithOwner (user .ID .String ()).InOrg (org .ID ), policy .ActionCreate ,
4878
- template , policy .ActionRead ,
4879
- template , policy .ActionUse ,
4880
- ).Errors (sql .ErrNoRows )
4881
- }))
4882
- s .Run ("GetPrebuildMetrics" , s .Subtest (func (_ database.Store , check * expects ) {
4883
- check .Args ().
4884
- Asserts (rbac .ResourceWorkspace .All (), policy .ActionRead )
4885
- }))
4886
- s .Run ("GetPrebuildsSettings" , s .Subtest (func (db database.Store , check * expects ) {
4887
- check .Args ().Asserts ()
4888
- }))
4889
- s .Run ("UpsertPrebuildsSettings" , s .Subtest (func (db database.Store , check * expects ) {
4890
- check .Args ("foo" ).Asserts (rbac .ResourceDeploymentConfig , policy .ActionUpdate )
4891
- }))
4892
- s .Run ("CountInProgressPrebuilds" , s .Subtest (func (_ database.Store , check * expects ) {
4893
- check .Args ().
4894
- Asserts (rbac .ResourceWorkspace .All (), policy .ActionRead )
4895
- }))
4896
- s .Run ("GetPresetsAtFailureLimit" , s .Subtest (func (_ database.Store , check * expects ) {
4897
- check .Args (int64 (0 )).
4898
- Asserts (rbac .ResourceTemplate .All (), policy .ActionViewInsights )
4899
- }))
4900
- s .Run ("GetPresetsBackoff" , s .Subtest (func (_ database.Store , check * expects ) {
4901
- check .Args (time.Time {}).
4902
- Asserts (rbac .ResourceTemplate .All (), policy .ActionViewInsights )
4903
- }))
4904
- s .Run ("GetRunningPrebuiltWorkspaces" , s .Subtest (func (_ database.Store , check * expects ) {
4905
- check .Args ().
4906
- Asserts (rbac .ResourceWorkspace .All (), policy .ActionRead )
4907
- }))
4908
- s .Run ("GetTemplatePresetsWithPrebuilds" , s .Subtest (func (db database.Store , check * expects ) {
4909
- user := dbgen .User (s .T (), db , database.User {})
4910
- check .Args (uuid.NullUUID {UUID : user .ID , Valid : true }).
4911
- Asserts (rbac .ResourceTemplate .All (), policy .ActionRead )
4912
- }))
4913
- s .Run ("GetPresetByID" , s .Subtest (func (db database.Store , check * expects ) {
4914
- org := dbgen .Organization (s .T (), db , database.Organization {})
4915
- user := dbgen .User (s .T (), db , database.User {})
4916
- template := dbgen .Template (s .T (), db , database.Template {
4917
- OrganizationID : org .ID ,
4918
- CreatedBy : user .ID ,
4919
- })
4920
- templateVersion := dbgen .TemplateVersion (s .T (), db , database.TemplateVersion {
4921
- TemplateID : uuid.NullUUID {
4922
- UUID : template .ID ,
4923
- Valid : true ,
4924
- },
4925
- OrganizationID : org .ID ,
4926
- CreatedBy : user .ID ,
4927
- })
4928
- preset := dbgen .Preset (s .T (), db , database.InsertPresetParams {
4929
- TemplateVersionID : templateVersion .ID ,
4930
- })
4931
- check .Args (preset .ID ).
4932
- Asserts (template , policy .ActionRead ).
4933
- Returns (database.GetPresetByIDRow {
4934
- ID : preset .ID ,
4935
- TemplateVersionID : preset .TemplateVersionID ,
4936
- Name : preset .Name ,
4937
- CreatedAt : preset .CreatedAt ,
4938
- TemplateID : uuid.NullUUID {
4939
- UUID : template .ID ,
4940
- Valid : true ,
4941
- },
4942
- InvalidateAfterSecs : preset .InvalidateAfterSecs ,
4943
- OrganizationID : org .ID ,
4944
- PrebuildStatus : database .PrebuildStatusHealthy ,
4945
- })
4946
- }))
4947
- s .Run ("UpdatePresetPrebuildStatus" , s .Subtest (func (db database.Store , check * expects ) {
4948
- org := dbgen .Organization (s .T (), db , database.Organization {})
4949
- user := dbgen .User (s .T (), db , database.User {})
4950
- template := dbgen .Template (s .T (), db , database.Template {
4951
- OrganizationID : org .ID ,
4952
- CreatedBy : user .ID ,
4953
- })
4954
- templateVersion := dbgen .TemplateVersion (s .T (), db , database.TemplateVersion {
4955
- TemplateID : uuid.NullUUID {
4956
- UUID : template .ID ,
4957
- Valid : true ,
4958
- },
4959
- OrganizationID : org .ID ,
4960
- CreatedBy : user .ID ,
4961
- })
4962
- preset := dbgen .Preset (s .T (), db , database.InsertPresetParams {
4963
- TemplateVersionID : templateVersion .ID ,
4964
- })
4965
- req := database.UpdatePresetPrebuildStatusParams {
4966
- PresetID : preset .ID ,
4967
- Status : database .PrebuildStatusHealthy ,
4968
- }
4969
- check .Args (req ).
4970
- Asserts (rbac .ResourceTemplate .WithID (template .ID ).InOrg (org .ID ), policy .ActionUpdate )
4971
- }))
4730
+ s .Run ("GetPresetByWorkspaceBuildID" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
4731
+ wbID := uuid .New ()
4732
+ dbm .EXPECT ().GetPresetByWorkspaceBuildID (gomock .Any (), wbID ).Return (testutil .Fake (s .T (), faker , database.TemplateVersionPreset {}), nil ).AnyTimes ()
4733
+ check .Args (wbID ).Asserts (rbac .ResourceTemplate , policy .ActionRead )
4734
+ }))
4735
+ s .Run ("GetPresetParametersByTemplateVersionID" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
4736
+ tpl := testutil .Fake (s .T (), faker , database.Template {})
4737
+ tv := testutil .Fake (s .T (), faker , database.TemplateVersion {TemplateID : uuid.NullUUID {UUID : tpl .ID , Valid : true }, OrganizationID : tpl .OrganizationID , CreatedBy : tpl .CreatedBy })
4738
+ params := []database.TemplateVersionPresetParameter {testutil .Fake (s .T (), faker , database.TemplateVersionPresetParameter {})}
4739
+ dbm .EXPECT ().GetTemplateVersionByID (gomock .Any (), tv .ID ).Return (tv , nil ).AnyTimes ()
4740
+ dbm .EXPECT ().GetTemplateByID (gomock .Any (), tpl .ID ).Return (tpl , nil ).AnyTimes ()
4741
+ dbm .EXPECT ().GetPresetParametersByTemplateVersionID (gomock .Any (), tv .ID ).Return (params , nil ).AnyTimes ()
4742
+ check .Args (tv .ID ).Asserts (tpl .RBACObject (), policy .ActionRead ).Returns (params )
4743
+ }))
4744
+ s .Run ("GetPresetParametersByPresetID" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
4745
+ tpl := testutil .Fake (s .T (), faker , database.Template {})
4746
+ presetID := uuid .New ()
4747
+ prow := database.GetPresetByIDRow {ID : presetID , TemplateID : uuid.NullUUID {UUID : tpl .ID , Valid : true }, OrganizationID : tpl .OrganizationID }
4748
+ params := []database.TemplateVersionPresetParameter {testutil .Fake (s .T (), faker , database.TemplateVersionPresetParameter {})}
4749
+ dbm .EXPECT ().GetPresetByID (gomock .Any (), presetID ).Return (prow , nil ).AnyTimes ()
4750
+ dbm .EXPECT ().GetTemplateByID (gomock .Any (), tpl .ID ).Return (tpl , nil ).AnyTimes ()
4751
+ dbm .EXPECT ().GetPresetParametersByPresetID (gomock .Any (), presetID ).Return (params , nil ).AnyTimes ()
4752
+ check .Args (presetID ).Asserts (tpl .RBACObject (), policy .ActionRead ).Returns (params )
4753
+ }))
4754
+ s .Run ("GetActivePresetPrebuildSchedules" , s .Mocked (func (dbm * dbmock.MockStore , _ * gofakeit.Faker , check * expects ) {
4755
+ dbm .EXPECT ().GetActivePresetPrebuildSchedules (gomock .Any ()).Return ([]database.TemplateVersionPresetPrebuildSchedule {}, nil ).AnyTimes ()
4756
+ check .Args ().Asserts (rbac .ResourceTemplate .All (), policy .ActionRead ).Returns ([]database.TemplateVersionPresetPrebuildSchedule {})
4757
+ }))
4758
+ s .Run ("GetPresetsByTemplateVersionID" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
4759
+ tpl := testutil .Fake (s .T (), faker , database.Template {})
4760
+ tv := testutil .Fake (s .T (), faker , database.TemplateVersion {TemplateID : uuid.NullUUID {UUID : tpl .ID , Valid : true }, OrganizationID : tpl .OrganizationID , CreatedBy : tpl .CreatedBy })
4761
+ presets := []database.TemplateVersionPreset {testutil .Fake (s .T (), faker , database.TemplateVersionPreset {TemplateVersionID : tv .ID })}
4762
+ dbm .EXPECT ().GetTemplateVersionByID (gomock .Any (), tv .ID ).Return (tv , nil ).AnyTimes ()
4763
+ dbm .EXPECT ().GetTemplateByID (gomock .Any (), tpl .ID ).Return (tpl , nil ).AnyTimes ()
4764
+ dbm .EXPECT ().GetPresetsByTemplateVersionID (gomock .Any (), tv .ID ).Return (presets , nil ).AnyTimes ()
4765
+ check .Args (tv .ID ).Asserts (tpl .RBACObject (), policy .ActionRead ).Returns (presets )
4766
+ }))
4767
+ s .Run ("ClaimPrebuiltWorkspace" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
4768
+ org := testutil .Fake (s .T (), faker , database.Organization {})
4769
+ user := testutil .Fake (s .T (), faker , database.User {})
4770
+ tpl := testutil .Fake (s .T (), faker , database.Template {OrganizationID : org .ID , CreatedBy : user .ID })
4771
+ arg := database.ClaimPrebuiltWorkspaceParams {NewUserID : user .ID , NewName : "" , PresetID : uuid .New ()}
4772
+ prow := database.GetPresetByIDRow {ID : arg .PresetID , TemplateID : uuid.NullUUID {UUID : tpl .ID , Valid : true }, OrganizationID : org .ID }
4773
+ dbm .EXPECT ().GetPresetByID (gomock .Any (), arg .PresetID ).Return (prow , nil ).AnyTimes ()
4774
+ dbm .EXPECT ().GetTemplateByID (gomock .Any (), tpl .ID ).Return (tpl , nil ).AnyTimes ()
4775
+ dbm .EXPECT ().ClaimPrebuiltWorkspace (gomock .Any (), arg ).Return (database.ClaimPrebuiltWorkspaceRow {}, sql .ErrNoRows ).AnyTimes ()
4776
+ check .Args (arg ).Asserts (
4777
+ rbac .ResourceWorkspace .WithOwner (user .ID .String ()).InOrg (org .ID ), policy .ActionCreate ,
4778
+ tpl , policy .ActionRead ,
4779
+ tpl , policy .ActionUse ,
4780
+ ).Errors (sql .ErrNoRows )
4781
+ }))
4782
+ s .Run ("GetPrebuildMetrics" , s .Mocked (func (dbm * dbmock.MockStore , _ * gofakeit.Faker , check * expects ) {
4783
+ dbm .EXPECT ().GetPrebuildMetrics (gomock .Any ()).Return ([]database.GetPrebuildMetricsRow {}, nil ).AnyTimes ()
4784
+ check .Args ().Asserts (rbac .ResourceWorkspace .All (), policy .ActionRead )
4785
+ }))
4786
+ s .Run ("GetPrebuildsSettings" , s .Mocked (func (dbm * dbmock.MockStore , _ * gofakeit.Faker , check * expects ) {
4787
+ dbm .EXPECT ().GetPrebuildsSettings (gomock .Any ()).Return ("{}" , nil ).AnyTimes ()
4788
+ check .Args ().Asserts ()
4789
+ }))
4790
+ s .Run ("UpsertPrebuildsSettings" , s .Mocked (func (dbm * dbmock.MockStore , _ * gofakeit.Faker , check * expects ) {
4791
+ dbm .EXPECT ().UpsertPrebuildsSettings (gomock .Any (), "foo" ).Return (nil ).AnyTimes ()
4792
+ check .Args ("foo" ).Asserts (rbac .ResourceDeploymentConfig , policy .ActionUpdate )
4793
+ }))
4794
+ s .Run ("CountInProgressPrebuilds" , s .Mocked (func (dbm * dbmock.MockStore , _ * gofakeit.Faker , check * expects ) {
4795
+ dbm .EXPECT ().CountInProgressPrebuilds (gomock .Any ()).Return ([]database.CountInProgressPrebuildsRow {}, nil ).AnyTimes ()
4796
+ check .Args ().Asserts (rbac .ResourceWorkspace .All (), policy .ActionRead )
4797
+ }))
4798
+ s .Run ("GetPresetsAtFailureLimit" , s .Mocked (func (dbm * dbmock.MockStore , _ * gofakeit.Faker , check * expects ) {
4799
+ dbm .EXPECT ().GetPresetsAtFailureLimit (gomock .Any (), int64 (0 )).Return ([]database.GetPresetsAtFailureLimitRow {}, nil ).AnyTimes ()
4800
+ check .Args (int64 (0 )).Asserts (rbac .ResourceTemplate .All (), policy .ActionViewInsights )
4801
+ }))
4802
+ s .Run ("GetPresetsBackoff" , s .Mocked (func (dbm * dbmock.MockStore , _ * gofakeit.Faker , check * expects ) {
4803
+ t0 := time.Time {}
4804
+ dbm .EXPECT ().GetPresetsBackoff (gomock .Any (), t0 ).Return ([]database.GetPresetsBackoffRow {}, nil ).AnyTimes ()
4805
+ check .Args (t0 ).Asserts (rbac .ResourceTemplate .All (), policy .ActionViewInsights )
4806
+ }))
4807
+ s .Run ("GetRunningPrebuiltWorkspaces" , s .Mocked (func (dbm * dbmock.MockStore , _ * gofakeit.Faker , check * expects ) {
4808
+ dbm .EXPECT ().GetRunningPrebuiltWorkspaces (gomock .Any ()).Return ([]database.GetRunningPrebuiltWorkspacesRow {}, nil ).AnyTimes ()
4809
+ check .Args ().Asserts (rbac .ResourceWorkspace .All (), policy .ActionRead )
4810
+ }))
4811
+ s .Run ("GetTemplatePresetsWithPrebuilds" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
4812
+ user := testutil .Fake (s .T (), faker , database.User {})
4813
+ arg := uuid.NullUUID {UUID : user .ID , Valid : true }
4814
+ dbm .EXPECT ().GetTemplatePresetsWithPrebuilds (gomock .Any (), arg ).Return ([]database.GetTemplatePresetsWithPrebuildsRow {}, nil ).AnyTimes ()
4815
+ check .Args (arg ).Asserts (rbac .ResourceTemplate .All (), policy .ActionRead )
4816
+ }))
4817
+ s .Run ("GetPresetByID" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
4818
+ org := testutil .Fake (s .T (), faker , database.Organization {})
4819
+ tpl := testutil .Fake (s .T (), faker , database.Template {OrganizationID : org .ID })
4820
+ presetID := uuid .New ()
4821
+ prow := database.GetPresetByIDRow {ID : presetID , TemplateVersionID : uuid .New (), Name : "test" , TemplateID : uuid.NullUUID {UUID : tpl .ID , Valid : true }, InvalidateAfterSecs : sql.NullInt32 {}, OrganizationID : org .ID , PrebuildStatus : database .PrebuildStatusHealthy }
4822
+ dbm .EXPECT ().GetPresetByID (gomock .Any (), presetID ).Return (prow , nil ).AnyTimes ()
4823
+ dbm .EXPECT ().GetTemplateByID (gomock .Any (), tpl .ID ).Return (tpl , nil ).AnyTimes ()
4824
+ check .Args (presetID ).Asserts (tpl , policy .ActionRead ).Returns (prow )
4825
+ }))
4826
+ s .Run ("UpdatePresetPrebuildStatus" , s .Mocked (func (dbm * dbmock.MockStore , faker * gofakeit.Faker , check * expects ) {
4827
+ org := testutil .Fake (s .T (), faker , database.Organization {})
4828
+ tpl := testutil .Fake (s .T (), faker , database.Template {OrganizationID : org .ID })
4829
+ presetID := uuid .New ()
4830
+ prow := database.GetPresetByIDRow {ID : presetID , TemplateID : uuid.NullUUID {UUID : tpl .ID , Valid : true }, OrganizationID : org .ID }
4831
+ req := database.UpdatePresetPrebuildStatusParams {PresetID : presetID , Status : database .PrebuildStatusHealthy }
4832
+ dbm .EXPECT ().GetPresetByID (gomock .Any (), presetID ).Return (prow , nil ).AnyTimes ()
4833
+ dbm .EXPECT ().UpdatePresetPrebuildStatus (gomock .Any (), req ).Return (nil ).AnyTimes ()
4834
+ check .Args (req ).Asserts (rbac .ResourceTemplate .WithID (tpl .ID ).InOrg (org .ID ), policy .ActionUpdate )
4835
+ }))
4972
4836
}
4973
4837
4974
4838
func (s * MethodTestSuite ) TestOAuth2ProviderApps () {
0 commit comments