fix: Database fake

mafredri · mafredri · commit 952b5d3f32af · 2022-08-09T19:33:32.000+03:00
diff --git a/coderd/database/databasefake/databasefake.go b/coderd/database/databasefake/databasefake.go
@@ -2045,7 +2045,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()
 
@@ -2058,17 +2058,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
`@@ -2045,7 +2045,7 @@ func (q *fakeQuerier) UpdateProvisionerJobWithCompleteByID(_ context.Context, ar`
`2045`	`2045`	`return sql.ErrNoRows`
`2046`	`2046`	`}`
`2047`	`2047`
`2048`		`-func (q *fakeQuerier) UpdateWorkspace(_ context.Context, arg database.UpdateWorkspaceParams) error {`
	`2048`	`+func (q *fakeQuerier) UpdateWorkspace(_ context.Context, arg database.UpdateWorkspaceParams) (database.Workspace, error) {`
`2049`	`2049`	`q.mutex.Lock()`
`2050`	`2050`	`defer q.mutex.Unlock()`
`2051`	`2051`
`@@ -2058,17 +2058,17 @@ func (q *fakeQuerier) UpdateWorkspace(_ context.Context, arg database.UpdateWork`
`2058`	`2058`	`continue`
`2059`	`2059`	`}`
`2060`	`2060`	`if other.Name == arg.Name {`
`2061`		`- return &pq.Error{Code: "23505", Message: "duplicate key value violates unique constraint"}`
	`2061`	`+ return database.Workspace{}, &pq.Error{Code: "23505", Message: "duplicate key value violates unique constraint"}`
`2062`	`2062`	`}`
`2063`	`2063`	`}`
`2064`	`2064`
`2065`	`2065`	`workspace.Name = arg.Name`
`2066`	`2066`	`q.workspaces[i] = workspace`
`2067`	`2067`
`2068`		`- return nil`
	`2068`	`+ return workspace, nil`
`2069`	`2069`	`}`
`2070`	`2070`
`2071`		`- return sql.ErrNoRows`
	`2071`	`+ return database.Workspace{}, sql.ErrNoRows`
`2072`	`2072`	`}`
`2073`	`2073`
`2074`	`2074`	`func (q *fakeQuerier) UpdateWorkspaceAutostart(_ context.Context, arg database.UpdateWorkspaceAutostartParams) error {`