Fix data race in unit tests

This commit is contained in:
Kegan Dougal 2022-03-02 10:49:29 +00:00
parent 00b3545b14
commit a4c918ee17

View File

@ -87,6 +87,12 @@ func (d *mockDeviceListUpdaterDatabase) MarkDeviceListStale(ctx context.Context,
return nil
}
func (d *mockDeviceListUpdaterDatabase) isStale(userID string) bool {
d.mu.Lock()
defer d.mu.Unlock()
return d.staleUsers[userID]
}
// StoreRemoteDeviceKeys persists the given keys. Keys with the same user ID and device ID will be replaced. An empty KeyJSON removes the key
// for this (user, device). Does not modify the stream ID for keys.
func (d *mockDeviceListUpdaterDatabase) StoreRemoteDeviceKeys(ctx context.Context, keys []api.DeviceMessage, clear []string) error {
@ -169,7 +175,7 @@ func TestUpdateHavePrevID(t *testing.T) {
if !reflect.DeepEqual(db.storedKeys, []api.DeviceMessage{want}) {
t.Errorf("DB didn't store correct event, got %v want %v", db.storedKeys, want)
}
if db.staleUsers[event.UserID] {
if db.isStale(event.UserID) {
t.Errorf("%s incorrectly marked as stale", event.UserID)
}
}
@ -243,7 +249,7 @@ func TestUpdateNoPrevID(t *testing.T) {
},
}
// Now we should have a fresh list and the keys and emitted something
if db.staleUsers[event.UserID] {
if db.isStale(event.UserID) {
t.Errorf("%s still marked as stale", event.UserID)
}
if !reflect.DeepEqual(producer.events, []api.DeviceMessage{want}) {
@ -304,7 +310,7 @@ func TestDebounce(t *testing.T) {
}
// user should be marked as stale
if !db.staleUsers[userID] {
if !db.isStale(userID) {
t.Errorf("user %s not marked as stale", userID)
}
// now send the response over federation
@ -330,7 +336,7 @@ func TestDebounce(t *testing.T) {
wg.Wait()
// user is no longer stale now
if db.staleUsers[userID] {
if db.isStale(userID) {
t.Errorf("user %s is marked as stale", userID)
}
}