def mock_api(cls): api = AuroraClientAPI(Cluster(name="foo"), 'test-client') mock_proxy = create_autospec(spec=SchedulerProxyApiSpec, spec_set=True, instance=True) api._scheduler_proxy = mock_proxy return api, mock_proxy
def test_handles_api_auth_error(): context = AuroraCommandContext() mock_scheduler_proxy = mock.create_autospec(spec=SchedulerProxyApiSpec, instance=True) mock_scheduler_proxy.killTasks.side_effect = SchedulerProxy.AuthError() mock_api = AuroraClientAPI(TEST_CLUSTER, "user-agent") mock_api._scheduler_proxy = mock_scheduler_proxy context.apis = {TEST_CLUSTER.name: mock_api} api = context.get_api(TEST_CLUSTER.name, clusters={TEST_CLUSTER.name: TEST_CLUSTER}) with pytest.raises(Context.CommandError) as e: api.kill_job(AuroraJobKey(TEST_CLUSTER.name, "role", "env", "job")) assert e.value.code == EXIT_AUTH_ERROR assert mock_scheduler_proxy.killTasks.call_count == 1
def test_handles_api_auth_error(): context = AuroraCommandContext() mock_scheduler_proxy = mock.create_autospec(spec=SchedulerProxyApiSpec, instance=True) mock_scheduler_proxy.killTasks.side_effect = SchedulerProxy.AuthError() mock_api = AuroraClientAPI(TEST_CLUSTER, 'user-agent') mock_api._scheduler_proxy = mock_scheduler_proxy context.apis = {TEST_CLUSTER.name: mock_api} api = context.get_api(TEST_CLUSTER.name, clusters={TEST_CLUSTER.name: TEST_CLUSTER}) with pytest.raises(Context.CommandError) as e: api.kill_job(AuroraJobKey(TEST_CLUSTER.name, 'role', 'env', 'job')) assert e.value.code == EXIT_AUTH_ERROR assert mock_scheduler_proxy.killTasks.call_count == 1
def mock_api(cls): api = AuroraClientAPI(Cluster(name="foo")) mock_proxy = Mock() api._scheduler_proxy = mock_proxy return api, mock_proxy