fix unit tests

2017-11-25 21:07:38 -08:00
parent f09c0393ba
commit 049f12096d
2 changed files with 2 additions and 2 deletions
--- 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