Add lockdown events in other environments

MetaMask · Mar 9, 2023 · 4e89e4e · 4e89e4e
1 parent bad541d
commit 4e89e4e
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 2 deletions.
diff --git a/packages/snaps-execution-environments/coverage.json b/packages/snaps-execution-environments/coverage.json
@@ -1,6 +1,6 @@
 {
   "branches": 77.6,
   "functions": 91.4,
-  "lines": 86.34,
-  "statements": 86.42
+  "lines": 89.65,
+  "statements": 89.71
 }
diff --git a/packages/snaps-execution-environments/jest.config.js b/packages/snaps-execution-environments/jest.config.js
@@ -8,6 +8,9 @@ module.exports = deepmerge(baseConfig, {
   coveragePathIgnorePatterns: [
     './src/index.ts',
     './src/iframe/index.ts',
+    './src/node-process/index.ts',
+    './src/node-thread/index.ts',
+    './src/offscreen/index.ts',
     './src/common/test-utils',
   ],
   coverageDirectory: './coverage/jest',

diff --git a/packages/snaps-execution-environments/src/node-process/index.ts b/packages/snaps-execution-environments/src/node-process/index.ts
@@ -1,7 +1,9 @@
 import { executeLockdown } from '../common/lockdown/lockdown';
+import { executeLockdownEvents } from '../common/lockdown/lockdown-events';
 import { executeLockdownMore } from '../common/lockdown/lockdown-more';
 import { ChildProcessSnapExecutor } from './ChildProcessSnapExecutor';
 
+executeLockdownEvents();
 executeLockdown();
 executeLockdownMore();
 

diff --git a/packages/snaps-execution-environments/src/node-thread/index.ts b/packages/snaps-execution-environments/src/node-thread/index.ts
@@ -1,7 +1,9 @@
 import { executeLockdown } from '../common/lockdown/lockdown';
+import { executeLockdownEvents } from '../common/lockdown/lockdown-events';
 import { executeLockdownMore } from '../common/lockdown/lockdown-more';
 import { ThreadSnapExecutor } from './ThreadSnapExecutor';
 
+executeLockdownEvents();
 executeLockdown();
 executeLockdownMore();
 

diff --git a/packages/snaps-execution-environments/src/offscreen/index.ts b/packages/snaps-execution-environments/src/offscreen/index.ts
@@ -1,9 +1,11 @@
 import { BrowserRuntimePostMessageStream } from '@metamask/post-message-stream';
 
 import { executeLockdown } from '../common/lockdown/lockdown';
+import { executeLockdownEvents } from '../common/lockdown/lockdown-events';
 import { executeLockdownMore } from '../common/lockdown/lockdown-more';
 import { OffscreenSnapExecutor } from './OffscreenSnapExecutor';
 
+executeLockdownEvents();
 executeLockdown();
 executeLockdownMore();