Skip to content

Commit 53094ad

Browse files
kacpersawkacpersaw
committed
replace dbgen with testutil.Fake
1 parent 31dcf69 commit 53094ad

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
@@ -1523,19 +1523,22 @@ func (s *MethodTestSuite) TestTemplate() {
15231523
ID: t1.ID,
15241524
}).Asserts(t1, policy.ActionUpdate)
15251525
}))
1526-
s.Run("UpdateTemplateVersionFlagsByJobID", s.Subtest(func(db database.Store, check *expects) {
1527-
dbtestutil.DisableForeignKeysAndTriggers(s.T(), db)
1528-
o := dbgen.Organization(s.T(), db, database.Organization{})
1529-
u := dbgen.User(s.T(), db, database.User{})
1530-
_ = dbgen.OrganizationMember(s.T(), db, database.OrganizationMember{OrganizationID: o.ID, UserID: u.ID})
1531-
t := dbgen.Template(s.T(), db, database.Template{OrganizationID: o.ID, CreatedBy: u.ID})
1532-
job := dbgen.ProvisionerJob(s.T(), db, nil, database.ProvisionerJob{OrganizationID: o.ID})
1533-
_ = dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
1526+
s.Run("UpdateTemplateVersionFlagsByJobID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
1527+
o := testutil.Fake(s.T(), faker, database.Organization{})
1528+
u := testutil.Fake(s.T(), faker, database.User{})
1529+
_ = testutil.Fake(s.T(), faker, database.OrganizationMember{OrganizationID: o.ID, UserID: u.ID})
1530+
t := testutil.Fake(s.T(), faker, database.Template{OrganizationID: o.ID, CreatedBy: u.ID})
1531+
job := testutil.Fake(s.T(), faker, database.ProvisionerJob{OrganizationID: o.ID})
1532+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{
15341533
OrganizationID: o.ID,
15351534
CreatedBy: u.ID,
15361535
JobID: job.ID,
15371536
TemplateID: uuid.NullUUID{UUID: t.ID, Valid: true},
15381537
})
1538+
1539+
dbm.EXPECT().GetTemplateByID(gomock.Any(), t.ID).Return(t, nil).AnyTimes()
1540+
dbm.EXPECT().GetTemplateVersionByJobID(gomock.Any(), job.ID).Return(tv, nil).AnyTimes()
1541+
dbm.EXPECT().UpdateTemplateVersionFlagsByJobID(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()
15391542
check.Args(database.UpdateTemplateVersionFlagsByJobIDParams{
15401543
JobID: job.ID,
15411544
HasAITask: sql.NullBool{Bool: true, Valid: true},
@@ -3230,34 +3233,38 @@ func (s *MethodTestSuite) TestWorkspace() {
32303233
Deadline: b.Deadline,
32313234
}).Asserts(w, policy.ActionUpdate)
32323235
}))
3233-
s.Run("UpdateWorkspaceBuildFlagsByID", s.Subtest(func(db database.Store, check *expects) {
3234-
u := dbgen.User(s.T(), db, database.User{})
3235-
o := dbgen.Organization(s.T(), db, database.Organization{})
3236-
tpl := dbgen.Template(s.T(), db, database.Template{
3236+
s.Run("UpdateWorkspaceBuildFlagsByID", s.Mocked(func(dbm *dbmock.MockStore, faker *gofakeit.Faker, check *expects) {
3237+
u := testutil.Fake(s.T(), faker, database.User{})
3238+
o := testutil.Fake(s.T(), faker, database.Organization{})
3239+
tpl := testutil.Fake(s.T(), faker, database.Template{
32373240
OrganizationID: o.ID,
32383241
CreatedBy: u.ID,
32393242
})
3240-
tv := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
3243+
tv := testutil.Fake(s.T(), faker, database.TemplateVersion{
32413244
TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true},
32423245
OrganizationID: o.ID,
32433246
CreatedBy: u.ID,
32443247
})
3245-
w := dbgen.Workspace(s.T(), db, database.WorkspaceTable{
3248+
w := testutil.Fake(s.T(), faker, database.Workspace{
32463249
TemplateID: tpl.ID,
32473250
OrganizationID: o.ID,
32483251
OwnerID: u.ID,
32493252
})
3250-
j := dbgen.ProvisionerJob(s.T(), db, nil, database.ProvisionerJob{
3253+
j := testutil.Fake(s.T(), faker, database.ProvisionerJob{
32513254
Type: database.ProvisionerJobTypeWorkspaceBuild,
32523255
})
3253-
b := dbgen.WorkspaceBuild(s.T(), db, database.WorkspaceBuild{
3256+
b := testutil.Fake(s.T(), faker, database.WorkspaceBuild{
32543257
JobID: j.ID,
32553258
WorkspaceID: w.ID,
32563259
TemplateVersionID: tv.ID,
32573260
})
3258-
res := dbgen.WorkspaceResource(s.T(), db, database.WorkspaceResource{JobID: b.JobID})
3259-
agt := dbgen.WorkspaceAgent(s.T(), db, database.WorkspaceAgent{ResourceID: res.ID})
3260-
app := dbgen.WorkspaceApp(s.T(), db, database.WorkspaceApp{AgentID: agt.ID})
3261+
res := testutil.Fake(s.T(), faker, database.WorkspaceResource{JobID: b.JobID})
3262+
agt := testutil.Fake(s.T(), faker, database.WorkspaceAgent{ResourceID: res.ID})
3263+
app := testutil.Fake(s.T(), faker, database.WorkspaceApp{AgentID: agt.ID})
3264+
3265+
dbm.EXPECT().GetWorkspaceByID(gomock.Any(), w.ID).Return(w, nil).AnyTimes()
3266+
dbm.EXPECT().GetWorkspaceBuildByID(gomock.Any(), b.ID).Return(b, nil).AnyTimes()
3267+
dbm.EXPECT().UpdateWorkspaceBuildFlagsByID(gomock.Any(), gomock.Any()).Return(nil).AnyTimes()
32613268
check.Args(database.UpdateWorkspaceBuildFlagsByIDParams{
32623269
ID: b.ID,
32633270
HasAITask: sql.NullBool{Bool: true, Valid: true},

0 commit comments

Comments
 (0)