diff --git a/graphene_sqlalchemy/tests/test_batching.py b/graphene_sqlalchemy/tests/test_batching.py index ab540225..a8993c2d 100644 --- a/graphene_sqlalchemy/tests/test_batching.py +++ b/graphene_sqlalchemy/tests/test_batching.py @@ -125,6 +125,8 @@ def test_many_to_one(session_factory): # The batched SQL statement generated is different in 1.2.x # SQLAlchemy 1.3+ optimizes out a JOIN statement in `selectin` # See https://git.io/JewQu + sql_statements = [message for message in messages if 'SELECT' in message and ' FROM articles'] + assert len(sql_statements) == 2 return assert messages == [ @@ -214,6 +216,8 @@ def test_one_to_one(session_factory): # The batched SQL statement generated is different in 1.2.x # SQLAlchemy 1.3+ optimizes out a JOIN statement in `selectin` # See https://git.io/JewQu + sql_statements = [message for message in messages if 'SELECT' in message and ' FROM articles'] + assert len(sql_statements) == 2 return assert messages == [ @@ -316,6 +320,8 @@ def test_one_to_many(session_factory): # The batched SQL statement generated is different in 1.2.x # SQLAlchemy 1.3+ optimizes out a JOIN statement in `selectin` # See https://git.io/JewQu + sql_statements = [message for message in messages if 'SELECT' in message and ' FROM articles'] + assert len(sql_statements) == 2 return assert messages == [ @@ -442,6 +448,8 @@ def test_many_to_many(session_factory): # The batched SQL statement generated is different in 1.2.x # SQLAlchemy 1.3+ optimizes out a JOIN statement in `selectin` # See https://git.io/JewQu + sql_statements = [message for message in messages if 'SELECT' in message and ' FROM pets'] + assert len(sql_statements) == 2 return assert messages == [