diff --git a/tests/unit/test_catalog.py b/tests/unit/test_catalog.py index 2cb53612c..448bb6717 100644 --- a/tests/unit/test_catalog.py +++ b/tests/unit/test_catalog.py @@ -40,9 +40,7 @@ class TestMembershipPlans: session.execute.return_value.fetchall.return_value = [row] sl = MagicMock(return_value=session) - with patch( - "packages.adapters.sqlalchemy_impl.session.SessionLocal", sl, create=True - ): + with patch("packages.adapters.sqlalchemy_impl.session.SessionLocal", sl, create=True): plans = admin_catalog.get_membership_plans() assert len(plans) == 1 @@ -72,9 +70,7 @@ class TestMembershipPlans: session.execute.return_value.fetchall.return_value = [row] sl = MagicMock(return_value=session) - with patch( - "packages.adapters.sqlalchemy_impl.session.SessionLocal", sl, create=True - ): + with patch("packages.adapters.sqlalchemy_impl.session.SessionLocal", sl, create=True): plans = admin_catalog.get_membership_plans() assert len(plans) == 1 @@ -101,9 +97,7 @@ class TestMembershipPlans: session.execute.return_value.fetchall.return_value = [row] sl = MagicMock(return_value=session) - with patch( - "packages.adapters.sqlalchemy_impl.session.SessionLocal", sl, create=True - ): + with patch("packages.adapters.sqlalchemy_impl.session.SessionLocal", sl, create=True): plans = admin_catalog.get_membership_plans() cycles = {p["billing_cycle"] for p in plans} @@ -112,9 +106,7 @@ class TestMembershipPlans: def test_no_session_returns_empty(self): from packages.application.catalog import admin_catalog - with patch( - "packages.adapters.sqlalchemy_impl.session.SessionLocal", None, create=True - ): + with patch("packages.adapters.sqlalchemy_impl.session.SessionLocal", None, create=True): assert admin_catalog.get_membership_plans() == [] @@ -136,9 +128,7 @@ class TestPointsPackages: session.execute.return_value.fetchall.return_value = [row] sl = MagicMock(return_value=session) - with patch( - "packages.adapters.sqlalchemy_impl.session.SessionLocal", sl, create=True - ): + with patch("packages.adapters.sqlalchemy_impl.session.SessionLocal", sl, create=True): packages = admin_catalog.get_points_packages() assert len(packages) == 1 @@ -166,9 +156,7 @@ class TestPointsPackages: session.execute.return_value.fetchall.return_value = [row] sl = MagicMock(return_value=session) - with patch( - "packages.adapters.sqlalchemy_impl.session.SessionLocal", sl, create=True - ): + with patch("packages.adapters.sqlalchemy_impl.session.SessionLocal", sl, create=True): packages = admin_catalog.get_points_packages() assert packages[0]["points"] == 0 @@ -178,9 +166,7 @@ class TestPointsPackages: def test_no_session_returns_empty(self): from packages.application.catalog import admin_catalog - with patch( - "packages.adapters.sqlalchemy_impl.session.SessionLocal", None, create=True - ): + with patch("packages.adapters.sqlalchemy_impl.session.SessionLocal", None, create=True): assert admin_catalog.get_points_packages() == []