Skip to content

Commit b5ba222

Browse files
EmyrkEmyrk
committed
Passing tests
1 parent 55fbd8f commit b5ba222

File tree

1 file changed

+59
-1
lines changed

1 file changed

+59
-1
lines changed

‎coderd/database/dbauthz/dbauthz_test.go

+59-1
Original file line numberDiff line numberDiff line change
@@ -1178,6 +1178,34 @@ func (s *MethodTestSuite) TestWorkspace() {
11781178
agt := dbgen.WorkspaceAgent(s.T(), db, database.WorkspaceAgent{ResourceID: res.ID})
11791179
check.Args(agt.ID).Asserts(ws, rbac.ActionRead).Returns(agt)
11801180
}))
1181+
s.Run("GetWorkspaceAgentLifecycleStateByID", s.Subtest(func(db database.Store, check *expects) {
1182+
tpl := dbgen.Template(s.T(), db, database.Template{})
1183+
ws := dbgen.Workspace(s.T(), db, database.Workspace{
1184+
TemplateID: tpl.ID,
1185+
})
1186+
build := dbgen.WorkspaceBuild(s.T(), db, database.WorkspaceBuild{WorkspaceID: ws.ID, JobID: uuid.New()})
1187+
res := dbgen.WorkspaceResource(s.T(), db, database.WorkspaceResource{JobID: build.JobID})
1188+
agt := dbgen.WorkspaceAgent(s.T(), db, database.WorkspaceAgent{ResourceID: res.ID})
1189+
check.Args(agt.ID).Asserts(ws, rbac.ActionRead)
1190+
}))
1191+
s.Run("GetWorkspaceAgentMetadata", s.Subtest(func(db database.Store, check *expects) {
1192+
tpl := dbgen.Template(s.T(), db, database.Template{})
1193+
ws := dbgen.Workspace(s.T(), db, database.Workspace{
1194+
TemplateID: tpl.ID,
1195+
})
1196+
build := dbgen.WorkspaceBuild(s.T(), db, database.WorkspaceBuild{WorkspaceID: ws.ID, JobID: uuid.New()})
1197+
res := dbgen.WorkspaceResource(s.T(), db, database.WorkspaceResource{JobID: build.JobID})
1198+
agt := dbgen.WorkspaceAgent(s.T(), db, database.WorkspaceAgent{ResourceID: res.ID})
1199+
_ = db.InsertWorkspaceAgentMetadata(context.Background(), database.InsertWorkspaceAgentMetadataParams{
1200+
WorkspaceAgentID: agt.ID,
1201+
DisplayName: "test",
1202+
Key: "test",
1203+
})
1204+
check.Args(database.GetWorkspaceAgentMetadataParams{
1205+
WorkspaceAgentID: agt.ID,
1206+
Keys: []string{"test"},
1207+
}).Asserts(ws, rbac.ActionRead)
1208+
}))
11811209
s.Run("GetWorkspaceAgentByInstanceID", s.Subtest(func(db database.Store, check *expects) {
11821210
tpl := dbgen.Template(s.T(), db, database.Template{})
11831211
ws := dbgen.Workspace(s.T(), db, database.Workspace{
@@ -1878,6 +1906,15 @@ func (s *MethodTestSuite) TestSystemFunctions() {
18781906
Asserts(rbac.ResourceSystem, rbac.ActionRead).
18791907
Returns(slice.New(tv1, tv2, tv3))
18801908
}))
1909+
s.Run("GetParameterSchemasByJobID", s.Subtest(func(db database.Store, check *expects) {
1910+
tpl := dbgen.Template(s.T(), db, database.Template{})
1911+
tv := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
1912+
TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true},
1913+
})
1914+
job := dbgen.ProvisionerJob(s.T(), db, nil, database.ProvisionerJob{ID: tv.JobID})
1915+
check.Args(job.ID).
1916+
Asserts(tpl, rbac.ActionRead).Errors(sql.ErrNoRows)
1917+
}))
18811918
s.Run("GetWorkspaceAppsByAgentIDs", s.Subtest(func(db database.Store, check *expects) {
18821919
aWs := dbgen.Workspace(s.T(), db, database.Workspace{})
18831920
aBuild := dbgen.WorkspaceBuild(s.T(), db, database.WorkspaceBuild{WorkspaceID: aWs.ID, JobID: uuid.New()})
@@ -2128,6 +2165,27 @@ func (s *MethodTestSuite) TestSystemFunctions() {
21282165
check.Args(database.InsertTemplateVersionVariableParams{}).Asserts(rbac.ResourceSystem, rbac.ActionCreate)
21292166
}))
21302167
s.Run("UpdateInactiveUsersToDormant", s.Subtest(func(db database.Store, check *expects) {
2131-
check.Args(database.UpdateInactiveUsersToDormantParams{}).Asserts(rbac.ResourceSystem, rbac.ActionCreate)
2168+
check.Args(database.UpdateInactiveUsersToDormantParams{}).Asserts(rbac.ResourceSystem, rbac.ActionCreate).Errors(sql.ErrNoRows)
2169+
}))
2170+
s.Run("GetWorkspaceUniqueOwnerCountByTemplateIDs", s.Subtest(func(db database.Store, check *expects) {
2171+
check.Args([]uuid.UUID{uuid.New()}).Asserts(rbac.ResourceSystem, rbac.ActionRead)
2172+
}))
2173+
s.Run("GetWorkspaceAgentScriptsByAgentIDs", s.Subtest(func(db database.Store, check *expects) {
2174+
check.Args([]uuid.UUID{uuid.New()}).Asserts(rbac.ResourceSystem, rbac.ActionRead)
2175+
}))
2176+
s.Run("GetWorkspaceAgentLogSourcesByAgentIDs", s.Subtest(func(db database.Store, check *expects) {
2177+
check.Args([]uuid.UUID{uuid.New()}).Asserts(rbac.ResourceSystem, rbac.ActionRead)
2178+
}))
2179+
s.Run("GetProvisionerJobsByIDsWithQueuePosition", s.Subtest(func(db database.Store, check *expects) {
2180+
check.Args([]uuid.UUID{}).Asserts()
2181+
}))
2182+
s.Run("GetReplicaByID", s.Subtest(func(db database.Store, check *expects) {
2183+
check.Args(uuid.New()).Asserts(rbac.ResourceSystem, rbac.ActionRead).Errors(sql.ErrNoRows)
2184+
}))
2185+
s.Run("GetWorkspaceAgentAndOwnerByAuthToken", s.Subtest(func(db database.Store, check *expects) {
2186+
check.Args(uuid.New()).Asserts(rbac.ResourceSystem, rbac.ActionRead).Errors(sql.ErrNoRows)
2187+
}))
2188+
s.Run("GetUserLinksByUserID", s.Subtest(func(db database.Store, check *expects) {
2189+
check.Args(uuid.New()).Asserts(rbac.ResourceSystem, rbac.ActionRead)
21322190
}))
21332191
}

0 commit comments

Comments
 (0)