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