added default in switch cases and use spread in rewireContext

nktnet1 · Oct 12, 2023 · c412424 · c412424
1 parent bddf581
commit c412424
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/files.ts b/src/files.ts
@@ -75,6 +75,9 @@ const retrieveSymbolsFromAst = (node: Statement, symbols: Symbols): void => {
       if (node.id !== null) {
         symbols.classes.push(node.id.name);
       }
+      break;
+    default:
+      break;
   }
 };
 

diff --git a/src/jewire.ts b/src/jewire.ts
@@ -24,9 +24,8 @@ const jewire = (relativePath: string, options: Options = {}): Record<string, any
   const rewiredModule = rewire(filePath);
 
   const rewireContext = {
+    ...rewiredModule,
     __get__: (name: string) => entityClone(rewiredModule.__get__(name), options.objectClone),
-    __set__: rewiredModule.__set__,
-    __with__: rewiredModule.__with__,
   };
 
   const entities: Record<string, any> = {};