fix: Database fake

mafredri · mafredri · commit 7ce1b3bff587 · 2022-08-24T15:43:29.000+03:00
diff --git a/coderd/database/databasefake/databasefake.go b/coderd/database/databasefake/databasefake.go
@@ -2091,7 +2091,7 @@ func (q *fakeQuerier) UpdateProvisionerJobWithCompleteByID(_ context.Context, ar
 	return sql.ErrNoRows
 }
 
-func (q *fakeQuerier) UpdateWorkspace(_ context.Context, arg database.UpdateWorkspaceParams) error {
+func (q *fakeQuerier) UpdateWorkspace(_ context.Context, arg database.UpdateWorkspaceParams) (database.Workspace, error) {
 	q.mutex.Lock()
 	defer q.mutex.Unlock()
 
@@ -2104,17 +2104,17 @@ func (q *fakeQuerier) UpdateWorkspace(_ context.Context, arg database.UpdateWork
 				continue
 			}
 			if other.Name == arg.Name {
-				return &pq.Error{Code: "23505", Message: "duplicate key value violates unique constraint"}
+				return database.Workspace{}, &pq.Error{Code: "23505", Message: "duplicate key value violates unique constraint"}
 			}
 		}
 
 		workspace.Name = arg.Name
 		q.workspaces[i] = workspace
 
-		return nil
+		return workspace, nil
 	}
 
-	return sql.ErrNoRows
+	return database.Workspace{}, sql.ErrNoRows
 }
 
 func (q *fakeQuerier) UpdateWorkspaceAutostart(_ context.Context, arg database.UpdateWorkspaceAutostartParams) error {

Original file line number	Diff line number	Diff line change
`@@ -2091,7 +2091,7 @@ func (q *fakeQuerier) UpdateProvisionerJobWithCompleteByID(_ context.Context, ar`
`2091`	`2091`	`return sql.ErrNoRows`
`2092`	`2092`	`}`
`2093`	`2093`
`2094`		`-func (q *fakeQuerier) UpdateWorkspace(_ context.Context, arg database.UpdateWorkspaceParams) error {`
	`2094`	`+func (q *fakeQuerier) UpdateWorkspace(_ context.Context, arg database.UpdateWorkspaceParams) (database.Workspace, error) {`
`2095`	`2095`	`q.mutex.Lock()`
`2096`	`2096`	`defer q.mutex.Unlock()`
`2097`	`2097`
`@@ -2104,17 +2104,17 @@ func (q *fakeQuerier) UpdateWorkspace(_ context.Context, arg database.UpdateWork`
`2104`	`2104`	`continue`
`2105`	`2105`	`}`
`2106`	`2106`	`if other.Name == arg.Name {`
`2107`		`- return &pq.Error{Code: "23505", Message: "duplicate key value violates unique constraint"}`
	`2107`	`+ return database.Workspace{}, &pq.Error{Code: "23505", Message: "duplicate key value violates unique constraint"}`
`2108`	`2108`	`}`
`2109`	`2109`	`}`
`2110`	`2110`
`2111`	`2111`	`workspace.Name = arg.Name`
`2112`	`2112`	`q.workspaces[i] = workspace`
`2113`	`2113`
`2114`		`- return nil`
	`2114`	`+ return workspace, nil`
`2115`	`2115`	`}`
`2116`	`2116`
`2117`		`- return sql.ErrNoRows`
	`2117`	`+ return database.Workspace{}, sql.ErrNoRows`
`2118`	`2118`	`}`
`2119`	`2119`
`2120`	`2120`	`func (q *fakeQuerier) UpdateWorkspaceAutostart(_ context.Context, arg database.UpdateWorkspaceAutostartParams) error {`