Skip to content

Commit 5d2f9b1

Browse files
EmyrkEmyrk
committed
Another chunk of tests
1 parent 2e2f6be commit 5d2f9b1

File tree

1 file changed

+47
-0
lines changed

1 file changed

+47
-0
lines changed

‎coderd/database/dbauthz/dbauthz_test.go

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -987,6 +987,12 @@ func (s *MethodTestSuite) TestUser() {
987987
ID: u.ID,
988988
}).Asserts(u.UserDataRBACObject(), rbac.ActionUpdate).Returns()
989989
}))
990+
s.Run("UpdateUserQuietHoursSchedule", s.Subtest(func(db database.Store, check *expects) {
991+
u := dbgen.User(s.T(), db, database.User{})
992+
check.Args(database.UpdateUserQuietHoursScheduleParams{
993+
ID: u.ID,
994+
}).Asserts(u.UserDataRBACObject(), rbac.ActionUpdate)
995+
}))
990996
s.Run("UpdateUserLastSeenAt", s.Subtest(func(db database.Store, check *expects) {
991997
u := dbgen.User(s.T(), db, database.User{})
992998
check.Args(database.UpdateUserLastSeenAtParams{
@@ -1152,6 +1158,18 @@ func (s *MethodTestSuite) TestWorkspace() {
11521158
LifecycleState: database.WorkspaceAgentLifecycleStateCreated,
11531159
}).Asserts(ws, rbac.ActionUpdate).Returns()
11541160
}))
1161+
s.Run("UpdateWorkspaceAgentMetadata", s.Subtest(func(db database.Store, check *expects) {
1162+
tpl := dbgen.Template(s.T(), db, database.Template{})
1163+
ws := dbgen.Workspace(s.T(), db, database.Workspace{
1164+
TemplateID: tpl.ID,
1165+
})
1166+
build := dbgen.WorkspaceBuild(s.T(), db, database.WorkspaceBuild{WorkspaceID: ws.ID, JobID: uuid.New()})
1167+
res := dbgen.WorkspaceResource(s.T(), db, database.WorkspaceResource{JobID: build.JobID})
1168+
agt := dbgen.WorkspaceAgent(s.T(), db, database.WorkspaceAgent{ResourceID: res.ID})
1169+
check.Args(database.UpdateWorkspaceAgentMetadataParams{
1170+
WorkspaceAgentID: agt.ID,
1171+
}).Asserts(ws, rbac.ActionUpdate).Returns()
1172+
}))
11551173
s.Run("UpdateWorkspaceAgentLogOverflowByID", s.Subtest(func(db database.Store, check *expects) {
11561174
tpl := dbgen.Template(s.T(), db, database.Template{})
11571175
ws := dbgen.Workspace(s.T(), db, database.Workspace{
@@ -1264,6 +1282,16 @@ func (s *MethodTestSuite) TestWorkspace() {
12641282
TemplateName: tpl.Name,
12651283
})
12661284
}))
1285+
s.Run("GetWorkspaceAgentsInLatestBuildByWorkspaceID", s.Subtest(func(db database.Store, check *expects) {
1286+
tpl := dbgen.Template(s.T(), db, database.Template{})
1287+
ws := dbgen.Workspace(s.T(), db, database.Workspace{
1288+
TemplateID: tpl.ID,
1289+
})
1290+
build := dbgen.WorkspaceBuild(s.T(), db, database.WorkspaceBuild{WorkspaceID: ws.ID, JobID: uuid.New()})
1291+
res := dbgen.WorkspaceResource(s.T(), db, database.WorkspaceResource{JobID: build.JobID})
1292+
dbgen.WorkspaceAgent(s.T(), db, database.WorkspaceAgent{ResourceID: res.ID})
1293+
check.Args(ws.ID).Asserts(ws, rbac.ActionRead)
1294+
}))
12671295
s.Run("GetWorkspaceByOwnerIDAndName", s.Subtest(func(db database.Store, check *expects) {
12681296
ws := dbgen.Workspace(s.T(), db, database.Workspace{})
12691297
check.Args(database.GetWorkspaceByOwnerIDAndNameParams{
@@ -1987,6 +2015,9 @@ func (s *MethodTestSuite) TestSystemFunctions() {
19872015
s.Run("GetHealthSettings", s.Subtest(func(db database.Store, check *expects) {
19882016
check.Args().Asserts()
19892017
}))
2018+
s.Run("UpsertHealthSettings", s.Subtest(func(db database.Store, check *expects) {
2019+
check.Args("foo").Asserts(rbac.ResourceDeploymentValues, rbac.ActionCreate)
2020+
}))
19902021
s.Run("GetDeploymentWorkspaceAgentStats", s.Subtest(func(db database.Store, check *expects) {
19912022
check.Args(time.Time{}).Asserts()
19922023
}))
@@ -1999,11 +2030,27 @@ func (s *MethodTestSuite) TestSystemFunctions() {
19992030
s.Run("GetHungProvisionerJobs", s.Subtest(func(db database.Store, check *expects) {
20002031
check.Args(time.Time{}).Asserts()
20012032
}))
2033+
s.Run("UpsertOAuthSigningKey", s.Subtest(func(db database.Store, check *expects) {
2034+
check.Args("foo").Asserts(rbac.ResourceSystem, rbac.ActionUpdate)
2035+
}))
20022036
s.Run("GetOAuthSigningKey", s.Subtest(func(db database.Store, check *expects) {
20032037
db.UpsertOAuthSigningKey(context.Background(), "foo")
20042038
check.Args().Asserts(rbac.ResourceSystem, rbac.ActionUpdate)
20052039
}))
20062040
s.Run("InsertMissingGroups", s.Subtest(func(db database.Store, check *expects) {
20072041
check.Args(database.InsertMissingGroupsParams{}).Asserts(rbac.ResourceSystem, rbac.ActionCreate).Errors(matchAnyError)
20082042
}))
2043+
s.Run("UpdateUserLoginType", s.Subtest(func(db database.Store, check *expects) {
2044+
u := dbgen.User(s.T(), db, database.User{})
2045+
check.Args(database.UpdateUserLoginTypeParams{
2046+
NewLoginType: database.LoginTypePassword,
2047+
UserID: u.ID,
2048+
}).Asserts(rbac.ResourceSystem, rbac.ActionUpdate)
2049+
}))
2050+
s.Run("GetWorkspaceAgentStatsAndLabels", s.Subtest(func(db database.Store, check *expects) {
2051+
check.Args(time.Time{}).Asserts()
2052+
}))
2053+
s.Run("GetWorkspaceAgentStats", s.Subtest(func(db database.Store, check *expects) {
2054+
check.Args(time.Time{}).Asserts()
2055+
}))
20092056
}

0 commit comments

Comments
 (0)