def test_stored_proc_multiple_result_sets(self):
"""
Verify the results of running a stored proc with multiple result sets
"""
try:
client = create_mssql_cli_client()
create_stored_proc = u"CREATE PROC sp_mssqlcli_multiple_results " \
u"AS " \
u"BEGIN " \
u"SELECT 'Morning' as [Name] UNION ALL select 'Evening' " \
u"SELECT 'Dawn' as [Name] UNION ALL select 'Dusk' UNION ALL select 'Midnight' " \
u"END"
exec_stored_proc = u"EXEC sp_mssqlcli_multiple_results"
del_stored_proc = u"DROP PROCEDURE sp_mssqlcli_multiple_results"
list(client.execute_single_batch_query(create_stored_proc))
row_counts = []
for rows, columns, message, query, is_error in client.execute_single_batch_query(
exec_stored_proc):
row_counts.append(len(rows))
self.assertTrue(row_counts[0] == 2)
self.assertTrue(row_counts[1] == 3)
list(client.execute_single_batch_query(del_stored_proc))
finally:
shutdown(client)
def test_mssqlcliclient_multiple_statement(self):
"""
Verify correct execution of queries separated by semi-colon
"""
try:
client = create_mssql_cli_client()
multi_statement_query = u"select 'Morning' as [Name] UNION ALL select 'Evening'; select 1;"
multi_statement_query2 = u"select 1; select 'foo' from teapot;"
multi_statement_query3 = u"select 'foo' from teapot; select 2;"
for rows, col, message, query, is_error in \
client.execute_multi_statement_single_batch(multi_statement_query):
if query == u"select 'Morning' as [Name] UNION ALL select 'Evening'":
self.assertTrue(len(rows), 2)
else:
self.assertTrue(len(rows), 1)
for rows, col, message, query, is_error in \
client.execute_multi_statement_single_batch(multi_statement_query2):
if query == u"select 1":
self.assertTrue(len(rows) == 1)
else:
self.assertTrue(is_error)
for rows, col, message, query, is_error in \
client.execute_multi_statement_single_batch(multi_statement_query3):
if query == u"select 2":
self.assertTrue(len(rows) == 1)
else:
self.assertTrue(is_error)
finally:
shutdown(client)
def test_schema_table_views_and_columns_query(self):
"""
Verify mssqlcliclient's tables, views, columns, and schema are populated.
Note: This test should run against a database that the credentials
MSSQL_CLI_USER and MSSQL_CLI_PASSWORD have write access to.
"""
try:
client = create_mssql_cli_client()
client.execute_single_batch_query(
'CREATE TABLE tabletest1 (a int, b varchar(25));')
client.execute_single_batch_query(
'CREATE TABLE tabletest2 (x int, y varchar(25), z bit);')
client.execute_single_batch_query(
'CREATE VIEW viewtest as SELECT a from tabletest1;')
client.execute_single_batch_query('CREATE SCHEMA schematest;')
client.execute_single_batch_query(
'CREATE TABLE schematest.tabletest1 (a int);')
assert ('schematest', 'tabletest1') in set(client.tables())
assert ('dbo', 'viewtest') in set(client.views())
assert ('schematest', 'tabletest1', 'a', 'int',
'NULL') in set(client.table_columns())
assert ('dbo', 'viewtest', 'a', 'int',
'NULL') in set(client.view_columns())
assert 'schematest' in client.schemas()
finally:
client.execute_single_batch_query('DROP TABLE tabletest1;')
client.execute_single_batch_query('DROP TABLE tabletest2;')
client.execute_single_batch_query('DROP VIEW viewtest IF EXISTS;')
client.execute_single_batch_query(
'DROP TABLE schematest.tabletest1;')
client.execute_single_batch_query('DROP SCHEMA schematest;')
shutdown(client)
def test_mssqlcliclient_reset_connection(self):
"""
Verify if the MssqlCliClient can successfully reset its connection
"""
try:
client = create_mssql_cli_client()
mssqlcliclient.reset_connection_and_clients(
client.sql_tools_client, client)
finally:
shutdown(client)
def test_connection(self):
"""
Verify a successful connection via returned owner uri.
"""
try:
client = create_mssql_cli_client(
owner_uri=u'connectionservicetest')
self.assertEqual(client.owner_uri, u'connectionservicetest')
finally:
shutdown(client)
def test_json_writer_extra_params(self):
"""
Verify JSON RPC accepts extra paramaters.
"""
try:
client = create_mssql_cli_client()
extra_params = client.extra_params
json_writer = JsonRpcWriter(io.BytesIO())
json_writer.send_request(u'test/method', extra_params, id=1)
except Exception as ex:
self.fail(u'Exception from JsonRpcWriter %s' % ex)
finally:
json_writer.close()
shutdown(client)
def test_get_query_results(self):
"""
Verify number of rows returned and returned query.
"""
try:
client = create_mssql_cli_client()
test_query = u"""
select 1 as [ShiftID], 'Day' as [Name] UNION ALL
select 2, N'魚' UNION ALL
select 3, 'Night'
"""
for rows, col, message, query, is_error in client.execute_single_batch_query(
test_query):
self.assertTrue(len(rows), 3)
self.assertTrue(query, test_query)
finally:
shutdown(client)
def test_format_output_live_connection():
sleep(7)
statement = u"""
select 1 as [ShiftID], 'Day' as [Name] UNION ALL
select 2, N'魚' UNION ALL
select 3, 'Night'
"""
try:
client = create_mssql_cli_client()
result = run_and_return_string_from_formatter(client, statement)
expected = [
u'+-----------+--------+', u'| ShiftID | Name |',
u'|-----------+--------|', u'| 1 | Day |',
u'| 2 | 魚 |', u'| 3 | Night |',
u'+-----------+--------+', u'(3 rows affected)'
]
assert list(result) == expected
finally:
shutdown(client)
def test_format_output_expanded_live_connection():
statement = u"""
select N'配列' as [Name] UNION ALL
select 'Evening' UNION ALL
select 'Night'
"""
try:
client = create_mssql_cli_client()
result = run_and_return_string_from_formatter(client,
statement,
expanded=True)
expected = [
'-[ RECORD 1 ]-------------------------', 'Name | 配列',
'-[ RECORD 2 ]-------------------------', 'Name | Evening',
'-[ RECORD 3 ]-------------------------', 'Name | Night',
'(3 rows affected)'
]
assert list(result) == expected
finally:
shutdown(client)
