def test_cr_status_bad_instance_type():
with pytest.raises(RuntimeError) as excinfo:
pik.cr_status(
service="",
instance="",
verbose=0,
instance_type="marathon",
kube_client=mock.Mock(),
)
assert "Unknown instance type" in str(excinfo.value)
def test_cr_status_happy_path(mock_get_cr):
mock_status = mock.Mock()
mock_metadata = mock.Mock()
mock_return = dict(status=mock_status, metadata=mock_metadata)
mock_get_cr.return_value = mock_return
status = pik.cr_status(
service="",
instance="",
verbose=0,
instance_type="flink",
kube_client=mock.Mock(),
)
assert status == mock_return