11import pymysql
22
33import pytest
4- import testing .mysqld
54
65from aws_xray_sdk .core import patch
76from aws_xray_sdk .core import xray_recorder
87from aws_xray_sdk .core .context import Context
98from aws_xray_sdk .ext .pymysql import unpatch
109
10+ MYSQL_USER = "root"
11+ MYSQL_PASSWORD = "root"
12+ MYSQL_HOST = "localhost"
13+ MYSQL_PORT = 3306
14+ MYSQL_DB_NAME = "test_db"
1115
1216@pytest .fixture (scope = 'module' , autouse = True )
1317def patch_module ():
@@ -32,46 +36,41 @@ def construct_ctx():
3236
3337def test_execute_dsn_kwargs ():
3438 q = 'SELECT 1'
35- with testing .mysqld .Mysqld () as mysqld :
36- dsn = mysqld .dsn ()
37- conn = pymysql .connect (database = dsn ['db' ],
38- user = dsn ['user' ],
39- password = '' ,
40- host = dsn ['host' ],
41- port = dsn ['port' ])
42- cur = conn .cursor ()
43- cur .execute (q )
39+ conn = pymysql .connect (database = MYSQL_DB_NAME ,
40+ user = MYSQL_USER ,
41+ password = MYSQL_PASSWORD ,
42+ host = MYSQL_HOST ,
43+ port = MYSQL_PORT )
44+ cur = conn .cursor ()
45+ cur .execute (q )
4446
4547 subsegment = xray_recorder .current_segment ().subsegments [- 1 ]
4648 assert subsegment .name == 'execute'
4749 sql = subsegment .sql
4850 assert sql ['database_type' ] == 'MySQL'
49- assert sql ['user' ] == dsn [ 'user' ]
51+ assert sql ['user' ] == MYSQL_USER
5052 assert sql ['driver_version' ] == 'PyMySQL'
5153 assert sql ['database_version' ]
5254
5355
5456def test_execute_bad_query ():
5557 q = "SELECT blarg"
56- with testing .mysqld .Mysqld () as mysqld :
57- dsn = mysqld .dsn ()
58- conn = pymysql .connect (database = dsn ['db' ],
59- user = dsn ['user' ],
60- password = '' ,
61- host = dsn ['host' ],
62- port = dsn ['port' ])
63-
64- cur = conn .cursor ()
65- try :
66- cur .execute (q )
67- except Exception :
68- pass
69-
58+ conn = pymysql .connect (database = MYSQL_DB_NAME ,
59+ user = MYSQL_USER ,
60+ password = MYSQL_PASSWORD ,
61+ host = MYSQL_HOST ,
62+ port = MYSQL_PORT )
63+ cur = conn .cursor ()
64+ try :
65+ cur .execute (q )
66+ except Exception :
67+ pass
68+
7069 subsegment = xray_recorder .current_segment ().subsegments [- 1 ]
7170 assert subsegment .name == "execute"
7271 sql = subsegment .sql
7372 assert sql ['database_type' ] == 'MySQL'
74- assert sql ['user' ] == dsn [ 'user' ]
73+ assert sql ['user' ] == MYSQL_USER
7574 assert sql ['driver_version' ] == 'PyMySQL'
7675 assert sql ['database_version' ]
7776
0 commit comments