def test_add_run_type_arg(self, m_worker):
m_worker().cli = Mock(name="cli")
upload_fvcom_atmos_forcing.main()
args, kwargs = m_worker().cli.add_argument.call_args_list[2]
assert args == ("run_type", )
assert kwargs["choices"] == {"nowcast", "forecast"}
assert "help" in kwargs
def test_add_run_date_option(self, m_worker):
m_worker().cli = Mock(name="cli")
upload_fvcom_atmos_forcing.main()
args, kwargs = m_worker().cli.add_date_option.call_args_list[0]
assert args == ("--run-date", )
assert kwargs["default"] == arrow.now().floor("day")
assert "help" in kwargs
def test_add_model_config_arg(self, m_worker):
m_worker().cli = Mock(name="cli")
upload_fvcom_atmos_forcing.main()
args, kwargs = m_worker().cli.add_argument.call_args_list[1]
assert args == ("model_config", )
assert kwargs["choices"] == {"r12", "x2"}
assert "help" in kwargs
def test_run_worker(self, m_worker):
m_worker().cli = Mock(name="cli")
upload_fvcom_atmos_forcing.main()
args, kwargs = m_worker().run.call_args
assert args == (
upload_fvcom_atmos_forcing.upload_fvcom_atmos_forcing,
upload_fvcom_atmos_forcing.success,
upload_fvcom_atmos_forcing.failure,
)
def test_add_host_name_arg(self, m_worker):
m_worker().cli = Mock(name="cli")
upload_fvcom_atmos_forcing.main()
args, kwargs = m_worker().cli.add_argument.call_args_list[0]
assert args == ("host_name", )
assert "help" in kwargs
def test_init_cli(self, m_worker):
m_worker().cli = Mock(name="cli")
upload_fvcom_atmos_forcing.main()
m_worker().init_cli.assert_called_once_with()
def test_instantiate_worker(self, m_worker):
m_worker().cli = Mock(name="cli")
upload_fvcom_atmos_forcing.main()
args, kwargs = m_worker.call_args
assert args == ("upload_fvcom_atmos_forcing", )
assert list(kwargs.keys()) == ["description"]