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)
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy