Skip to content
Snippets Groups Projects
Commit fcca4d31 authored by Daniel Nemergut's avatar Daniel Nemergut
Browse files

Cleanup from pre-commit hook

parent 0d55d3a4
No related branches found
No related tags found
2 merge requests!1706merge 2.8.4 to main,!1670WS-1405 CASA matrix service
......@@ -17,19 +17,16 @@
# along with Workspaces. If not, see <https://www.gnu.org/licenses/>.
""" Tests for CASA Matrix Service """
# pylint: disable=C0103, E0401, R0201, W0621
import http
import json
import pytest
import pytest
from pycapo import CapoConfig
from sqlalchemy import create_engine
from sqlalchemy.orm import Session
from unittest.mock import MagicMock, patch
from workspaces.system.services.casa_matrix_service import CasaMatrixService
from workspaces.capability.schema import CasaMatrixCapability, CasaMatrixCasaVersion
from workspaces.capability.schema import CasaMatrixCasaVersion
from workspaces.capability.schema import mapper_registry as capability_mapper_registry
from workspaces.system.services.casa_matrix_service import CasaMatrixService
settings = CapoConfig().settings("edu.nrao.archive.workflow.config.CasaVersions")
......@@ -44,20 +41,18 @@ ROUTER_PATCH = "workspaces.system.services.casa_matrix_service.Router"
@pytest.fixture(scope="module")
def fake_request(casa_version: str, capability: str, telescope: str) -> json:
return json.dumps({
"casa_version": casa_version,
'capability': capability,
'telescope': telescope
})
return json.dumps({"casa_version": casa_version, "capability": capability, "telescope": telescope})
@pytest.fixture(scope="module")
def fake_version() -> CasaMatrixCasaVersion:
return CasaMatrixCasaVersion.from_json({
"casa_version": "1.2.3-4|default",
"capabilities": ["restore_cms", "std_calibration"],
"is_cluster_compatible": True,
})
return CasaMatrixCasaVersion.from_json(
{
"casa_version": "1.2.3-4|default",
"capabilities": ["restore_cms", "std_calibration"],
"is_cluster_compatible": True,
}
)
@pytest.fixture(scope="module")
......@@ -87,26 +82,20 @@ class TestCasaMatrixService:
ret_version = mock_matrix_service.get_version(None, fake_request["capability"])
assert ret_version["version"] == expected_version
@pytest.mark.parametrize("fake_request", [
{"5.6.1-8|default", "restore_cms", "VLA"},
{"5.6.1-8|default", "restore_basic_cms", "VLA"}
])
@pytest.mark.parametrize(
"fake_request", [{"5.6.1-8|default", "restore_cms", "VLA"}, {"5.6.1-8|default", "restore_basic_cms", "VLA"}]
)
def test_vla_default_version(self, mock_matrix_service: CasaMatrixService, fake_request: dict[str, str, str]):
expected_version = VLA_DEFAULT_CASA
ret_version = mock_matrix_service.get_version(
None, fake_request["capability"], fake_request["telescope"]
)
ret_version = mock_matrix_service.get_version(None, fake_request["capability"], fake_request["telescope"])
assert ret_version["version"] == expected_version
@pytest.mark.parametrize("fake_request", [
{"5.6.1-8|default", "restore_cms", "ALMA"},
{"5.6.1-8|default", "restore_basic_cms", "ALMA"}
])
@pytest.mark.parametrize(
"fake_request", [{"5.6.1-8|default", "restore_cms", "ALMA"}, {"5.6.1-8|default", "restore_basic_cms", "ALMA"}]
)
def test_alma_default_version(self, mock_matrix_service: CasaMatrixService, fake_request: dict[str, str, str]):
expected_version = ALMA_DEFAULT_CASA
ret_version = mock_matrix_service.get_version(
None, fake_request["capability"], fake_request["telescope"]
)
ret_version = mock_matrix_service.get_version(None, fake_request["capability"], fake_request["telescope"])
assert ret_version["version"] == expected_version
@pytest.mark.parametrize("fake_request", [{"5.6.1-8|default", None, None}, {"6.1.1-15|2020.1.0.40", None, None}])
......@@ -121,10 +110,9 @@ class TestCasaMatrixService:
ret_version = mock_matrix_service.get_version(fake_request["casa_version"])
assert ret_version["version"] == expected_version
@pytest.mark.parametrize("fake_request", [
{"0.0.0-0|default", "restore_cms", "VLA"},
{"x.x.x-x|default", "restore_basic_cms", "VLA"}
])
@pytest.mark.parametrize(
"fake_request", [{"0.0.0-0|default", "restore_cms", "VLA"}, {"x.x.x-x|default", "restore_basic_cms", "VLA"}]
)
def test_vla_invalid_version(self, mock_matrix_service: CasaMatrixService, fake_request: dict[str, str, str]):
expected_version = VLA_DEFAULT_CASA
ret_version = mock_matrix_service.get_version(
......@@ -132,10 +120,9 @@ class TestCasaMatrixService:
)
assert ret_version["version"] == expected_version
@pytest.mark.parametrize("fake_request", [
{"0.0.0-0|default", "restore_cms", "ALMA"},
{"x.x.x-x|default", "restore_basic_cms", "ALMA"}
])
@pytest.mark.parametrize(
"fake_request", [{"0.0.0-0|default", "restore_cms", "ALMA"}, {"x.x.x-x|default", "restore_basic_cms", "ALMA"}]
)
def test_alma_invalid_version(self, mock_matrix_service: CasaMatrixService, fake_request: dict[str, str, str]):
expected_version = ALMA_DEFAULT_CASA
ret_version = mock_matrix_service.get_version(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment