From 049f12096d6cb40726bf4d68a9c98ed0438f8512 Mon Sep 17 00:00:00 2001
From: Yun Xu <yunx@zillowgroup.com>
Date: Sat, 25 Nov 2017 21:07:38 -0800
Subject: [PATCH] fix unit tests

---
 setup.py             | 2 +-
 tests/test_worker.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 17f2c67e..3f8cd7ce 100644
--- a/setup.py
+++ b/setup.py
@@ -59,7 +59,7 @@ requirements = [
     uvloop,
     ujson,
     'aiofiles>=0.3.0',
-    'websockets>=3.2',
+    'websockets>=4.0',
 ]
 if strtobool(os.environ.get("SANIC_NO_UJSON", "no")):
     print("Installing without uJSON")
diff --git a/tests/test_worker.py b/tests/test_worker.py
index e2b301ec..af00982e 100644
--- a/tests/test_worker.py
+++ b/tests/test_worker.py
@@ -131,5 +131,5 @@ def test_worker_close(worker):
     loop.run_until_complete(_close)
 
     assert worker.signal.stopped == True
-    conn.websocket.close_connection.assert_called_with(force=True)
+    conn.websocket.close_connection.assert_called_with(after_handshake=False)
     assert len(worker.servers) == 0