Added test for pymysql connection library. Renamed and modified the mysql.connector test based on discoveries writing this test.

This commit is contained in:
Timothy Sehn
2020-08-24 16:47:12 -07:00
parent fd5280eb72
commit 9d04bc471d
3 changed files with 53 additions and 6 deletions

View File

@@ -12,7 +12,6 @@ QUERY_RESPONSE = [
]
def main():
print(sys.argv)
user = sys.argv[1]
port = sys.argv[2]
db = sys.argv[3]
@@ -22,7 +21,6 @@ def main():
port=port,
database=db)
for query_response in QUERY_RESPONSE:
query = list(query_response.keys())[0]
exp_results = query_response[query]
@@ -33,6 +31,8 @@ def main():
print(exp_results)
print(results)
if ( results != exp_results ):
print("Query:")
print(query)
print("Expected:")
print(exp_results)
print("Received:")

View File

@@ -0,0 +1,43 @@
import pymysql
import sys
QUERY_RESPONSE = [
{ "create table test (pk int, value int, primary key(pk))": () },
{ "describe test": (
('pk', 'int', 'NO', 'PRI', '', ''),
('value', 'int', 'YES', '', '', '')
) },
{ "insert into test (pk, value) values (0,0)": () },
{ "select * from test": ((0,0),) }
]
def main():
user = sys.argv[1]
port = int(sys.argv[2])
db = sys.argv[3]
connection = pymysql.connect(host="127.0.0.1",
port=port,
user=user,
db=db)
for query_response in QUERY_RESPONSE:
query = list(query_response.keys())[0]
exp_results = query_response[query]
with connection.cursor() as cursor:
cursor.execute(query)
results = cursor.fetchall()
if ( results != exp_results ):
print("Query:")
print(query)
print("Expected:")
print(exp_results)
print("Received:")
print(results)
exit(1)
connection.close()
exit(0)
main()