Update test.py

zhewenshen · Jul 20, 2024 · 2a16f8e · 2a16f8e
1 parent 303e001
commit 2a16f8e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/test.py b/src/test.py
@@ -10,7 +10,12 @@
 
 @pytest.fixture
 def client_server():
-    return Client(), Server()
+    client = Client()
+    server = Server()
+    key = Fernet.generate_key();
+    client.set_key(key)
+    server.set_key(key)
+    return client, server
 
 
 def print_server_response(message):