def setUp(self): self._service_under_test = EditDataService() self._mock_connection = mock.MagicMock() self._service_provider = ServiceProviderMock({ 'query_execution': {}, 'connection': self._mock_connection }) self.cursor = utils.MockCursor(None) self.connection = utils.MockPsycopgConnection(cursor=self.cursor) self.cursor.connection = self.connection self.connection_service = ConnectionService() self.connection_service.get_connection = mock.Mock( return_value=self.connection) self.query_execution_service = QueryExecutionService() self._service_provider._services = { constants.CONNECTION_SERVICE_NAME: self.connection_service, constants.QUERY_EXECUTION_SERVICE_NAME: self.query_execution_service } self._service_provider._is_initialized = True self._service_under_test.register(self._service_provider) # self._connection = MockConnection({"port": "8080", "host": "test", "dbname": "test"}) self._initialize_edit_request = InitializeEditParams() self._initialize_edit_request.schema_name = 'public' self._initialize_edit_request.object_name = 'Employee' self._initialize_edit_request.object_type = 'Table' self._initialize_edit_request.owner_uri = 'testuri'
def setUp(self): """Set up the test by creating a query with multiple batches""" self.statement_list = statement_list = [ 'select version;', 'select * from t1;' ] self.statement_str = ''.join(statement_list) self.query_uri = 'test_uri' self.query = Query( self.query_uri, self.statement_str, QueryExecutionSettings(ExecutionPlanOptions(), ResultSetStorageType.FILE_STORAGE), QueryEvents()) self.mock_query_results = [('Id1', 'Value1'), ('Id2', 'Value2')] self.cursor = utils.MockCursor(self.mock_query_results) self.connection = utils.MockConnection(cursor=self.cursor) self.columns_info = [] db_column_id = DbColumn() db_column_id.data_type = 'text' db_column_id.column_name = 'Id' db_column_value = DbColumn() db_column_value.data_type = 'text' db_column_value.column_name = 'Value' self.columns_info = [db_column_id, db_column_value] self.get_columns_info_mock = mock.Mock(return_value=self.columns_info)
def setUp(self): """Set up mock objects for testing the query execution service. Ran before each unit test. """ # set up mock connection self.rows = [(1, 'Text 1'), (2, 'Text 2')] self.cursor = utils.MockCursor(self.rows) self.mock_pymysql_connection = utils.MockPyMySQLConnection( parameters={ 'host': 'test', 'dbname': 'test', }) self.connection = MockMySQLServerConnection() self.connection.cursor.return_value = self.cursor self.cursor.connection = self.connection self.connection_service = ConnectionService() self.request_context = utils.MockRequestContext() # setup mock query_execution_service self.query_execution_service = QueryExecutionService() self.service_provider = ServiceProvider(None, {}, constants.MYSQL_PROVIDER_NAME) self.service_provider._services = { constants.CONNECTION_SERVICE_NAME: self.connection_service } self.service_provider._is_initialized = True self.query_execution_service._service_provider = self.service_provider def connection_side_effect(owner_uri: str, connection_type: ConnectionType): return self.connection self.connection_service.get_connection = mock.Mock( side_effect=connection_side_effect)
def setUp(self): self._cursor = utils.MockCursor(None) self._connection = utils.MockConnection(cursor=self._cursor) self._batch_text = 'Select * from t1' self._batch_id = 1 self._batch_events = BatchEvents() self._selection_data = SelectionData() self._result_set = mock.MagicMock()
def setUp(self): self._rows = [(1, 'Some text 1', 'Some valid xml', b'Hello bytes1'), (2, 'Some Text 2', 'Some Valid xml', b'Hello bytes2')] self._cursor = utils.MockCursor(self._rows) self._columns_info = [] self._get_columns_info_mock = mock.Mock( return_value=self._columns_info) self._reader = StorageDataReader(self._cursor)
def setUp(self): self._id = 1 self._batch_id = 1 self._events = ResultSetEvents() self._first_row = tuple([1, 2, 3]) self._second_row = tuple([5, 6, 7]) self._cursor = utils.MockCursor([self._first_row, self._second_row]) self._result_set = InMemoryResultSet(self._id, self._batch_id, self._events)
def setUp(self): self._id = 1 self._batch_id = 1 self._events = ResultSetEvents() self._bytes_to_write = 10 self._writer = MockWriter(self._bytes_to_write) self._row: List[DbCellValue] = ['Column_Val1', 'Column_Val2'] self._reader = MockReader(self._row) self._file = 'TestFile' self._cursor = utils.MockCursor([tuple([1, 2, 3]), tuple([5, 6, 7])]) self._result_set = None
def setUp(self): self._rows = [( 1, 'int4', ), (2, 'bool')] self._cursor = utils.MockCursor(self._rows) column = namedtuple('Column', [ 'name', 'type_code', 'display_size', 'internal_size', 'precision', 'scale', 'null_ok' ]) self._cursor.description = [ column('id', 1, None, None, None, None, True), column('is_valid', 2, None, None, None, None, True) ] self._connection = utils.MockConnection(cursor=self._cursor)
def setUp(self): self._file_stream = io.BytesIO() self._writer = ServiceBufferFileStreamWriter(self._file_stream) self._cursor = utils.MockCursor([tuple([11, 22, 33]), tuple([55, 66, 77])])