From 748a19e546cdebe6c9d9d08a8fb7ab7c0b17dfcf Mon Sep 17 00:00:00 2001 From: Matteo Cencini Date: Thu, 31 Oct 2024 13:41:33 +0100 Subject: [PATCH] bug fix in server --- src/pulserver/_server/_server.py | 2 +- tests/_server/test_server.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/pulserver/_server/_server.py b/src/pulserver/_server/_server.py index b01d10b..f03dad5 100644 --- a/src/pulserver/_server/_server.py +++ b/src/pulserver/_server/_server.py @@ -238,7 +238,7 @@ def handle_client_connection(config, client_socket, plugins, logger): logger.info(f"Calling {function_name} with args {kwargs}") # Run design function - result_buffer, optional_buffer = function(**kwargs) + result_buffer, optional_buffer = function(kwargs) # Log the output to the function-specific log file function_logger.info(f"Output buffer: {result_buffer}") diff --git a/tests/_server/test_server.py b/tests/_server/test_server.py index 4b16cea..0d9ec74 100644 --- a/tests/_server/test_server.py +++ b/tests/_server/test_server.py @@ -345,7 +345,7 @@ def test_handle_client_connection_valid_function( handle_client_connection(sample_config, client_socket, plugins, logger) # Check that the function was called correctly - plugins["foo"].assert_called_once_with(FOVx=256.0, Nx=128) + plugins["foo"].assert_called_once_with({"FOVx": 256.0, "Nx": 128}) # Check that logging was done correctly logger.info.assert_called_once_with( @@ -403,7 +403,7 @@ def test_handle_client_connection_no_optional_buffer( logger = main_logger # Mock plugins with an example function that returns no optional buffer - def mock_plugin(**kwargs): + def mock_plugin(kwargs): return b"result_buffer", None plugins["foo"] = MagicMock(side_effect=mock_plugin) @@ -420,7 +420,7 @@ def mock_plugin(**kwargs): handle_client_connection(sample_config, client_socket, plugins, logger) # Check that the function was called correctly - plugins["foo"].assert_called_once_with(FOVx=256.0, Nx=128) + plugins["foo"].assert_called_once_with({"FOVx": 256.0, "Nx": 128}) # Check that logging was done correctly logger.info.assert_called_once_with(