Skip to content

Commit 3667c52

Browse files
hugodutkahugodutka
committed
format
1 parent 48a23e6 commit 3667c52

File tree

1 file changed

+34
-32
lines changed

1 file changed

+34
-32
lines changed

‎coderd/database/dbauthz/dbauthz_test.go

Lines changed: 34 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -4468,15 +4468,15 @@ func (s *MethodTestSuite) TestSystemFunctions() {
44684468
check.Args(arg).Asserts(rbac.ResourceSystem, policy.ActionCreate)
44694469
}))
44704470
s.Run("GetProvisionerJobTimingsByJobID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4471-
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeWorkspaceBuild})
4472-
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{JobID: j.ID})
4473-
ws := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4474-
dbm.EXPECT().GetProvisionerJobByID(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4475-
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4476-
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(ws, nil).AnyTimes()
4477-
dbm.EXPECT().GetProvisionerJobTimingsByJobID(gomock.Any(), j.ID).Return([]database.ProvisionerJobTiming{}, nil).AnyTimes()
4478-
check.Args(j.ID).Asserts(ws, policy.ActionRead)
4479-
}))
4471+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeWorkspaceBuild})
4472+
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{JobID: j.ID})
4473+
ws := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4474+
dbm.EXPECT().GetProvisionerJobByID(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4475+
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4476+
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(ws, nil).AnyTimes()
4477+
dbm.EXPECT().GetProvisionerJobTimingsByJobID(gomock.Any(), j.ID).Return([]database.ProvisionerJobTiming{}, nil).AnyTimes()
4478+
check.Args(j.ID).Asserts(ws, policy.ActionRead)
4479+
}))
44804480
s.Run("GetWorkspaceAgentScriptTimingsByBuildID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
44814481
build := testutil.Fake(s.T(), faker, database.WorkspaceBuild{})
44824482
dbm.EXPECT().GetWorkspaceAgentScriptTimingsByBuildID(gomock.Any(), build.ID).Return([]database.GetWorkspaceAgentScriptTimingsByBuildIDRow{}, nil).AnyTimes()
@@ -4542,31 +4542,33 @@ func (s *MethodTestSuite) TestSystemFunctions() {
45424542
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
45434543
// Minimal assertion check argument
45444544
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{JobID: j.ID})
4545-
w := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4546-
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4547-
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(w, nil).AnyTimes()
4548-
check.Args(j.ID).Asserts(w, policy.ActionRead).Returns(j)
4545+
w := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4546+
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4547+
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(w, nil).AnyTimes()
4548+
check.Args(j.ID).Asserts(w, policy.ActionRead).Returns(j)
45494549
}))
45504550
s.Run("TemplateVersion/GetProvisionerJobByIDForUpdate", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4551-
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeTemplateVersionImport})
4552-
tpl := testutil.Fake(s.T(), faker, database.Template{})
4553-
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4554-
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4555-
dbm.EXPECT().GetTemplateVersionByJobID(gomock.Any(), j.ID).Return(tv, nil).AnyTimes()
4556-
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4557-
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4558-
}))
4559-
s.Run("TemplateVersionDryRun/GetProvisionerJobByIDForUpdate", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4560-
tpl := testutil.Fake(s.T(), faker, database.Template{})
4561-
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4562-
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{})
4563-
j.Type = database.ProvisionerJobTypeTemplateVersionDryRun
4564-
j.Input = must(json.Marshal(struct { TemplateVersionID uuid.UUID `json:"template_version_id"` }{TemplateVersionID: tv.ID}))
4565-
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4566-
dbm.EXPECT().GetTemplateVersionByID(gomock.Any(), tv.ID).Return(tv, nil).AnyTimes()
4567-
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4568-
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4569-
}))
4551+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeTemplateVersionImport})
4552+
tpl := testutil.Fake(s.T(), faker, database.Template{})
4553+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4554+
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4555+
dbm.EXPECT().GetTemplateVersionByJobID(gomock.Any(), j.ID).Return(tv, nil).AnyTimes()
4556+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4557+
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4558+
}))
4559+
s.Run("TemplateVersionDryRun/GetProvisionerJobByIDForUpdate", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4560+
tpl := testutil.Fake(s.T(), faker, database.Template{})
4561+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4562+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{})
4563+
j.Type = database.ProvisionerJobTypeTemplateVersionDryRun
4564+
j.Input = must(json.Marshal(struct {
4565+
TemplateVersionID uuid.UUID `json:"template_version_id"`
4566+
}{TemplateVersionID: tv.ID}))
4567+
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4568+
dbm.EXPECT().GetTemplateVersionByID(gomock.Any(), tv.ID).Return(tv, nil).AnyTimes()
4569+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4570+
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4571+
}))
45704572
}
45714573

45724574
func (s *MethodTestSuite) TestNotifications() {

0 commit comments

Comments
 (0)