@@ -38,13 +38,23 @@ def pytest_addoption(parser):
3838 action = "store_true" ,
3939 default = False ,
4040 )
41+ parser .addoption (
42+ "--sql-tests" ,
43+ action = "store_true" ,
44+ default = False ,
45+ )
4146
4247
4348@fixture (scope = "session" )
4449def is_unit_tests_run (request ):
4550 return request .config .getoption ("--unit-tests" )
4651
4752
53+ @fixture (scope = "session" )
54+ def is_sql_tests_run (request ):
55+ return request .config .getoption ("--sql-tests" )
56+
57+
4858@fixture (scope = "session" )
4959def raw_data ():
5060 user_ids = count (1 )
@@ -75,8 +85,8 @@ def entities(raw_data):
7585
7686
7787@fixture (scope = "session" )
78- def cassandra_session (cassandra_address , is_unit_tests_run ):
79- if is_unit_tests_run :
88+ def cassandra_session (cassandra_address , is_unit_tests_run , is_sql_tests_run ):
89+ if is_unit_tests_run or is_sql_tests_run :
8090 return
8191
8292 with Cluster (
@@ -180,8 +190,8 @@ async def _setup_sqlite(sqlite_file, raw_data, is_unit_tests_run):
180190
181191
182192@async_fixture (scope = "session" , autouse = True )
183- async def _setup_mongodb (mongodb_url , raw_data , is_unit_tests_run ):
184- if is_unit_tests_run :
193+ async def _setup_mongodb (mongodb_url , raw_data , is_unit_tests_run , is_sql_tests_run ):
194+ if is_unit_tests_run or is_sql_tests_run :
185195 return
186196
187197 client = AsyncIOMotorClient (mongodb_url )
0 commit comments