Skip to content

Commit 2628005

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 b190c3d commit 2628005

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
@@ -4724,248 +4724,112 @@ func (s *MethodTestSuite) TestNotifications() {
47244724
}
47254725

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

49714835
func (s *MethodTestSuite) TestOAuth2ProviderApps() {

0 commit comments

Comments
 (0)