Skip to content

Commit 3dd08e2

Browse files
hugodutkahugodutka
committed
format
1 parent a2db47d commit 3dd08e2

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
@@ -4308,15 +4308,15 @@ func (s *MethodTestSuite) TestSystemFunctions() {
43084308
check.Args(arg).Asserts(rbac.ResourceSystem, policy.ActionCreate)
43094309
}))
43104310
s.Run("GetProvisionerJobTimingsByJobID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4311-
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeWorkspaceBuild})
4312-
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{JobID: j.ID})
4313-
ws := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4314-
dbm.EXPECT().GetProvisionerJobByID(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4315-
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4316-
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(ws, nil).AnyTimes()
4317-
dbm.EXPECT().GetProvisionerJobTimingsByJobID(gomock.Any(), j.ID).Return([]database.ProvisionerJobTiming{}, nil).AnyTimes()
4318-
check.Args(j.ID).Asserts(ws, policy.ActionRead)
4319-
}))
4311+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeWorkspaceBuild})
4312+
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{JobID: j.ID})
4313+
ws := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4314+
dbm.EXPECT().GetProvisionerJobByID(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4315+
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4316+
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(ws, nil).AnyTimes()
4317+
dbm.EXPECT().GetProvisionerJobTimingsByJobID(gomock.Any(), j.ID).Return([]database.ProvisionerJobTiming{}, nil).AnyTimes()
4318+
check.Args(j.ID).Asserts(ws, policy.ActionRead)
4319+
}))
43204320
s.Run("GetWorkspaceAgentScriptTimingsByBuildID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
43214321
build := testutil.Fake(s.T(), faker, database.WorkspaceBuild{})
43224322
dbm.EXPECT().GetWorkspaceAgentScriptTimingsByBuildID(gomock.Any(), build.ID).Return([]database.GetWorkspaceAgentScriptTimingsByBuildIDRow{}, nil).AnyTimes()
@@ -4382,31 +4382,33 @@ func (s *MethodTestSuite) TestSystemFunctions() {
43824382
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
43834383
// Minimal assertion check argument
43844384
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{JobID: j.ID})
4385-
w := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4386-
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4387-
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(w, nil).AnyTimes()
4388-
check.Args(j.ID).Asserts(w, policy.ActionRead).Returns(j)
4385+
w := testutil.Fake(s.T(), faker, database.Workspace{ID: b.WorkspaceID})
4386+
dbm.EXPECT().GetWorkspaceBuildByJobID(gomock.Any(), j.ID).Return(b, nil).AnyTimes()
4387+
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), b.WorkspaceID).Return(w, nil).AnyTimes()
4388+
check.Args(j.ID).Asserts(w, policy.ActionRead).Returns(j)
43894389
}))
43904390
s.Run("TemplateVersion/GetProvisionerJobByIDForUpdate", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4391-
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeTemplateVersionImport})
4392-
tpl := testutil.Fake(s.T(), faker, database.Template{})
4393-
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4394-
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4395-
dbm.EXPECT().GetTemplateVersionByJobID(gomock.Any(), j.ID).Return(tv, nil).AnyTimes()
4396-
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4397-
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4398-
}))
4399-
s.Run("TemplateVersionDryRun/GetProvisionerJobByIDForUpdate", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4400-
tpl := testutil.Fake(s.T(), faker, database.Template{})
4401-
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4402-
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{})
4403-
j.Type = database.ProvisionerJobTypeTemplateVersionDryRun
4404-
j.Input = must(json.Marshal(struct { TemplateVersionID uuid.UUID `json:"template_version_id"` }{TemplateVersionID: tv.ID}))
4405-
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4406-
dbm.EXPECT().GetTemplateVersionByID(gomock.Any(), tv.ID).Return(tv, nil).AnyTimes()
4407-
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4408-
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4409-
}))
4391+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{Type: database.ProvisionerJobTypeTemplateVersionImport})
4392+
tpl := testutil.Fake(s.T(), faker, database.Template{})
4393+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4394+
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4395+
dbm.EXPECT().GetTemplateVersionByJobID(gomock.Any(), j.ID).Return(tv, nil).AnyTimes()
4396+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4397+
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4398+
}))
4399+
s.Run("TemplateVersionDryRun/GetProvisionerJobByIDForUpdate", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
4400+
tpl := testutil.Fake(s.T(), faker, database.Template{})
4401+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true}})
4402+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{})
4403+
j.Type = database.ProvisionerJobTypeTemplateVersionDryRun
4404+
j.Input = must(json.Marshal(struct {
4405+
TemplateVersionID uuid.UUID `json:"template_version_id"`
4406+
}{TemplateVersionID: tv.ID}))
4407+
dbm.EXPECT().GetProvisionerJobByIDForUpdate(gomock.Any(), j.ID).Return(j, nil).AnyTimes()
4408+
dbm.EXPECT().GetTemplateVersionByID(gomock.Any(), tv.ID).Return(tv, nil).AnyTimes()
4409+
dbm.EXPECT().GetTemplateByID(gomock.Any(), tpl.ID).Return(tpl, nil).AnyTimes()
4410+
check.Args(j.ID).Asserts(tv.RBACObject(tpl), policy.ActionRead).Returns(j)
4411+
}))
44104412
}
44114413

44124414
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