Skip to content
Snippets Groups Projects

WS-474: REST calls and CLI for calibration ingestion

Merged Charlotte Hausman requested to merge Calibration_Ingestion into main
Files
13
@@ -5,6 +5,8 @@ import argparse
from unittest.mock import patch, MagicMock
import conveyor.conveyor as con
from conveyor.deliver import DeliveryConveyor
from conveyor.retrieve import RetrievalConveyor
expected_settings = {
"qa_delivery_area": "/lustre/aoc/cluster/pipeline/docker/workspaces/qa2",
@@ -24,17 +26,14 @@ class TestConveyor:
settings = con._get_settings("test/test.json")
assert settings["qa_delivery_area"] == expected_settings["qa_delivery_area"]
assert settings["weblog_cache_area"] == expected_settings["weblog_cache_area"]
assert (
settings["workspaces_lustre_root_dir"]
== expected_settings["workspaces_lustre_root_dir"]
)
assert settings["workspaces_lustre_root_dir"] == expected_settings["workspaces_lustre_root_dir"]
# mock calls to cwd and count
assert cwd.call_count == 2
assert settings["destination_dir"] == expected_settings["destination_dir"]
assert settings["destination_subdir"] == expected_settings["destination_subdir"]
def test_main_deliver(self):
args.deliver = ["test/test.json"]
args.deliver = ['test/test.json']
with patch("argparse.ArgumentParser.parse_args", MagicMock(return_value=args)) as mock_args:
assert args.deliver[0] == "test/test.json"
@@ -46,7 +45,7 @@ class TestConveyor:
args.deliver = None
def test_main_retrieve(self):
args.retrieve = ["test/test.json"]
args.retrieve = ['test/test.json']
with patch("argparse.ArgumentParser.parse_args", MagicMock(return_value=args)) as mock_args:
with patch("conveyor.retrieve.RetrievalConveyor.convey") as ret_convey:
Loading