Skip to content

Commit 69bf6bd

Browse files
hugodutkahugodutka
committed
coderd: dbauthz_test: migrate TestPrebuilds suite to use mock DB
- Replace Subtest with Mocked and add gomock expectations - Mock underlying calls used by querier (e.g., GetTemplateVersionByID, GetPresetByID, GetTemplateByID) - Preserve RBAC assertions and expected returns/errors - Verified by running TestMethodTestSuite/TestPrebuilds locally with GOCACHE
1 parent 8192e6b commit 69bf6bd

File tree

1 file changed

+106
-242
lines changed

1 file changed

+106
-242
lines changed

‎coderd/database/dbauthz/dbauthz_test.go

Lines changed: 106 additions & 242 deletions
Original file line numberDiff line numberDiff line change
@@ -4651,248 +4651,112 @@ func (s *MethodTestSuite) TestNotifications() {
46514651
}
46524652

46534653
func (s *MethodTestSuite) TestPrebuilds() {
4654-
s.Run("GetPresetByWorkspaceBuildID", s.Subtest(func(db database.Store, check *expects) {
4655-
org := dbgen.Organization(s.T(), db, database.Organization{})
4656-
user := dbgen.User(s.T(), db, database.User{})
4657-
template := dbgen.Template(s.T(), db, database.Template{
4658-
CreatedBy: user.ID,
4659-
OrganizationID: org.ID,
4660-
})
4661-
templateVersion := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
4662-
TemplateID: uuid.NullUUID{UUID: template.ID, Valid: true},
4663-
OrganizationID: org.ID,
4664-
CreatedBy: user.ID,
4665-
})
4666-
preset, err := db.InsertPreset(context.Background(), database.InsertPresetParams{
4667-
TemplateVersionID: templateVersion.ID,
4668-
Name: "test",
4669-
})
4670-
require.NoError(s.T(), err)
4671-
workspace := dbgen.Workspace(s.T(), db, database.WorkspaceTable{
4672-
OrganizationID: org.ID,
4673-
OwnerID: user.ID,
4674-
TemplateID: template.ID,
4675-
})
4676-
job := dbgen.ProvisionerJob(s.T(), db, nil, database.ProvisionerJob{
4677-
OrganizationID: org.ID,
4678-
})
4679-
workspaceBuild := dbgen.WorkspaceBuild(s.T(), db, database.WorkspaceBuild{
4680-
WorkspaceID: workspace.ID,
4681-
TemplateVersionID: templateVersion.ID,
4682-
TemplateVersionPresetID: uuid.NullUUID{UUID: preset.ID, Valid: true},
4683-
InitiatorID: user.ID,
4684-
JobID: job.ID,
4685-
})
4686-
_, err = db.GetPresetByWorkspaceBuildID(context.Background(), workspaceBuild.ID)
4687-
require.NoError(s.T(), err)
4688-
check.Args(workspaceBuild.ID).Asserts(rbac.ResourceTemplate, policy.ActionRead)
4689-
}))
4690-
s.Run("GetPresetParametersByTemplateVersionID", s.Subtest(func(db database.Store, check *expects) {
4691-
ctx := context.Background()
4692-
org := dbgen.Organization(s.T(), db, database.Organization{})
4693-
user := dbgen.User(s.T(), db, database.User{})
4694-
template := dbgen.Template(s.T(), db, database.Template{
4695-
CreatedBy: user.ID,
4696-
OrganizationID: org.ID,
4697-
})
4698-
templateVersion := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
4699-
TemplateID: uuid.NullUUID{UUID: template.ID, Valid: true},
4700-
OrganizationID: org.ID,
4701-
CreatedBy: user.ID,
4702-
})
4703-
preset, err := db.InsertPreset(ctx, database.InsertPresetParams{
4704-
TemplateVersionID: templateVersion.ID,
4705-
Name: "test",
4706-
})
4707-
require.NoError(s.T(), err)
4708-
insertedParameters, err := db.InsertPresetParameters(ctx, database.InsertPresetParametersParams{
4709-
TemplateVersionPresetID: preset.ID,
4710-
Names: []string{"test"},
4711-
Values: []string{"test"},
4712-
})
4713-
require.NoError(s.T(), err)
4714-
check.
4715-
Args(templateVersion.ID).
4716-
Asserts(template.RBACObject(), policy.ActionRead).
4717-
Returns(insertedParameters)
4718-
}))
4719-
s.Run("GetPresetParametersByPresetID", s.Subtest(func(db database.Store, check *expects) {
4720-
ctx := context.Background()
4721-
org := dbgen.Organization(s.T(), db, database.Organization{})
4722-
user := dbgen.User(s.T(), db, database.User{})
4723-
template := dbgen.Template(s.T(), db, database.Template{
4724-
CreatedBy: user.ID,
4725-
OrganizationID: org.ID,
4726-
})
4727-
templateVersion := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
4728-
TemplateID: uuid.NullUUID{UUID: template.ID, Valid: true},
4729-
OrganizationID: org.ID,
4730-
CreatedBy: user.ID,
4731-
})
4732-
preset, err := db.InsertPreset(ctx, database.InsertPresetParams{
4733-
TemplateVersionID: templateVersion.ID,
4734-
Name: "test",
4735-
})
4736-
require.NoError(s.T(), err)
4737-
insertedParameters, err := db.InsertPresetParameters(ctx, database.InsertPresetParametersParams{
4738-
TemplateVersionPresetID: preset.ID,
4739-
Names: []string{"test"},
4740-
Values: []string{"test"},
4741-
})
4742-
require.NoError(s.T(), err)
4743-
check.
4744-
Args(preset.ID).
4745-
Asserts(template.RBACObject(), policy.ActionRead).
4746-
Returns(insertedParameters)
4747-
}))
4748-
s.Run("GetActivePresetPrebuildSchedules", s.Subtest(func(db database.Store, check *expects) {
4749-
check.Args().
4750-
Asserts(rbac.ResourceTemplate.All(), policy.ActionRead).
4751-
Returns([]database.TemplateVersionPresetPrebuildSchedule{})
4752-
}))
4753-
s.Run("GetPresetsByTemplateVersionID", s.Subtest(func(db database.Store, check *expects) {
4754-
ctx := context.Background()
4755-
org := dbgen.Organization(s.T(), db, database.Organization{})
4756-
user := dbgen.User(s.T(), db, database.User{})
4757-
template := dbgen.Template(s.T(), db, database.Template{
4758-
CreatedBy: user.ID,
4759-
OrganizationID: org.ID,
4760-
})
4761-
templateVersion := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
4762-
TemplateID: uuid.NullUUID{UUID: template.ID, Valid: true},
4763-
OrganizationID: org.ID,
4764-
CreatedBy: user.ID,
4765-
})
4766-
4767-
_, err := db.InsertPreset(ctx, database.InsertPresetParams{
4768-
TemplateVersionID: templateVersion.ID,
4769-
Name: "test",
4770-
})
4771-
require.NoError(s.T(), err)
4772-
4773-
presets, err := db.GetPresetsByTemplateVersionID(ctx, templateVersion.ID)
4774-
require.NoError(s.T(), err)
4775-
4776-
check.Args(templateVersion.ID).Asserts(template.RBACObject(), policy.ActionRead).Returns(presets)
4777-
}))
4778-
s.Run("ClaimPrebuiltWorkspace", s.Subtest(func(db database.Store, check *expects) {
4779-
org := dbgen.Organization(s.T(), db, database.Organization{})
4780-
user := dbgen.User(s.T(), db, database.User{})
4781-
template := dbgen.Template(s.T(), db, database.Template{
4782-
OrganizationID: org.ID,
4783-
CreatedBy: user.ID,
4784-
})
4785-
templateVersion := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
4786-
TemplateID: uuid.NullUUID{
4787-
UUID: template.ID,
4788-
Valid: true,
4789-
},
4790-
OrganizationID: org.ID,
4791-
CreatedBy: user.ID,
4792-
})
4793-
preset := dbgen.Preset(s.T(), db, database.InsertPresetParams{
4794-
TemplateVersionID: templateVersion.ID,
4795-
})
4796-
check.Args(database.ClaimPrebuiltWorkspaceParams{
4797-
NewUserID: user.ID,
4798-
NewName: "",
4799-
PresetID: preset.ID,
4800-
}).Asserts(
4801-
rbac.ResourceWorkspace.WithOwner(user.ID.String()).InOrg(org.ID), policy.ActionCreate,
4802-
template, policy.ActionRead,
4803-
template, policy.ActionUse,
4804-
).Errors(sql.ErrNoRows)
4805-
}))
4806-
s.Run("GetPrebuildMetrics", s.Subtest(func(_ database.Store, check *expects) {
4807-
check.Args().
4808-
Asserts(rbac.ResourceWorkspace.All(), policy.ActionRead)
4809-
}))
4810-
s.Run("GetPrebuildsSettings", s.Subtest(func(db database.Store, check *expects) {
4811-
check.Args().Asserts()
4812-
}))
4813-
s.Run("UpsertPrebuildsSettings", s.Subtest(func(db database.Store, check *expects) {
4814-
check.Args("foo").Asserts(rbac.ResourceDeploymentConfig, policy.ActionUpdate)
4815-
}))
4816-
s.Run("CountInProgressPrebuilds", s.Subtest(func(_ database.Store, check *expects) {
4817-
check.Args().
4818-
Asserts(rbac.ResourceWorkspace.All(), policy.ActionRead)
4819-
}))
4820-
s.Run("GetPresetsAtFailureLimit", s.Subtest(func(_ database.Store, check *expects) {
4821-
check.Args(int64(0)).
4822-
Asserts(rbac.ResourceTemplate.All(), policy.ActionViewInsights)
4823-
}))
4824-
s.Run("GetPresetsBackoff", s.Subtest(func(_ database.Store, check *expects) {
4825-
check.Args(time.Time{}).
4826-
Asserts(rbac.ResourceTemplate.All(), policy.ActionViewInsights)
4827-
}))
4828-
s.Run("GetRunningPrebuiltWorkspaces", s.Subtest(func(_ database.Store, check *expects) {
4829-
check.Args().
4830-
Asserts(rbac.ResourceWorkspace.All(), policy.ActionRead)
4831-
}))
4832-
s.Run("GetTemplatePresetsWithPrebuilds", s.Subtest(func(db database.Store, check *expects) {
4833-
user := dbgen.User(s.T(), db, database.User{})
4834-
check.Args(uuid.NullUUID{UUID: user.ID, Valid: true}).
4835-
Asserts(rbac.ResourceTemplate.All(), policy.ActionRead)
4836-
}))
4837-
s.Run("GetPresetByID", s.Subtest(func(db database.Store, check *expects) {
4838-
org := dbgen.Organization(s.T(), db, database.Organization{})
4839-
user := dbgen.User(s.T(), db, database.User{})
4840-
template := dbgen.Template(s.T(), db, database.Template{
4841-
OrganizationID: org.ID,
4842-
CreatedBy: user.ID,
4843-
})
4844-
templateVersion := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
4845-
TemplateID: uuid.NullUUID{
4846-
UUID: template.ID,
4847-
Valid: true,
4848-
},
4849-
OrganizationID: org.ID,
4850-
CreatedBy: user.ID,
4851-
})
4852-
preset := dbgen.Preset(s.T(), db, database.InsertPresetParams{
4853-
TemplateVersionID: templateVersion.ID,
4854-
})
4855-
check.Args(preset.ID).
4856-
Asserts(template, policy.ActionRead).
4857-
Returns(database.GetPresetByIDRow{
4858-
ID: preset.ID,
4859-
TemplateVersionID: preset.TemplateVersionID,
4860-
Name: preset.Name,
4861-
CreatedAt: preset.CreatedAt,
4862-
TemplateID: uuid.NullUUID{
4863-
UUID: template.ID,
4864-
Valid: true,
4865-
},
4866-
InvalidateAfterSecs: preset.InvalidateAfterSecs,
4867-
OrganizationID: org.ID,
4868-
PrebuildStatus: database.PrebuildStatusHealthy,
4869-
})
4870-
}))
4871-
s.Run("UpdatePresetPrebuildStatus", s.Subtest(func(db database.Store, check *expects) {
4872-
org := dbgen.Organization(s.T(), db, database.Organization{})
4873-
user := dbgen.User(s.T(), db, database.User{})
4874-
template := dbgen.Template(s.T(), db, database.Template{
4875-
OrganizationID: org.ID,
4876-
CreatedBy: user.ID,
4877-
})
4878-
templateVersion := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
4879-
TemplateID: uuid.NullUUID{
4880-
UUID: template.ID,
4881-
Valid: true,
4882-
},
4883-
OrganizationID: org.ID,
4884-
CreatedBy: user.ID,
4885-
})
4886-
preset := dbgen.Preset(s.T(), db, database.InsertPresetParams{
4887-
TemplateVersionID: templateVersion.ID,
4888-
})
4889-
req := database.UpdatePresetPrebuildStatusParams{
4890-
PresetID: preset.ID,
4891-
Status: database.PrebuildStatusHealthy,
4892-
}
4893-
check.Args(req).
4894-
Asserts(rbac.ResourceTemplate.WithID(template.ID).InOrg(org.ID), policy.ActionUpdate)
4895-
}))
4654+
s.Run("GetPresetByWorkspaceBuildID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4655+
wbID := uuid.New()
4656+
dbm.EXPECT().GetPresetByWorkspaceBuildID(gomock.Any(), wbID).Return(testutil.Fake(s.T(), faker, database.TemplateVersionPreset{}), nil).AnyTimes()
4657+
check.Args(wbID).Asserts(rbac.ResourceTemplate, policy.ActionRead)
4658+
}))
4659+
s.Run("GetPresetParametersByTemplateVersionID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4660+
tpl := testutil.Fake(s.T(), faker, database.Template{})
4661+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}, OrganizationID: tpl.OrganizationID, CreatedBy: tpl.CreatedBy})
4662+
params := []database.TemplateVersionPresetParameter{testutil.Fake(s.T(), faker, database.TemplateVersionPresetParameter{})}
4663+
dbm.EXPECT().GetTemplateVersionByID(gomock.Any(), tv.ID).Return(tv, nil).AnyTimes()
4664+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4665+
dbm.EXPECT().GetPresetParametersByTemplateVersionID(gomock.Any(), tv.ID).Return(params, nil).AnyTimes()
4666+
check.Args(tv.ID).Asserts(tpl.RBACObject(), policy.ActionRead).Returns(params)
4667+
}))
4668+
s.Run("GetPresetParametersByPresetID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4669+
tpl := testutil.Fake(s.T(), faker, database.Template{})
4670+
presetID := uuid.New()
4671+
prow := database.GetPresetByIDRow{ID: presetID, TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}, OrganizationID: tpl.OrganizationID}
4672+
params := []database.TemplateVersionPresetParameter{testutil.Fake(s.T(), faker, database.TemplateVersionPresetParameter{})}
4673+
dbm.EXPECT().GetPresetByID(gomock.Any(), presetID).Return(prow, nil).AnyTimes()
4674+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4675+
dbm.EXPECT().GetPresetParametersByPresetID(gomock.Any(), presetID).Return(params, nil).AnyTimes()
4676+
check.Args(presetID).Asserts(tpl.RBACObject(), policy.ActionRead).Returns(params)
4677+
}))
4678+
s.Run("GetActivePresetPrebuildSchedules", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) {
4679+
dbm.EXPECT().GetActivePresetPrebuildSchedules(gomock.Any()).Return([]database.TemplateVersionPresetPrebuildSchedule{}, nil).AnyTimes()
4680+
check.Args().Asserts(rbac.ResourceTemplate.All(), policy.ActionRead).Returns([]database.TemplateVersionPresetPrebuildSchedule{})
4681+
}))
4682+
s.Run("GetPresetsByTemplateVersionID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4683+
tpl := testutil.Fake(s.T(), faker, database.Template{})
4684+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}, OrganizationID: tpl.OrganizationID, CreatedBy: tpl.CreatedBy})
4685+
presets := []database.TemplateVersionPreset{testutil.Fake(s.T(), faker, database.TemplateVersionPreset{TemplateVersionID: tv.ID})}
4686+
dbm.EXPECT().GetTemplateVersionByID(gomock.Any(), tv.ID).Return(tv, nil).AnyTimes()
4687+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4688+
dbm.EXPECT().GetPresetsByTemplateVersionID(gomock.Any(), tv.ID).Return(presets, nil).AnyTimes()
4689+
check.Args(tv.ID).Asserts(tpl.RBACObject(), policy.ActionRead).Returns(presets)
4690+
}))
4691+
s.Run("ClaimPrebuiltWorkspace", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4692+
org := testutil.Fake(s.T(), faker, database.Organization{})
4693+
user := testutil.Fake(s.T(), faker, database.User{})
4694+
tpl := testutil.Fake(s.T(), faker, database.Template{OrganizationID: org.ID, CreatedBy: user.ID})
4695+
arg := database.ClaimPrebuiltWorkspaceParams{NewUserID: user.ID, NewName: "", PresetID: uuid.New()}
4696+
prow := database.GetPresetByIDRow{ID: arg.PresetID, TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}, OrganizationID: org.ID}
4697+
dbm.EXPECT().GetPresetByID(gomock.Any(), arg.PresetID).Return(prow, nil).AnyTimes()
4698+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4699+
dbm.EXPECT().ClaimPrebuiltWorkspace(gomock.Any(), arg).Return(database.ClaimPrebuiltWorkspaceRow{}, sql.ErrNoRows).AnyTimes()
4700+
check.Args(arg).Asserts(
4701+
rbac.ResourceWorkspace.WithOwner(user.ID.String()).InOrg(org.ID), policy.ActionCreate,
4702+
tpl, policy.ActionRead,
4703+
tpl, policy.ActionUse,
4704+
).Errors(sql.ErrNoRows)
4705+
}))
4706+
s.Run("GetPrebuildMetrics", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) {
4707+
dbm.EXPECT().GetPrebuildMetrics(gomock.Any()).Return([]database.GetPrebuildMetricsRow{}, nil).AnyTimes()
4708+
check.Args().Asserts(rbac.ResourceWorkspace.All(), policy.ActionRead)
4709+
}))
4710+
s.Run("GetPrebuildsSettings", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) {
4711+
dbm.EXPECT().GetPrebuildsSettings(gomock.Any()).Return("{}", nil).AnyTimes()
4712+
check.Args().Asserts()
4713+
}))
4714+
s.Run("UpsertPrebuildsSettings", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) {
4715+
dbm.EXPECT().UpsertPrebuildsSettings(gomock.Any(), "foo").Return(nil).AnyTimes()
4716+
check.Args("foo").Asserts(rbac.ResourceDeploymentConfig, policy.ActionUpdate)
4717+
}))
4718+
s.Run("CountInProgressPrebuilds", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) {
4719+
dbm.EXPECT().CountInProgressPrebuilds(gomock.Any()).Return([]database.CountInProgressPrebuildsRow{}, nil).AnyTimes()
4720+
check.Args().Asserts(rbac.ResourceWorkspace.All(), policy.ActionRead)
4721+
}))
4722+
s.Run("GetPresetsAtFailureLimit", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) {
4723+
dbm.EXPECT().GetPresetsAtFailureLimit(gomock.Any(), int64(0)).Return([]database.GetPresetsAtFailureLimitRow{}, nil).AnyTimes()
4724+
check.Args(int64(0)).Asserts(rbac.ResourceTemplate.All(), policy.ActionViewInsights)
4725+
}))
4726+
s.Run("GetPresetsBackoff", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) {
4727+
t0 := time.Time{}
4728+
dbm.EXPECT().GetPresetsBackoff(gomock.Any(), t0).Return([]database.GetPresetsBackoffRow{}, nil).AnyTimes()
4729+
check.Args(t0).Asserts(rbac.ResourceTemplate.All(), policy.ActionViewInsights)
4730+
}))
4731+
s.Run("GetRunningPrebuiltWorkspaces", s.Mocked(func(dbm *dbmock.MockStore, _ *gofakeit.Faker, check *expects) {
4732+
dbm.EXPECT().GetRunningPrebuiltWorkspaces(gomock.Any()).Return([]database.GetRunningPrebuiltWorkspacesRow{}, nil).AnyTimes()
4733+
check.Args().Asserts(rbac.ResourceWorkspace.All(), policy.ActionRead)
4734+
}))
4735+
s.Run("GetTemplatePresetsWithPrebuilds", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4736+
user := testutil.Fake(s.T(), faker, database.User{})
4737+
arg := uuid.NullUUID{UUID: user.ID, Valid: true}
4738+
dbm.EXPECT().GetTemplatePresetsWithPrebuilds(gomock.Any(), arg).Return([]database.GetTemplatePresetsWithPrebuildsRow{}, nil).AnyTimes()
4739+
check.Args(arg).Asserts(rbac.ResourceTemplate.All(), policy.ActionRead)
4740+
}))
4741+
s.Run("GetPresetByID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4742+
org := testutil.Fake(s.T(), faker, database.Organization{})
4743+
tpl := testutil.Fake(s.T(), faker, database.Template{OrganizationID: org.ID})
4744+
presetID := uuid.New()
4745+
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}
4746+
dbm.EXPECT().GetPresetByID(gomock.Any(), presetID).Return(prow, nil).AnyTimes()
4747+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4748+
check.Args(presetID).Asserts(tpl, policy.ActionRead).Returns(prow)
4749+
}))
4750+
s.Run("UpdatePresetPrebuildStatus", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4751+
org := testutil.Fake(s.T(), faker, database.Organization{})
4752+
tpl := testutil.Fake(s.T(), faker, database.Template{OrganizationID: org.ID})
4753+
presetID := uuid.New()
4754+
prow := database.GetPresetByIDRow{ID: presetID, TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}, OrganizationID: org.ID}
4755+
req := database.UpdatePresetPrebuildStatusParams{PresetID: presetID, Status: database.PrebuildStatusHealthy}
4756+
dbm.EXPECT().GetPresetByID(gomock.Any(), presetID).Return(prow, nil).AnyTimes()
4757+
dbm.EXPECT().UpdatePresetPrebuildStatus(gomock.Any(), req).Return(nil).AnyTimes()
4758+
check.Args(req).Asserts(rbac.ResourceTemplate.WithID(tpl.ID).InOrg(org.ID), policy.ActionUpdate)
4759+
}))
48964760
}
48974761

48984762
func (s *MethodTestSuite) TestOAuth2ProviderApps() {

0 commit comments

Comments
 (0)