Skip to content

Commit b53fe57

Browse files
committed
fix: replace getPersistentState with deriveStateFromMetadata
1 parent f6f1394 commit b53fe57

File tree

1 file changed

+12
-10
lines changed

1 file changed

+12
-10
lines changed

packages/snaps-controllers/src/snaps/SnapController.test.tsx

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,4 @@
1-
import {
2-
getPersistentState,
3-
deriveStateFromMetadata,
4-
} from '@metamask/base-controller';
1+
import { deriveStateFromMetadata } from '@metamask/base-controller';
52
import { encrypt } from '@metamask/browser-passworder';
63
import {
74
createAsyncMiddleware,
@@ -88,7 +85,10 @@ import {
8885
STATE_DEBOUNCE_TIMEOUT,
8986
} from './constants';
9087
import { SnapsRegistryStatus } from './registry';
91-
import type { SnapControllerState } from './SnapController';
88+
import type {
89+
PersistedSnapControllerState,
90+
SnapControllerState,
91+
} from './SnapController';
9292
import {
9393
SNAP_APPROVAL_INSTALL,
9494
SNAP_APPROVAL_RESULT,
@@ -12622,7 +12622,7 @@ describe('SnapController', () => {
1262212622
deriveStateFromMetadata(
1262312623
controller.state,
1262412624
controller.metadata,
12625-
'anonymous',
12625+
'includeInDebugSnapshot',
1262612626
),
1262712627
).toMatchInlineSnapshot(`{}`);
1262812628
});
@@ -12748,15 +12748,16 @@ describe('SnapController', () => {
1274812748
);
1274912749

1275012750
// persist the state somewhere
12751-
const persistedState = getPersistentState<SnapControllerState>(
12751+
const persistedState = deriveStateFromMetadata<SnapControllerState>(
1275212752
firstSnapController.state,
1275312753
firstSnapController.metadata,
12754+
'persist',
1275412755
);
1275512756

1275612757
// create a new controller
1275712758
const secondSnapController = getSnapController(
1275812759
getSnapControllerOptions({
12759-
state: persistedState,
12760+
state: persistedState as PersistedSnapControllerState,
1276012761
}),
1276112762
);
1276212763

@@ -12787,15 +12788,16 @@ describe('SnapController', () => {
1278712788
expect(firstSnapController.state.snaps[MOCK_SNAP_ID]).toBeDefined();
1278812789

1278912790
// persist the state somewhere
12790-
const persistedState = getPersistentState<SnapControllerState>(
12791+
const persistedState = deriveStateFromMetadata<SnapControllerState>(
1279112792
firstSnapController.state,
1279212793
firstSnapController.metadata,
12794+
'persist',
1279312795
);
1279412796

1279512797
// create a new controller
1279612798
const secondSnapController = getSnapController(
1279712799
getSnapControllerOptions({
12798-
state: persistedState,
12800+
state: persistedState as PersistedSnapControllerState,
1279912801
}),
1280012802
);
1280112803

0 commit comments

Comments
 (0)