Add test

MetaMask · Dec 3, 2024 · ae074db · ae074db
1 parent d882f64
commit ae074db
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/packages/snaps-controllers/coverage.json b/packages/snaps-controllers/coverage.json
@@ -1,6 +1,6 @@
 {
-  "branches": 92.89,
+  "branches": 92.91,
   "functions": 96.71,
-  "lines": 98,
+  "lines": 98.01,
   "statements": 97.71
 }
diff --git a/packages/snaps-controllers/src/test-utils/webview.ts b/packages/snaps-controllers/src/test-utils/webview.ts
@@ -1,4 +1,9 @@
-import { base64ToBytes, bytesToString } from '@metamask/utils';
+import {
+  base64ToBytes,
+  bytesToBase64,
+  bytesToString,
+  stringToBytes,
+} from '@metamask/utils';
 
 import { WebViewMessageStream } from '../services/webview/WebViewMessageStream';
 
@@ -52,6 +57,11 @@ export function createWebViewObjects() {
     name: 'a',
     target: 'b',
     getWebView: mockGetWebViewA,
+    // For one of the streams, we test that a custom btoa function is used.
+    btoa: (data: string) => {
+      const bytes = stringToBytes(data);
+      return bytesToBase64(bytes);
+    },
   });
 
   const streamB = new WebViewMessageStream({