diff --git a/libs/common/spec/fake-state-provider.ts b/libs/common/spec/fake-state-provider.ts index ae3233405f..5b0be6ccf9 100644 --- a/libs/common/spec/fake-state-provider.ts +++ b/libs/common/spec/fake-state-provider.ts @@ -32,7 +32,8 @@ export class FakeGlobalStateProvider implements GlobalStateProvider { states: Map> = new Map(); get(keyDefinition: KeyDefinition): GlobalState { this.mock.get(keyDefinition); - let result = this.states.get(keyDefinition.fullName); + const cacheKey = `${keyDefinition.fullName}_${keyDefinition.stateDefinition.defaultStorageLocation}`; + let result = this.states.get(cacheKey); if (result == null) { let fake: FakeGlobalState; @@ -44,10 +45,10 @@ export class FakeGlobalStateProvider implements GlobalStateProvider { } fake.keyDefinition = keyDefinition; result = fake; - this.states.set(keyDefinition.fullName, result); + this.states.set(cacheKey, result); result = new FakeGlobalState(); - this.states.set(keyDefinition.fullName, result); + this.states.set(cacheKey, result); } return result as GlobalState; } @@ -76,7 +77,8 @@ export class FakeSingleUserStateProvider implements SingleUserStateProvider { if (keyDefinition instanceof KeyDefinition) { keyDefinition = UserKeyDefinition.fromBaseKeyDefinition(keyDefinition); } - let result = this.states.get(`${keyDefinition.fullName}_${userId}`); + const cacheKey = `${keyDefinition.fullName}_${keyDefinition.stateDefinition.defaultStorageLocation}_${userId}`; + let result = this.states.get(cacheKey); if (result == null) { let fake: FakeSingleUserState; @@ -88,7 +90,7 @@ export class FakeSingleUserStateProvider implements SingleUserStateProvider { } fake.keyDefinition = keyDefinition; result = fake; - this.states.set(`${keyDefinition.fullName}_${userId}`, result); + this.states.set(cacheKey, result); } return result as SingleUserState; } @@ -119,7 +121,8 @@ export class FakeActiveUserStateProvider implements ActiveUserStateProvider { if (keyDefinition instanceof KeyDefinition) { keyDefinition = UserKeyDefinition.fromBaseKeyDefinition(keyDefinition); } - let result = this.states.get(keyDefinition.fullName); + const cacheKey = `${keyDefinition.fullName}_${keyDefinition.stateDefinition.defaultStorageLocation}`; + let result = this.states.get(cacheKey); if (result == null) { // Look for established mock @@ -129,7 +132,7 @@ export class FakeActiveUserStateProvider implements ActiveUserStateProvider { result = new FakeActiveUserState(this.accountService); } result.keyDefinition = keyDefinition; - this.states.set(keyDefinition.fullName, result); + this.states.set(cacheKey, result); } return result as ActiveUserState; }