Function generate_query_baseline

tests/jsonrpc/generatequerybaseline.py:12–50

(file_name)

# Helper to generate a query baseline
#
def generate_query_baseline(file_name):
    sqltoolsservice_args = [mssqltoolsservice.get_executable_path()]
    with io.open(file_name, 'wb') as baseline:
        tools_service_process = subprocess.Popen(
            sqltoolsservice_args,
            bufsize=0,
            stdin=subprocess.PIPE,
            stdout=baseline)

        # Before running this script, enter a real server name.
        parameters = {u'OwnerUri': u'connectionservicetest',
                      u'Connection':
                          {u'ServerName': u'*',
                           u'DatabaseName': u'AdventureWorks2014',
                           u'UserName': u'*',
                           u'Password': u'*',
                           u'AuthenticationType': u'Integrated'}
                      }

        writer = json_rpc_client.JsonRpcWriter(tools_service_process.stdin)
        writer.send_request(u'connection/connect', parameters, request_id=1)

        time.sleep(2)

        parameters = {u'OwnerUri': u'connectionservicetest',
                      u'Query': "select * from HumanResources.Department"}
        writer.send_request(u'query/executeString', parameters, request_id=2)

        time.sleep(5)
        parameters = {u'OwnerUri': u'connectionservicetest',
                      u'BatchIndex': 0,
                      u'ResultSetIndex': 0,
                      u'RowsStartIndex': 0,
                      u'RowsCount': 16}

        writer.send_request(u'query/subset', parameters, request_id=3)
        # submit raw request.
        time.sleep(5)
        tools_service_process.kill()


if __name__ == '__main__':