matlab-proxy 0.15.0__py3-none-any.whl → 0.16.0__py3-none-any.whl

tests/unit/test_app_state.py

@@ -0,0 +1,586 @@
+# Copyright 2023-2024 The MathWorks, Inc.
+
+import json
+import os
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Optional
+
+import pytest
+
+from matlab_proxy import settings
+from matlab_proxy.app_state import AppState
+from matlab_proxy.constants import MWI_AUTH_TOKEN_NAME_FOR_HTTP
+from matlab_proxy.util.mwi.exceptions import LicensingError, MatlabError
+from tests.unit.util import MockResponse
+
+
+@pytest.fixture
+def sample_settings_fixture(tmp_path):
+    """A pytest fixture which returns a dict containing sample settings for the AppState class.
+
+    Args:
+        tmp_path : Builtin pytest fixture
+
+    Returns:
+        dict: A dictionary of sample settings
+    """
+    tmp_file = tmp_path / "parent_1" / "parent_2" / "tmp_file.json"
+    return {
+        "error": None,
+        "warnings": [],
+        "matlab_config_file": tmp_file,
+        "is_xvfb_available": True,
+        "mwi_server_url": "dummy",
+        "mwi_logs_root_dir": Path(settings.get_mwi_config_folder(dev=True)),
+        "app_port": 12345,
+        "mwapikey": "asdf",
+    }
+
+
+@pytest.fixture
+def app_state_fixture(sample_settings_fixture):
+    """A pytest fixture which returns an instance of AppState class with no errors.
+
+    Args:
+        sample_settings_fixture (dict): A dictionary of sample settings to be used by
+
+    Returns:
+        AppState: An object of the AppState class
+    """
+    app_state = AppState(settings=sample_settings_fixture)
+    app_state.processes = {"matlab": None, "xvfb": None}
+    app_state.licensing = {"type": "existing_license"}
+    return app_state
+
+
+@pytest.fixture
+def sample_token_headers_fixture():
+    return {MWI_AUTH_TOKEN_NAME_FOR_HTTP: "asdf"}
+
+
+@pytest.fixture
+def app_state_with_token_auth_fixture(
+    app_state_fixture, sample_token_headers_fixture, tmp_path
+):
+    """Pytest fixture which returns AppState instance with token authentication enabled.
+
+    Args:
+        app_state_fixture (AppState): Pytest fixture
+        tmp_path (str): Built-in pytest fixture
+
+    Returns:
+        (AppState, dict): Instance of the AppState class with token authentication enabled and token headers
+    """
+    tmp_matlab_ready_file = Path(tmp_path) / "tmp_file.txt"
+    tmp_matlab_ready_file.touch()
+    ((mwi_auth_token_name, mwi_auth_token_hash),) = sample_token_headers_fixture.items()
+    app_state_fixture.matlab_session_files["matlab_ready_file"] = tmp_matlab_ready_file
+    app_state_fixture.settings["mwi_is_token_auth_enabled"] = True
+    app_state_fixture.settings["mwi_auth_token_name_for_env"] = mwi_auth_token_name
+    app_state_fixture.settings["mwi_auth_token_name_for_http"] = (
+        MWI_AUTH_TOKEN_NAME_FOR_HTTP
+    )
+    app_state_fixture.settings["mwi_auth_token_hash"] = mwi_auth_token_hash
+    app_state_fixture.settings["mwi_server_url"] = "http://localhost:8888"
+
+    return app_state_fixture
+
+
+@pytest.fixture
+def mocker_os_patching_fixture(mocker, platform):
+    """A pytest fixture which patches the is_* functions in system.py module
+
+    Args:
+        mocker : Built in pytest fixture
+        platform (str): A string representing "windows", "linux" or "mac"
+
+    Returns:
+        mocker: Built in pytest fixture with patched calls to system.py module.
+    """
+    mocker.patch("matlab_proxy.app_state.system.is_linux", return_value=False)
+    mocker.patch("matlab_proxy.app_state.system.is_windows", return_value=False)
+    mocker.patch("matlab_proxy.app_state.system.is_mac", return_value=False)
+    mocker.patch("matlab_proxy.app_state.system.is_posix", return_value=False)
+
+    if platform == "linux":
+        mocker.patch("matlab_proxy.app_state.system.is_linux", return_value=True)
+        mocker.patch("matlab_proxy.app_state.system.is_posix", return_value=True)
+
+    elif platform == "windows":
+        mocker.patch("matlab_proxy.app_state.system.is_windows", return_value=True)
+        mocker.patch("matlab_proxy.app_state.system.is_posix", return_value=False)
+
+    else:
+        mocker.patch("matlab_proxy.app_state.system.is_mac", return_value=True)
+        mocker.patch("matlab_proxy.app_state.system.is_posix", return_value=True)
+
+    return mocker
+
+
+@dataclass(frozen=True)
+class Mock_xvfb:
+    """An immutable dataclass representing a mocked Xvfb process"""
+
+    returncode: Optional[int]
+    pid: Optional[int]
+
+
+@dataclass(frozen=True)
+class Mock_matlab:
+    """An immutable dataclass representing a mocked MATLAB process"""
+
+    returncode: Optional[int]
+    pid: Optional[int]
+
+
+@pytest.mark.parametrize(
+    "licensing, expected",
+    [
+        (None, False),
+        ({"type": "nlm", "conn_str": "123@host"}, True),
+        ({"type": "nlm"}, False),
+        ({"type": "mhlm", "identity_token": "random_token"}, False),
+        (
+            {
+                "type": "mhlm",
+                "identity_token": "random_token",
+                "source_id": "dummy_id",
+                "expiry": "Jan 1, 1970",
+                "entitlement_id": "123456",
+            },
+            True,
+        ),
+        ({"type": "existing_license"}, True),
+        ({"type": "invalid_type"}, False),
+    ],
+    ids=[
+        "None licensing",
+        "happy path-nlm",
+        "incomplete nlm data",
+        "incomplete mhlm data",
+        "happy path-mhlm",
+        "happy path-existing license",
+        "invalid license",
+    ],
+)
+def test_is_licensed(app_state_fixture, licensing, expected):
+    """Test to check is_licensed()
+
+    Args:
+        app_state_fixture (AppState): Object of AppState class with defaults set
+        licensing (dict): Represents licensing information
+        expected (bool): Expected return value.
+    """
+    # Arrange
+    # Nothing to arrange
+
+    # Act
+    app_state_fixture.licensing = licensing
+
+    # Assert
+    assert app_state_fixture.is_licensed() == expected
+
+
+@pytest.mark.parametrize(
+    "err, expected_err",
+    [
+        (MatlabError(message="dummy error"), MatlabError(message="dummy")),
+        (LicensingError(message="license issue"), None),
+    ],
+    ids=["Any error except licensing error", "licensing error"],
+)
+def test_unset_licensing(err, app_state_fixture, expected_err):
+    """Test to check unset_liecnsing removes licensing from the AppState object
+
+    Args:
+        err (Exception): Custom exceptions defined in exceptions.py
+        licensing (bool): Whether licensing info is removed
+        expected_err (Exception): Expected exception
+    """
+    # Arrange
+    app_state_fixture.error = err
+
+    # Act
+    app_state_fixture.unset_licensing()
+
+    # Assert
+    assert app_state_fixture.licensing == None
+    assert type(app_state_fixture.error) is type(expected_err)
+
+
+# config file is deleted when licensing info is not set i.e. set to None
+def test_persist_licensing_when_licensing_info_is_not_set(app_state_fixture):
+    """Test to check if data is not persisted to a file if licensing info is not present
+
+    Args:
+        tmp_path (Path): Built in pytest fixture
+    """
+    # Arrange
+    # Nothing to arrange
+    app_state_fixture.licensing = None
+
+    # Act
+    app_state_fixture.persist_config_data()
+
+    # Assert
+    assert os.path.exists(app_state_fixture.settings["matlab_config_file"]) is False
+
+
+@pytest.mark.parametrize(
+    "licensing_data",
+    [
+        ({"type": "nlm", "conn_str": "123@host"}),
+        (
+            {
+                "type": "mhlm",
+                "identity_token": "random_token",
+                "source_id": "dummy_id",
+                "expiry": "Jan 1, 1970",
+                "entitlement_id": "123456",
+            }
+        ),
+        ({"type": "existing_license"}),
+    ],
+    ids=["nlm type", "mhlm type", "existing license type"],
+)
+def test_persist_config_data(licensing_data: dict, tmp_path):
+    """Test to check if persist_licensing() writes data to the file system
+
+    Args:
+        data (dict): Represents matlab-proxy licensing data
+        tmp_path : Built-in pytest fixture.
+    """
+    # Arrange
+    tmp_file = tmp_path / "parent_1" / "parent_2" / "tmp_file.json"
+    settings = {
+        "matlab_config_file": tmp_file,
+        "error": None,
+        "matlab_version": None,
+        "warnings": [],
+    }
+    app_state = AppState(settings=settings)
+    app_state.licensing = licensing_data
+
+    cached_data = {"licensing": licensing_data, "matlab": {"version": None}}
+
+    # Act
+    app_state.persist_config_data()
+    with open(tmp_file, "r") as file:
+        got = file.read()
+
+    # Assert
+    assert json.loads(got) == cached_data
+
+
+validate_required_processes_test_data = [
+    (None, None, "linux", False),  # xvfb is None == True
+    (None, Mock_xvfb(None, 1), "linux", False),  # matlab is None == True
+    (
+        Mock_matlab(None, 1),
+        Mock_xvfb(None, 1),
+        "linux",
+        True,
+    ),  # All branches are skipped and nothing returned
+    (
+        Mock_matlab(None, 1),
+        Mock_xvfb(123, 2),
+        "linux",
+        False,
+    ),  # xvfb.returncode is not None == True
+    (
+        Mock_matlab(123, 1),
+        Mock_xvfb(None, 2),
+        "linux",
+        False,
+    ),  # matlab.returncode is not None == True
+    (
+        Mock_matlab(None, 1),
+        None,
+        "linux",
+        True,
+    ),  # Xvfb not found on path
+]
+
+
+@pytest.mark.parametrize(
+    "matlab, xvfb, platform, expected",
+    validate_required_processes_test_data,
+    ids=[
+        "processes_not_running",
+        "matlab_not_running",
+        "All_required_processes_running",
+        "All_processes_running_with_xvfb_returning_non_zero_code",
+        "All_processes_running_with_matlab_returning_non_zero_code",
+        "xvfb_is_optional_matlab_starts_without_it",
+    ],
+)
+def test_are_required_processes_ready(
+    app_state_fixture, mocker_os_patching_fixture, matlab, xvfb, expected
+):
+    """Test to check if required processes are ready
+
+    Args:
+        app_state_fixture (AppState): Object of AppState class with defaults set
+        mocker_os_patching_fixture (mocker): Custom pytest fixture for mocking
+        matlab (Mock_matlab): Represents a mocked MATLAB process
+        xvfb (Mock_xvfb): Represents a mocked Xvfb process
+        expected (bool): Expected return value based on process return code
+    """
+    # Arrange
+    app_state_fixture.processes = {"matlab": matlab, "xvfb": xvfb}
+    if not xvfb:
+        app_state_fixture.settings["is_xvfb_available"] = False
+
+    # Act
+    actual = app_state_fixture._are_required_processes_ready()
+
+    # Assert
+    assert actual == expected
+
+
+get_matlab_status_based_on_connector_status_test_data = [
+    ("up", True, "up"),
+    ("down", True, "starting"),
+    ("up", False, "starting"),
+]
+
+
+@pytest.mark.parametrize(
+    "connector_status, ready_file_present, matlab_status",
+    get_matlab_status_based_on_connector_status_test_data,
+    ids=["connector_up", "connector_down", "connector_up_ready_file_not_present"],
+)
+async def test_get_matlab_status_based_on_connector_status(
+    mocker, app_state_fixture, connector_status, ready_file_present, matlab_status
+):
+    """Test to check matlab status based on connector status
+
+    Args:
+        mocker : Built in pytest fixture.
+        connector_status (str): Status of Embedded Connector.
+        ready_file_present (bool): Represents if the ready file has been created or not.
+        matlab_status (str): Represents the status of MATLAB process.
+    """
+    # Arrange
+    mocker.patch(
+        "matlab_proxy.app_state.mwi.embedded_connector.request.get_state",
+        return_value=connector_status,
+    )
+    mocker.patch.object(Path, "exists", return_value=ready_file_present)
+    app_state_fixture.settings["mwi_is_token_auth_enabled"] = False
+    app_state_fixture.matlab_session_files["matlab_ready_file"] = Path("dummy")
+
+    # Act
+    actual_matlab_status = await app_state_fixture._get_matlab_connector_status()
+
+    # Assert
+    assert actual_matlab_status == matlab_status
+
+
+@pytest.mark.parametrize(
+    "valid_processes, connector_status, expected",
+    [
+        (True, "up", "up"),
+        (False, "up", "down"),
+        (True, "down", "down"),
+    ],
+    ids=[
+        "valid_processes_connector_up",
+        "invalid_processes_connector_up",
+        "valid_processes_connector_down",
+    ],
+)
+async def test_get_matlab_state(
+    app_state_fixture, mocker, valid_processes, connector_status, expected
+):
+    """Test to check get_matlab_state returns the correct MATLAB state based on the connector status
+
+    Args:
+        app_state_fixture (AppState): Object of AppState class with defaults set
+        mocker : Built in pytest fixture
+        valid_processes (bool): Represents if the processes are valid or not
+        connector_status (str): Status of Embedded Connector.
+        expected (str): Expected status of MATLAB process.
+    """
+    # Arrange
+    mocker.patch.object(
+        AppState,
+        "_are_required_processes_ready",
+        return_value=valid_processes,
+    )
+    mocker.patch.object(
+        AppState,
+        "_get_matlab_connector_status",
+        return_value=connector_status,
+    )
+
+    # Act
+    actual_state = await app_state_fixture.get_matlab_state()
+
+    # Assert
+    assert actual_state == expected
+
+
+@pytest.mark.parametrize("platform", [("linux"), ("windows"), ("mac")])
+async def test_track_embedded_connector(mocker_os_patching_fixture, app_state_fixture):
+    """Test to check track_embedded_connector task
+
+    Args:
+        mocker_os_patching_fixture (mocker): Custom pytest fixture for mocking
+        app_state_fixture (AppState): Object of AppState class with defaults set
+    """
+
+    # Arrange
+    # patching embedded_connector_start_time to EPOCH+1 seconds and state to be "down"
+    mocker_os_patching_fixture.patch.object(
+        app_state_fixture, "embedded_connector_start_time", new=float(1.0)
+    )
+    mocker_os_patching_fixture.patch.object(
+        app_state_fixture, "embedded_connector_state", return_value="down"
+    )
+
+    # verify that stop_matlab() is called once
+    spy = mocker_os_patching_fixture.spy(app_state_fixture, "stop_matlab")
+
+    # Act
+    await app_state_fixture._AppState__track_embedded_connector_state()
+
+    # Assert
+    spy.assert_called_once()
+
+
+@pytest.mark.parametrize(
+    "env_var_name, filter_prefix, is_filtered",
+    [("MWI_AUTH_TOKEN", "MWI_", None), ("MWIFOO_AUTH_TOKEN", "MWI_", "foo")],
+    ids=["env_var_is_filtered", "env_var_is_not_filtered"],
+)
+def test_env_variables_filtration_for_xvfb_process(
+    monkeypatch, env_var_name, filter_prefix, is_filtered
+):
+    """Test to check if __filter_env_variables filters environment variables with a certain prefix correctly.
+
+    Args:
+        monkeypatch (Object): Built-in pytest fixture for monkeypatching
+        env_var_name (str): Name of the environment variable
+        filter_prefix (str): Prefix to check for filtering
+        is_filtered (bool): To check if the env variable with specified prefix is filtered.
+    """
+    # Arrange
+    env_var = env_var_name
+    monkeypatch.setenv(env_var, "foo")
+
+    # Act
+    filtered_env_vars: dict = AppState._AppState__filter_env_variables(
+        os.environ, filter_prefix
+    )
+
+    # Assert
+    assert filtered_env_vars.get(env_var) == is_filtered
+
+
+@pytest.mark.parametrize(
+    "platform, expected_output",
+    [("linux", "stdout"), ("windows", "file"), ("mac", "stdout")],
+)
+async def test_setup_env_for_matlab(
+    mocker_os_patching_fixture, platform, expected_output, app_state_fixture, tmp_path
+):
+    """Test to check MW_DIAGNOSTIC_DEST is set appropriately for posix and non-posix systems
+
+    Args:
+        mocker_os_patching_fixture (mocker): Custom pytest fixture for mocking
+        platform (str): string describing a platform
+        app_state_fixture (AppState): Object of AppState class with defaults set
+        tmp_path (Path): Built-in pytest fixture for temporary paths
+    """
+
+    # Arrange
+    app_state_fixture.licensing = {"type": "existing_license"}
+    app_state_fixture.settings = {"mwapikey": None, "matlab_display": ":1"}
+    app_state_fixture.mwi_logs_dir = tmp_path
+    mocker_os_patching_fixture.patch(
+        "matlab_proxy.app_state.logger.isEnabledFor", return_value=True
+    )
+
+    # Act
+    matlab_env = await app_state_fixture._AppState__setup_env_for_matlab()
+
+    # Assert
+    assert expected_output in matlab_env["MW_DIAGNOSTIC_DEST"]
+
+
+@pytest.mark.parametrize(
+    "function_to_call ,mock_response",
+    [
+        ("_get_matlab_connector_status", MockResponse(ok=True)),
+        (
+            "_AppState__send_stop_request_to_matlab",
+            MockResponse(
+                ok=True, payload={"messages": {"EvalResponse": [{"isError": None}]}}
+            ),
+        ),
+    ],
+    ids=["request matlab connector status", "send request to stop matlab"],
+)
+async def test_requests_sent_by_matlab_proxy_have_headers(
+    app_state_with_token_auth_fixture,
+    function_to_call,
+    mock_response,
+    mocker,
+    sample_token_headers_fixture,
+):
+    """Test to check if token headers are included in requests sent by matlab-proxy when authentication is enabled
+
+    Args:
+        app_state_fixture_with_token_auth (AppState): Instance of AppState class with token authentication enabled
+        mocker : Built-in pytest fixture
+    """
+    # Arrange
+    mocked_request = mocker.patch(
+        "aiohttp.ClientSession.request", return_value=mock_response
+    )
+
+    # Act
+    # Call the function passed as a string
+    method = getattr(app_state_with_token_auth_fixture, function_to_call)
+    _ = await method()
+
+    # Assert
+    connector_status_request_headers = list(mocked_request.call_args_list)[0].kwargs[
+        "headers"
+    ]
+    assert sample_token_headers_fixture == connector_status_request_headers
+
+
+async def test_start_matlab_without_xvfb(app_state_fixture, mocker):
+    """Test to check if Matlab process starts without throwing errors when Xvfb is not present
+
+    Args:
+        app_state_fixture (AppState): Object of AppState class with defaults set
+        mocker : Built-in pytest fixture
+    """
+    # Arrange
+    app_state_fixture.settings["is_xvfb_available"] = False
+    mock_matlab = Mock_matlab(None, 1)
+
+    # Starting asyncio tasks related to matlab is not required here as only Xvfb check is required.
+    mocker.patch.object(
+        AppState, "_AppState__start_matlab_process", return_value=mock_matlab
+    )
+    mocker.patch.object(
+        AppState, "_AppState__matlab_stderr_reader_posix", return_value=None
+    )
+    mocker.patch.object(
+        AppState, "_AppState__track_embedded_connector_state", return_value=None
+    )
+    mocker.patch.object(AppState, "_AppState__update_matlab_port", return_value=None)
+
+    # Act
+    await app_state_fixture.start_matlab()
+
+    # Assert
+    # Check if Xvfb has not started
+    assert app_state_fixture.processes["xvfb"] is None
+    # Check if Matlab started
+    assert app_state_fixture.processes["matlab"] is mock_matlab

tests/unit/test_constants.py

@@ -0,0 +1,6 @@
+# Copyright 2023-2024 The MathWorks, Inc.
+
+TWO_MAX_TRIES = 2
+FIVE_MAX_TRIES = 5
+HALF_SECOND_DELAY = 0.5
+ONE_SECOND_DELAY = 1

tests/unit/test_ddux.py

@@ -0,0 +1,22 @@
+# Copyright 2020-2022 The MathWorks, Inc.
+
+import matlab_proxy
+from matlab_proxy import util
+
+
+def test_get_mwi_ddux_value():
+    """Tests ddux value for matlab-proxy with different extension names"""
+    expected_result = matlab_proxy.__get_matlab_proxy_base_ddux_value()
+    actual_result = matlab_proxy.get_mwi_ddux_value(
+        matlab_proxy.get_default_config_name()
+    )
+
+    assert expected_result == actual_result
+
+    expected_result = f"MATLAB_PROXY:HELLO_WORLD:V1"
+    actual_result = matlab_proxy.get_mwi_ddux_value("hello world")
+
+    assert expected_result == actual_result
+
+    actual_result = matlab_proxy.get_mwi_ddux_value(" \n \t hello-world  \n")
+    assert expected_result == actual_result