test: try and close the demo server cleanly

snyk · Aug 28, 2019 · 3d90fb9 · 3d90fb9
1 parent 4b9407e
commit 3d90fb9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/test/disable.test.js b/test/disable.test.js
@@ -34,5 +34,5 @@ test('agent can be disabled', async (t) => {
   delete process.env.SNYK_BEACON_INTERVAL_MS;
   delete process.env.SNYK_RUNTIME_AGENT_DISABLE;
 
-  demoApp.close();
+  await new Promise((resolve) => demoApp.close(resolve));
 });
diff --git a/test/e2e.test.js b/test/e2e.test.js
@@ -183,5 +183,5 @@ test('demo app reports a vuln method when called', async (t) => {
   delete process.env.SNYK_SNAPSHOT_INTERVAL_MS;
   delete process.env.SNYK_TRIGGER_EXTRA_VULN;
 
-  demoApp.close();
+  await new Promise((resolve) => demoApp.close(resolve));
 });
diff --git a/test/failures.test.js b/test/failures.test.js
@@ -25,8 +25,7 @@ test('node agent does not crash the demo app', async (t) => {
   delete process.env.SNYK_HOMEBASE_URL;
   delete process.env.SNYK_BEACON_INTERVAL_MS;
 
-  demoApp.close();
-
+  await new Promise((resolve) => demoApp.close(resolve));
 });
 
 test('node agent does not crash the demo app', async (t) => {
@@ -45,5 +44,5 @@ test('node agent does not crash the demo app', async (t) => {
   delete process.env.SNYK_SNAPSHOT_INTERVAL_MS;
   delete process.env.SNYK_BEACON_INTERVAL_MS;
 
-  demoApp.close();
+  await new Promise((resolve) => demoApp.close(resolve));
 });