def test_by_dataset_id_and_empty_col_ids_should_query_correctly(
self, mocked):
return_val = [[1], [2], [3]]
mocked.query.filter \
.return_value.all.return_value = return_val
service = DatasetColumnService()
metadata_id = 123
column_ids = None
actual = service.columns_by_id_and_dataset_id(metadata_id, column_ids)
mocked.query.filter.assert_called_with(
mocked._metadata_id == metadata_id)
self.assertEqual(actual, return_val)
def test_column_by_metadata_id_and_names_should_query_correctly(
self, mocked):
mocked.query.with_entities.return_value.filter \
.return_value.all.return_value = [[1], [2], [3]]
service = DatasetColumnService()
metadata_id = 123
column_names = ["a", "b"]
actual = service.columns_by_dataset_id_and_names(
metadata_id, column_names)
mocked.query.with_entities.assert_called_with(mocked._id)
mocked.query.with_entities.return_value.filter.assert_called_with(
mocked._metadata_id == metadata_id, mocked._name.in_(column_names))
expected = [1, 2, 3]
self.assertEqual(actual, expected)
def __init__(self):
self._dataset_service = DatasetService()
self._column_service = DatasetColumnService()
self._udf_service = UdfService()
self._udf_io_service = UdfIOService()
def test_failure_delete_columns(self):
mocks = [MagicMock() for i in range(1)]
mocks[0].side_effect = Exception()
service = DatasetColumnService()
actual = service.delete_column(mocks)
self.assertEqual(actual, None)
def test_delete_all_columns(self):
mocks = [MagicMock() for i in range(5)]
service = DatasetColumnService()
service.delete_column(mocks)
for mock in mocks:
mock.delete.assert_called_once()
def test_create_should_create_all_columns(self):
mocks = [MagicMock() for i in range(5)]
service = DatasetColumnService()
service.create_column(mocks)
for mock in mocks:
mock.save.assert_called_once()