Skip to content

Commit cd55fd7

Browse files
committed
Refactor helper and test
1 parent f993b7b commit cd55fd7

File tree

2 files changed

+14
-19
lines changed

2 files changed

+14
-19
lines changed

packages/snaps-jest/src/helpers.test.tsx

+12-14
Original file line numberDiff line numberDiff line change
@@ -734,11 +734,20 @@ describe('installSnap', () => {
734734
describe('onKeyringRequest', () => {
735735
it('sends a keyring request and returns the result', async () => {
736736
jest.spyOn(console, 'log').mockImplementation();
737+
const mockRequestObject = {
738+
params: {
739+
options: {
740+
privateKey: 'foo bar',
741+
},
742+
},
743+
id: 1,
744+
method: 'keyring_listAccounts',
745+
jsonrpc: '2.0',
746+
} as const;
737747

738748
const { snapId, close: closeServer } = await getMockServer({
739749
sourceCode: `
740750
module.exports.onKeyringRequest = async ({ origin, request }) => {
741-
// return handleKeyringRequest(keyring, request);
742751
return request;
743752
}
744753
`,
@@ -747,24 +756,13 @@ describe('installSnap', () => {
747756
const { onKeyringRequest, close } = await installSnap(snapId);
748757
const response = await onKeyringRequest({
749758
origin: 'metamask.io',
750-
request: {
751-
params: {
752-
foo: 'bar',
753-
},
754-
},
759+
request: mockRequestObject,
755760
});
756761

757762
expect(response).toStrictEqual(
758763
expect.objectContaining({
759764
response: {
760-
result: {
761-
id: 1,
762-
jsonrpc: '2.0',
763-
method: '',
764-
params: {
765-
foo: 'bar',
766-
},
767-
},
765+
result: mockRequestObject,
768766
},
769767
}),
770768
);

packages/snaps-simulation/src/helpers.ts

+2-5
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ export function getHelpers({
233233
): Promise<SnapResponseWithInterface> => {
234234
log('Sending keyring request %o.', request);
235235

236-
const { origin: keyringRequestOrigin, ...keyringRequest } = create(
236+
const { origin: keyringRequestOrigin, request: keyringRequest } = create(
237237
request,
238238
KeyringOptionsStruct,
239239
);
@@ -245,10 +245,7 @@ export function getHelpers({
245245
runSaga,
246246
controllerMessenger,
247247
handler: HandlerType.OnKeyringRequest,
248-
request: {
249-
method: '',
250-
params: keyringRequest.request.params,
251-
},
248+
request: { origin: keyringRequestOrigin, ...keyringRequest },
252249
});
253250

254251
assertIsResponseWithInterface(response);

0 commit comments

Comments
 (0)