From 4a4b5e13031898440421cef7230c2742e6cd226a Mon Sep 17 00:00:00 2001
From: Shin'ya UEOKA <ueokande@i-beam.org>
Date: Fri, 27 Sep 2019 10:36:49 +0000
Subject: [PATCH] Add graceful shutdown to TestServer

---
 e2e/lib/TestServer.ts | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/e2e/lib/TestServer.ts b/e2e/lib/TestServer.ts
index ba9845b..c010e37 100644
--- a/e2e/lib/TestServer.ts
+++ b/e2e/lib/TestServer.ts
@@ -37,16 +37,28 @@ export default class TestServer {
     return `http://${addr.address}:${addr.port}${path}`
   }
 
-  listen() {
+  start(): Promise<void>  {
+    if (this.http) {
+      throw new Error('http server already started');
+    }
+
     this.http = http.createServer(this.app)
-    this.http.listen(this.port, this.address);
+    return new Promise((resolve) => {
+      this.http!!.listen(this.port, this.address, () => {
+        resolve();
+      })
+    });
   }
 
-  close(): void {
+  stop(): Promise<void> {
     if (!this.http) {
-      return;
+      return Promise.resolve();
     }
-    this.http.close();
-    this.http = undefined;
+    return new Promise((resolve) => {
+      this.http!!.close(() => {
+        this.http = undefined;
+        resolve();
+      });
+    })
   }
 }