Skip to content

Commit db898a1

Browse files
kacpersawkacpersaw
committed
replace dbgen with testutil.Fake
1 parent 763855e commit db898a1

File tree

1 file changed

+26
-19
lines changed

1 file changed

+26
-19
lines changed

‎coderd/database/dbauthz/dbauthz_test.go

Lines changed: 26 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1537,19 +1537,22 @@ func (s *MethodTestSuite) TestTemplate() {
15371537
ID: t1.ID,
15381538
}).Asserts(t1, policy.ActionUpdate)
15391539
}))
1540-
s.Run("UpdateTemplateVersionFlagsByJobID", s.Subtest(func(db database.Store, check *expects) {
1541-
dbtestutil.DisableForeignKeysAndTriggers(s.T(), db)
1542-
o := dbgen.Organization(s.T(), db, database.Organization{})
1543-
u := dbgen.User(s.T(), db, database.User{})
1544-
_ = dbgen.OrganizationMember(s.T(), db, database.OrganizationMember{OrganizationID: o.ID, UserID: u.ID})
1545-
t := dbgen.Template(s.T(), db, database.Template{OrganizationID: o.ID, CreatedBy: u.ID})
1546-
job := dbgen.ProvisionerJob(s.T(), db, nil, database.ProvisionerJob{OrganizationID: o.ID})
1547-
_ = dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
1540+
s.Run("UpdateTemplateVersionFlagsByJobID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
1541+
o := testutil.Fake(s.T(), faker, database.Organization{})
1542+
u := testutil.Fake(s.T(), faker, database.User{})
1543+
_ = testutil.Fake(s.T(), faker, database.OrganizationMember{OrganizationID: o.ID, UserID: u.ID})
1544+
t := testutil.Fake(s.T(), faker, database.Template{OrganizationID: o.ID, CreatedBy: u.ID})
1545+
job := testutil.Fake(s.T(), faker, database.ProvisionerJob{OrganizationID: o.ID})
1546+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{
15481547
OrganizationID: o.ID,
15491548
CreatedBy: u.ID,
15501549
JobID: job.ID,
15511550
TemplateID: uuid.NullUUID{UUID: t.ID, Valid: true},
15521551
})
1552+
1553+
dbm.EXPECT().GetTemplateByID(gomock.Any(), t.ID).Return(t, nil).AnyTimes()
1554+
dbm.EXPECT().GetTemplateVersionByJobID(gomock.Any(), job.ID).Return(tv, nil).AnyTimes()
1555+
dbm.EXPECT().UpdateTemplateVersionFlagsByJobID(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()
15531556
check.Args(database.UpdateTemplateVersionFlagsByJobIDParams{
15541557
JobID: job.ID,
15551558
HasAITask: sql.NullBool{Bool: true, Valid: true},
@@ -3162,34 +3165,38 @@ func (s *MethodTestSuite) TestWorkspace() {
31623165
Deadline: b.Deadline,
31633166
}).Asserts(w, policy.ActionUpdate)
31643167
}))
3165-
s.Run("UpdateWorkspaceBuildFlagsByID", s.Subtest(func(db database.Store, check *expects) {
3166-
u := dbgen.User(s.T(), db, database.User{})
3167-
o := dbgen.Organization(s.T(), db, database.Organization{})
3168-
tpl := dbgen.Template(s.T(), db, database.Template{
3168+
s.Run("UpdateWorkspaceBuildFlagsByID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
3169+
u := testutil.Fake(s.T(), faker, database.User{})
3170+
o := testutil.Fake(s.T(), faker, database.Organization{})
3171+
tpl := testutil.Fake(s.T(), faker, database.Template{
31693172
OrganizationID: o.ID,
31703173
CreatedBy: u.ID,
31713174
})
3172-
tv := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
3175+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{
31733176
TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true},
31743177
OrganizationID: o.ID,
31753178
CreatedBy: u.ID,
31763179
})
3177-
w := dbgen.Workspace(s.T(), db, database.WorkspaceTable{
3180+
w := testutil.Fake(s.T(), faker, database.Workspace{
31783181
TemplateID: tpl.ID,
31793182
OrganizationID: o.ID,
31803183
OwnerID: u.ID,
31813184
})
3182-
j := dbgen.ProvisionerJob(s.T(), db, nil, database.ProvisionerJob{
3185+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{
31833186
Type: database.ProvisionerJobTypeWorkspaceBuild,
31843187
})
3185-
b := dbgen.WorkspaceBuild(s.T(), db, database.WorkspaceBuild{
3188+
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{
31863189
JobID: j.ID,
31873190
WorkspaceID: w.ID,
31883191
TemplateVersionID: tv.ID,
31893192
})
3190-
res := dbgen.WorkspaceResource(s.T(), db, database.WorkspaceResource{JobID: b.JobID})
3191-
agt := dbgen.WorkspaceAgent(s.T(), db, database.WorkspaceAgent{ResourceID: res.ID})
3192-
app := dbgen.WorkspaceApp(s.T(), db, database.WorkspaceApp{AgentID: agt.ID})
3193+
res := testutil.Fake(s.T(), faker, database.WorkspaceResource{JobID: b.JobID})
3194+
agt := testutil.Fake(s.T(), faker, database.WorkspaceAgent{ResourceID: res.ID})
3195+
app := testutil.Fake(s.T(), faker, database.WorkspaceApp{AgentID: agt.ID})
3196+
3197+
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), w.ID).Return(w, nil).AnyTimes()
3198+
dbm.EXPECT().GetWorkspaceBuildByID(gomock.Any(), b.ID).Return(b, nil).AnyTimes()
3199+
dbm.EXPECT().UpdateWorkspaceBuildFlagsByID(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()
31933200
check.Args(database.UpdateWorkspaceBuildFlagsByIDParams{
31943201
ID: b.ID,
31953202
HasAITask: sql.NullBool{Bool: true, Valid: true},

0 commit comments

Comments
 (0)