From b0a7588135b38514501d294e7f256eebcba953c9 Mon Sep 17 00:00:00 2001 From: Wayne Date: Fri, 12 Jul 2019 00:22:33 +0200 Subject: [PATCH] [refactor] parametrized commnand_passwd function --- test/test_server.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/test/test_server.py b/test/test_server.py index 57d1c53..36f4539 100644 --- a/test/test_server.py +++ b/test/test_server.py @@ -1,15 +1,14 @@ """ tests the server functionality """ +import pytest 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): +@pytest.mark.parametrize("test_input,expected", [("password", True), ("wrong password", False)]) +def test_command_pass(server, test_input, expected): server.password = "password" - assert server.command_pass(server, "wrong password") == False + assert server.command_pass(server, test_input) == expected