fixup! feat: Add agent state reporting

mafredri · mafredri · commit 4da05730c5f8 · 2023-01-19T09:58:57.000Z
diff --git a/coderd/database/databasefake/databasefake.go b/coderd/database/databasefake/databasefake.go
@@ -3871,12 +3871,12 @@ func (q *fakeQuerier) GetQuotaConsumedForUser(_ context.Context, userID uuid.UUI
 	return sum, nil
 }
 
-func (q *fakeQuerier) UpdateWorkspaceAgentStateByID(_ context.Context, id uuid.UUID, state database.AgentState) error {
+func (q *fakeQuerier) UpdateWorkspaceAgentStateByID(_ context.Context, arg database.UpdateWorkspaceAgentStateByIDParams) error {
 	q.mutex.Lock()
 	defer q.mutex.Unlock()
 	for i, agent := range q.workspaceAgents {
-		if agent.ID == id {
-			agent.State = state
+		if agent.ID == arg.ID {
+			agent.State = arg.State
 			q.workspaceAgents[i] = agent
 			return nil
 		}
