def test_add_run_failure(mock_k8s_run_api_client: CustomObjectsApi): mock_k8s_run_api_client.create_namespaced_custom_object.side_effect = ApiException(status=500) run = Run(name=RUN_NAME, experiment_name='fake') with pytest.raises(ApiException): run.create(namespace=NAMESPACE)
def test_add_run(mock_k8s_run_api_client: CustomObjectsApi): mock_k8s_run_api_client.create_namespaced_custom_object.return_value = GET_RUN_RESPONSE_RAW run = Run(name=RUN_NAME, experiment_name='fake') added_run = run.create(namespace=NAMESPACE) assert added_run is not None and type(added_run) is KubernetesObject