Skip to content

Commit 9cd2b0a

Browse files
committed
Add ability to run only SQL tests
1 parent 1944dc3 commit 9cd2b0a

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

tests/conftest.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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")
4449
def 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")
4959
def 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

Comments
 (0)