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()