Fix update of got

mafredri · mafredri · commit 33ef00dc635b · 2023-06-09T14:33:35.000Z
diff --git a/coderd/database/dbfake/databasefake.go b/coderd/database/dbfake/databasefake.go
@@ -2751,6 +2751,10 @@ func (q *fakeQuerier) InsertAPIKey(_ context.Context, arg database.InsertAPIKeyP
 }
 
 func (q *fakeQuerier) UpdateWorkspaceAgentMetadata(_ context.Context, arg database.UpdateWorkspaceAgentMetadataParams) error {
+	if err := validateDatabaseType(arg); err != nil {
+		return err
+	}
+
 	q.mutex.Lock()
 	defer q.mutex.Unlock()
 
@@ -2774,6 +2778,10 @@ func (q *fakeQuerier) UpdateWorkspaceAgentMetadata(_ context.Context, arg databa
 }
 
 func (q *fakeQuerier) InsertWorkspaceAgentMetadata(_ context.Context, arg database.InsertWorkspaceAgentMetadataParams) error {
+	if err := validateDatabaseType(arg); err != nil {
+		return err
+	}
+
 	q.mutex.Lock()
 	defer q.mutex.Unlock()
 
diff --git a/coderd/workspaceagents_test.go b/coderd/workspaceagents_test.go
@@ -1294,7 +1294,13 @@ func TestWorkspaceAgent_Metadata(t *testing.T) {
 		// We can't trust the order of the updates due to timers and debounces,
 		// so let's check a few times once more.
 		for i := 0; i < 2 && (want.Value != got.Result.Value || want.Error != got.Result.Error); i++ {
-			recvUpdate()
+			update = recvUpdate()
+			for _, m := range update {
+				if m.Description.Key == got.Description.Key {
+					got = m
+					break
+				}
+			}
 		}
 		ok1 := assert.Equal(t, want.Value, got.Result.Value)
 		ok2 := assert.Equal(t, want.Error, got.Result.Error)
