def test_query_subset_response_AdventureWorks2014(self):
"""
Test the retrieval of the actual rows for "select * from HumanResources.Department"
"""
with open(self.get_baseline(\
u'test_simple_query.txt'),
u'r+b',
buffering=0) as response_file:
request_stream = io.BytesIO()
rpc_client = json_rpc_client.JsonRpcClient(request_stream,
response_file)
rpc_client.start()
# Submit a dummy request.
owner_uri = u'connectionservicetest'
parameters = {
u'OwnerUri': u'connectionservicetest',
u'BatchIndex': 0,
u'ResultSetIndex': 0,
u'RowsStartIndex': 0,
u'RowCount': 16
}
request = queryservice.QuerySubsetRequest(
request_id=u'3',
owner_uri=owner_uri,
json_rpc_client=rpc_client,
parameters=parameters)
self.verify_query_service_response(request=request,
expected_result_subsets=1,
expected_row_count=16)
rpc_client.shutdown()
def create_request(self, request_type, parameters):
"""
Create request of request type passed in.
"""
request = None
self.current_id = uuid.uuid4().int
if request_type == u'connection_request':
logger.info(u'SqlToolsClient connection request Id {0}'.format(
self.current_id))
request = connection.ConnectionRequest(self.current_id,
self.json_rpc_client,
parameters)
return request
if request_type == u'query_execute_string_request':
logger.info(u'SqlToolsClient execute string request Id {0}'.format(
self.current_id))
request = query.QueryExecuteStringRequest(self.current_id,
self.json_rpc_client,
parameters)
return request
if request_type == u'query_subset_request':
logger.info(u'SqlToolsClient subset request Id {0}'.format(
self.current_id))
request = query.QuerySubsetRequest(self.current_id,
self.json_rpc_client,
parameters)
return request
def test_query_subset_response_AdventureWorks2014(self):
"""
Test the retrieval of the actual rows for "select * from HumanResources.Department"
"""
with open(self.get_test_baseline(u'select_from_humanresources_department_adventureworks2014.txt'), u'r+b', buffering=0) as response_file:
request_stream = io.BytesIO()
rpc_client = json_rpc_client.JsonRpcClient(
request_stream, response_file)
rpc_client.start()
# Submit a dummy request.
parameters = {u'OwnerUri': u'connectionservicetest',
u'BatchIndex': 0,
u'ResultSetIndex': 0,
u'RowsStartIndex': 0,
u'RowCount': 16}
request = queryservice.QuerySubsetRequest(
3, rpc_client, parameters)
self.verify_subset_response(request=request)
rpc_client.shutdown()