Pass earlyExit parameter to run method so tests don't hang

src/restore.ts

@@ -1,22 +1,26 @@
 import restoreImpl from "./restoreImpl";
 import { StateProvider } from "./stateProvider";
 
-async function run(): Promise<void> {
+async function run(earlyExit?: boolean | undefined): Promise<void> {
     try {
         await restoreImpl(new StateProvider());
     } catch (err) {
         console.error(err);
+        if (earlyExit) {
             process.exit(1);
         }
+    }
 
     // node will stay alive if any promises are not resolved,
     // which is a possibility if HTTP requests are dangling
     // due to retries or timeouts. We know that if we got here
     // that all promises that we care about have successfully
     // resolved, so simply exit with success.
+    if (earlyExit) {
         process.exit(0);
+    }
 }
 
-run();
+run(true);
 
 export default run;

src/restoreOnly.ts

@@ -1,22 +1,26 @@
 import restoreImpl from "./restoreImpl";
 import { NullStateProvider } from "./stateProvider";
 
-async function run(): Promise<void> {
+async function run(earlyExit?: boolean | undefined): Promise<void> {
     try {
         await restoreImpl(new NullStateProvider());
     } catch (err) {
         console.error(err);
+        if (earlyExit) {
             process.exit(1);
         }
+    }
 
     // node will stay alive if any promises are not resolved,
     // which is a possibility if HTTP requests are dangling
     // due to retries or timeouts. We know that if we got here
     // that all promises that we care about have successfully
     // resolved, so simply exit with success.
+    if (earlyExit) {
         process.exit(0);
+    }
 }
 
-run();
+run(true);
 
 export default run;