Merge pull request #360 from matrix-org/dmr/invalidate-token-mutex

matrix-org · Nov 2, 2023 · 13d4e02 · 13d4e02
2 parents 0e8d4a8 + 5e65210
commit 13d4e02
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests-integration/v3_test.go b/tests-integration/v3_test.go
@@ -99,8 +99,10 @@ func (s *testV2Server) addAccountWithDeviceID(userID, deviceID, token string) {
 
 // like invalidateToken, but doesn't do any waiting.
 func (s *testV2Server) invalidateTokenImmediately(token string) {
+	s.mu.Lock()
 	delete(s.tokenToUser, token)
 	delete(s.tokenToDevice, token)
+	s.mu.Unlock()
 }
 
 // remove the token and wait until the proxy sends a request with this token, then 401 it and return.