def test_remove_dataset_backends__one_backend(self, mock_api_remove_dataset_backend):
fake_dataset_id = 'fake-dataset-id'
fake_backend_id = 'fake-backend-id'
fake_passed_args = FakeArgs(host='fake-host', user='******')
fake_args = FakeArgs(dataset_id=fake_dataset_id, backend_ids=[fake_backend_id], all=False, force=False, **vars(fake_passed_args))
cli.remove_dataset_backends(fake_args)
mock_api_remove_dataset_backend.assert_called_once_with(fake_dataset_id, fake_backend_id, **vars(fake_passed_args))
def test_remove_dataset_backends__remove_all_answer_no(self, mock_api_remove_dataset_backend, mock_api_list_dataset_backends, mock_input):
fake_dataset_id = 'fake-dataset-id'
fake_passed_args = FakeArgs(host='fake-host', user='******')
fake_args = FakeArgs(dataset_id=fake_dataset_id, backend_ids=[], all=True, force=False, **vars(fake_passed_args))
cli.remove_dataset_backends(fake_args)
mock_api_remove_dataset_backend.assert_not_called()
mock_api_list_dataset_backends.assert_called_once_with(fake_dataset_id, **vars(fake_passed_args))
def test_remove_dataset_backends__backend_list(self, mock_api_remove_dataset_backend):
fake_dataset_id = 'fake-dataset-id'
fake_backend_ids = ['fake-backend-id-1', 'fake-backend-id-2', 'fake-backend-id-3']
fake_passed_args = FakeArgs(host='fake-host', user='******')
fake_args = FakeArgs(dataset_id=fake_dataset_id, backend_ids=fake_backend_ids, all=False, force=False, **vars(fake_passed_args))
cli.remove_dataset_backends(fake_args)
expected_calls = []
for idx, id in enumerate(fake_backend_ids):
expected_calls.append(mock.call(fake_dataset_id, fake_backend_ids[idx], **vars(fake_passed_args)))
mock_api_remove_dataset_backend.assert_has_calls(expected_calls)
def test_remove_dataset_backends__remove_all_forced(self, mock_api_remove_dataset_backend, mock_api_list_dataset_backends):
fake_dataset_id = 'fake-dataset-id'
fake_passed_args = FakeArgs(host='fake-host', user='******')
fake_args = FakeArgs(dataset_id=fake_dataset_id, backend_ids=[], all=True, force=True, **vars(fake_passed_args))
cli.remove_dataset_backends(fake_args)
expected_calls = []
for idx, id in enumerate(listed_backends):
expected_calls.append(mock.call(fake_dataset_id, listed_backends[idx], **vars(fake_passed_args)))
mock_api_remove_dataset_backend.assert_has_calls(expected_calls)
mock_api_list_dataset_backends.assert_called_once_with(fake_dataset_id, **vars(fake_passed_args))
def test_remove_dataset_backends__raises_value_error(self):
fake_dataset_id = 'fake-dataset-id'
fake_passed_args = FakeArgs(host='fake-host', user='******')
fake_args = FakeArgs(dataset_id=fake_dataset_id, backend_ids=[], all=False, force=False, **vars(fake_passed_args))
with self.assertRaisesRegex(ValueError, 'You must provide a list of backend IDs or pass --all'):
cli.remove_dataset_backends(fake_args)