Skip to content

Commit bf65b40

Browse files
hugodutkahugodutka
committed
format
1 parent 8a24095 commit bf65b40

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
@@ -4465,15 +4465,15 @@ func (s *MethodTestSuite) TestSystemFunctions() {
44654465
check.Args(arg).Asserts(rbac.ResourceSystem, policy.ActionCreate)
44664466
}))
44674467
s.Run("GetProvisionerJobTimingsByJobID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4468-
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeWorkspaceBuild})
4469-
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{JobID: j.ID})
4470-
ws := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4471-
dbm.EXPECT().GetProvisionerJobByID(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4472-
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4473-
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(ws, nil).AnyTimes()
4474-
dbm.EXPECT().GetProvisionerJobTimingsByJobID(gomock.Any(), j.ID).Return([]database.ProvisionerJobTiming{}, nil).AnyTimes()
4475-
check.Args(j.ID).Asserts(ws, policy.ActionRead)
4476-
}))
4468+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeWorkspaceBuild})
4469+
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{JobID: j.ID})
4470+
ws := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4471+
dbm.EXPECT().GetProvisionerJobByID(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4472+
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4473+
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(ws, nil).AnyTimes()
4474+
dbm.EXPECT().GetProvisionerJobTimingsByJobID(gomock.Any(), j.ID).Return([]database.ProvisionerJobTiming{}, nil).AnyTimes()
4475+
check.Args(j.ID).Asserts(ws, policy.ActionRead)
4476+
}))
44774477
s.Run("GetWorkspaceAgentScriptTimingsByBuildID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
44784478
build := testutil.Fake(s.T(), faker, database.WorkspaceBuild{})
44794479
dbm.EXPECT().GetWorkspaceAgentScriptTimingsByBuildID(gomock.Any(), build.ID).Return([]database.GetWorkspaceAgentScriptTimingsByBuildIDRow{}, nil).AnyTimes()
@@ -4539,31 +4539,33 @@ func (s *MethodTestSuite) TestSystemFunctions() {
45394539
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
45404540
// Minimal assertion check argument
45414541
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{JobID: j.ID})
4542-
w := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4543-
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4544-
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(w, nil).AnyTimes()
4545-
check.Args(j.ID).Asserts(w, policy.ActionRead).Returns(j)
4542+
w := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4543+
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4544+
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(w, nil).AnyTimes()
4545+
check.Args(j.ID).Asserts(w, policy.ActionRead).Returns(j)
45464546
}))
45474547
s.Run("TemplateVersion/GetProvisionerJobByIDForUpdate", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4548-
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeTemplateVersionImport})
4549-
tpl := testutil.Fake(s.T(), faker, database.Template{})
4550-
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4551-
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4552-
dbm.EXPECT().GetTemplateVersionByJobID(gomock.Any(), j.ID).Return(tv, nil).AnyTimes()
4553-
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4554-
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4555-
}))
4556-
s.Run("TemplateVersionDryRun/GetProvisionerJobByIDForUpdate", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4557-
tpl := testutil.Fake(s.T(), faker, database.Template{})
4558-
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4559-
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{})
4560-
j.Type = database.ProvisionerJobTypeTemplateVersionDryRun
4561-
j.Input = must(json.Marshal(struct { TemplateVersionID uuid.UUID `json:"template_version_id"` }{TemplateVersionID: tv.ID}))
4562-
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4563-
dbm.EXPECT().GetTemplateVersionByID(gomock.Any(), tv.ID).Return(tv, nil).AnyTimes()
4564-
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4565-
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4566-
}))
4548+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeTemplateVersionImport})
4549+
tpl := testutil.Fake(s.T(), faker, database.Template{})
4550+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4551+
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4552+
dbm.EXPECT().GetTemplateVersionByJobID(gomock.Any(), j.ID).Return(tv, nil).AnyTimes()
4553+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4554+
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4555+
}))
4556+
s.Run("TemplateVersionDryRun/GetProvisionerJobByIDForUpdate", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4557+
tpl := testutil.Fake(s.T(), faker, database.Template{})
4558+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4559+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{})
4560+
j.Type = database.ProvisionerJobTypeTemplateVersionDryRun
4561+
j.Input = must(json.Marshal(struct {
4562+
TemplateVersionID uuid.UUID `json:"template_version_id"`
4563+
}{TemplateVersionID: tv.ID}))
4564+
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4565+
dbm.EXPECT().GetTemplateVersionByID(gomock.Any(), tv.ID).Return(tv, nil).AnyTimes()
4566+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4567+
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4568+
}))
45674569
}
45684570

45694571
func (s *MethodTestSuite) TestNotifications() {

0 commit comments

Comments
 (0)