fix test imports

haliphax · Feb 28, 2023 · 6565b48 · 6565b48
1 parent 96fa05c
commit 6565b48
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/ssh/test_start_server.py b/tests/ssh/test_start_server.py
@@ -7,6 +7,7 @@
 # target
 from xthulu.ssh import start_server
 from xthulu.ssh.server import SSHServer
+from xthulu.ssh.process_factory import handle_client
 
 # local
 from tests import run_coroutine
@@ -55,7 +56,7 @@ def test_server_args(self):
                 "port": int(ssh_config["port"]),
                 "server_factory": SSHServer,
                 "server_host_keys": ssh_config["host_keys"],
-                "process_factory": SSHServer.handle_client,
+                "process_factory": handle_client,
                 "encoding": None,
             }
         )