synced/coder
1
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2025-07-13 21:36:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix dbmem tests

Browse Source
This commit is contained in:
Sas Swart
2025-01-21 06:37:38 +00:00
parent f320807a16
commit 353f5a9555
2 changed files with 19 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
coderd/database/dbauthz/dbauthz_test.go
View File

@ -3741,6 +3741,10 @@ func (s *MethodTestSuite) TestSystemFunctions() {
OrganizationID: org.ID,
CreatedBy: user.ID,
})
_, err := db.InsertPreset(context.Background(), database.InsertPresetParams{
TemplateVersionID: templateVersion.ID,
Name: "test",
})
workspace := dbgen.Workspace(s.T(), db, database.WorkspaceTable{
OrganizationID: org.ID,
OwnerID: user.ID,
@ -3755,10 +3759,6 @@ func (s *MethodTestSuite) TestSystemFunctions() {
InitiatorID: user.ID,
JobID: job.ID,
})
_, err := db.InsertPreset(context.Background(), database.InsertPresetParams{
TemplateVersionID: workspaceBuild.TemplateVersionID,
Name: "test",
})
require.NoError(s.T(), err)
db.GetPresetByWorkspaceBuildID(context.Background(), workspaceBuild.ID)
check.Args(workspaceBuild.ID).Asserts(rbac.ResourceTemplate, policy.ActionRead)

8
coderd/database/dbmem/dbmem.go
View File

@ -3785,7 +3785,9 @@ func (q *FakeQuerier) GetPresetByWorkspaceBuildID(_ context.Context, workspaceBu
defer q.mutex.RUnlock()
for _, workspaceBuild := range q.workspaceBuilds {
if workspaceBuild.ID == workspaceBuildID {
if workspaceBuild.ID != workspaceBuildID {
continue
}
for _, preset := range q.presets {
if preset.TemplateVersionID == workspaceBuild.TemplateVersionID {
return database.GetPresetByWorkspaceBuildIDRow{
@ -3797,7 +3799,6 @@ func (q *FakeQuerier) GetPresetByWorkspaceBuildID(_ context.Context, workspaceBu
}
}
}
}
return database.GetPresetByWorkspaceBuildIDRow{}, sql.ErrNoRows
}
@ -8112,8 +8113,7 @@ func (q *FakeQuerier) InsertPreset(_ context.Context, arg database.InsertPresetP
defer q.mutex.Unlock()
preset := database.TemplateVersionPreset{
// TODO (sasswart): double check how we generate these IDs in postgres.
// They should not be params here.
TemplateVersionID: arg.TemplateVersionID,
Name: arg.Name,
CreatedAt: arg.CreatedAt,
UpdatedAt: arg.UpdatedAt,