Check indexes in dbmem

code-asher · code-asher · commit a590352ed9fc · 2024-02-08T17:58:12.000-09:00
Might have no match, and a -1 will panic.
diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go
@@ -1312,7 +1312,8 @@ func (q *FakeQuerier) DeleteOAuth2ProviderAppTokensByAppAndUserID(_ context.Cont
 		keyIdx := slices.IndexFunc(q.apiKeys, func(key database.APIKey) bool {
 			return key.ID == token.APIKeyID
 		})
-		if q.oauth2ProviderAppSecrets[secretIdx].AppID == arg.AppID && q.apiKeys[keyIdx].UserID == arg.UserID {
+		if secretIdx != -1 && q.oauth2ProviderAppSecrets[secretIdx].AppID == arg.AppID &&
+			keyIdx != -1 && q.apiKeys[keyIdx].UserID == arg.UserID {
 			keyIDsToDelete = append(keyIDsToDelete, token.APIKeyID)
 		} else {
 			tokens = append(tokens, token)
@@ -2351,7 +2352,7 @@ func (q *FakeQuerier) GetOAuth2ProviderAppsByUserID(_ context.Context, userID uu
 						keyIdx := slices.IndexFunc(q.apiKeys, func(key database.APIKey) bool {
 							return key.ID == token.APIKeyID
 						})
-						if q.apiKeys[keyIdx].UserID == userID {
+						if keyIdx != -1 && q.apiKeys[keyIdx].UserID == userID {
 							tokens = append(tokens, token)
 						}
 					}
