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