From b1f302151b32e14d77da9604a19dbfb1b22306e0 Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Wed, 13 Dec 2017 21:09:43 +0200
Subject: [PATCH] tests: fix test_server.py

---
 libagent/tests/test_server.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/libagent/tests/test_server.py b/libagent/tests/test_server.py
index 8e6717e..4c010a9 100644
--- a/libagent/tests/test_server.py
+++ b/libagent/tests/test_server.py
@@ -1,4 +1,3 @@
-import functools
 import io
 import os
 import socket
@@ -74,26 +73,27 @@ def test_handle():
 
 
 def test_server_thread():
-    connections = [FakeSocket()]
+    sock = FakeSocket()
+    connections = [sock]
     quit_event = threading.Event()
 
     class FakeServer(object):
         def accept(self):  # pylint: disable=no-self-use
             if connections:
                 return connections.pop(), 'address'
-            quit_event.set()
             raise socket.timeout()
 
         def getsockname(self):  # pylint: disable=no-self-use
             return 'fake_server'
 
-    handler = protocol.Handler(conn=empty_device()),
-    handle_conn = functools.partial(server.handle_connection,
-                                    handler=handler,
-                                    mutex=None)
+    def handle_conn(conn):
+        assert conn is sock
+        quit_event.set()
+
     server.server_thread(sock=FakeServer(),
                          handle_conn=handle_conn,
                          quit_event=quit_event)
+    quit_event.wait()
 
 
 def test_spawn():