From 73266ab5ea4655105cb40f36b64b4c12b2d08923 Mon Sep 17 00:00:00 2001 From: Sheogorath Date: Fri, 12 Jul 2019 00:06:53 +0200 Subject: [PATCH] Add RFC1459 4.1.1 password message In order to allow the server to handle password message properly, it should take care of the correct parameters being passed to the function. Tokenizer is still missing, but should be fine for basic functionality. --- src/server.py | 3 +++ test/test_server.py | 7 +++++++ 2 files changed, 10 insertions(+) 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