diff --git a/src/server.py b/src/server.py index 3bc1a77..58447d2 100644 --- a/src/server.py +++ b/src/server.py @@ -21,6 +21,9 @@ class Handler_TCPServer(socketserver.BaseRequestHandler): def ping(self, param): return "PONG" + def command_pass(self, param): + return param == self.password + if __name__ == "__main__": HOST, PORT = "localhost", 9999 diff --git a/test/test_server.py b/test/test_server.py index e569018..57d1c53 100644 --- a/test/test_server.py +++ b/test/test_server.py @@ -6,3 +6,10 @@ def test_ping(server): assert server.ping(server, "some message") == "PONG" +def test_command_pass(server): + server.password = "password" + assert server.command_pass(server, "password") == True + +def test_command_pass_wrong(server): + server.password = "password" + assert server.command_pass(server, "wrong password") == False