Add relationship tests

Author: piyush-jaiswal

tests/conftest.py

@@ -10,7 +10,6 @@
 from tests import utils
 
 
-
 @pytest.fixture
 def client():
     app.config["TESTING"] = True
@@ -49,3 +48,46 @@ def _get_headers(email="[email protected]", password="testpassword"):
         return utils.get_auth_header(tokens["access_token"])
 
     return _get_headers
+
+
+@pytest.fixture
+def create_category(client, create_authenticated_headers):
+    def _create(name, subcategories=None, headers=None):
+        if headers is None:
+            headers = create_authenticated_headers()
+        payload = {"name": name}
+        if subcategories is not None:
+            payload["subcategories"] = subcategories
+        return client.post("/category/create", json=payload, headers=headers)
+
+    return _create
+
+
+@pytest.fixture
+def create_subcategory(client, create_authenticated_headers):
+    def _create(name, categories=None, products=None, headers=None):
+        if headers is None:
+            headers = create_authenticated_headers()
+        payload = {"name": name}
+        if categories is not None:
+            payload["categories"] = categories
+        if products is not None:
+            payload["products"] = products
+        return client.post("/subcategory/create", json=payload, headers=headers)
+
+    return _create
+
+
+@pytest.fixture
+def create_product(client, create_authenticated_headers):
+    def _create(name, description=None, subcategories=None, headers=None):
+        if headers is None:
+            headers = create_authenticated_headers()
+        payload = {"name": name}
+        if description is not None:
+            payload["description"] = description
+        if subcategories is not None:
+            payload["subcategories"] = subcategories
+        return client.post("/product/create", json=payload, headers=headers)
+
+    return _create

tests/test_relationships.py

@@ -0,0 +1,186 @@
+import pytest
+
+from app.models import Category, Subcategory, Product
+
+
+class TestRelationships:
+    def _category_subcategory_ids(self, client, category_id):
+        with client.application.app_context():
+            category = Category.query.get(category_id)
+            assert category is not None
+            return sorted([subcategory.id for subcategory in category.subcategories])
+
+    def _subcategory_category_ids(self, client, subcategory_id):
+        with client.application.app_context():
+            subcategory = Subcategory.query.get(subcategory_id)
+            assert subcategory is not None
+            return sorted([category.id for category in subcategory.categories])
+
+    def _subcategory_product_ids(self, client, subcategory_id):
+        with client.application.app_context():
+            subcategory = Subcategory.query.get(subcategory_id)
+            assert subcategory is not None
+            return sorted([product.id for product in subcategory.products])
+
+    def _product_subcategory_ids(self, client, product_id):
+        with client.application.app_context():
+            product = Product.query.get(product_id)
+            assert product is not None
+            return sorted([subcategory.id for subcategory in product.subcategories])
+
+    def _category_product_ids_via_subcategories(self, client, category_id):
+        with client.application.app_context():
+            category = Category.query.get(category_id)
+            assert category is not None
+            return sorted({product.id for subcategory in category.subcategories for product in subcategory.products.all()})
+
+    def test_create_category_with_subcategories(self, client, create_category, create_subcategory):
+        subcategory1 = create_subcategory("SC_A").get_json()
+        subcategory2 = create_subcategory("SC_B").get_json()
+        response = create_category("Cat_AB", subcategories=[subcategory1["id"], subcategory2["id"]])
+        assert response.status_code == 201
+        category = response.get_json()
+
+        assert self._category_subcategory_ids(client, category["id"]) == sorted([subcategory1["id"], subcategory2["id"]])
+        assert category["id"] in self._subcategory_category_ids(client, subcategory1["id"])
+        assert category["id"] in self._subcategory_category_ids(client, subcategory2["id"])
+
+    def test_create_subcategory_with_categories_and_products(self, client, create_subcategory, create_category, create_product):
+        category1 = create_category("C1").get_json()
+        category2 = create_category("C2").get_json()
+        product1 = create_product("P1", "des").get_json()
+        product2 = create_product("P2", "desc").get_json()
+
+        response = create_subcategory("SC_C1C2_P1P2", categories=[category1["id"], category2["id"]], products=[product1["id"], product2["id"]])
+        assert response.status_code == 201
+        subcategory = response.get_json()
+
+        assert self._subcategory_category_ids(client, subcategory["id"]) == sorted([category1["id"], category2["id"]])
+        assert self._subcategory_product_ids(client, subcategory["id"]) == sorted([product1["id"], product2["id"]])
+
+    def test_create_product_with_subcategories_links_to_category_products(self, client, create_category, create_subcategory, create_product):
+        category = create_category("C").get_json()
+        subcategory = create_subcategory("SC", categories=[category["id"]]).get_json()
+        response = create_product("P", "desc", subcategories=[subcategory["id"]])
+        assert response.status_code == 201
+        product = response.get_json()
+
+        assert subcategory["id"] in self._product_subcategory_ids(client, product["id"])
+        assert product["id"] in self._category_product_ids_via_subcategories(client, category["id"])
+        assert product["id"] in self._subcategory_product_ids(client, subcategory["id"])
+
+    def test_update_category_adds_subcategories(self, client, create_authenticated_headers, create_category, create_subcategory):
+        subcategory1 = create_subcategory("U_SC1").get_json()
+        subcategory2 = create_subcategory("U_SC2").get_json()
+        category = create_category("U_Cat", subcategories=[subcategory1["id"]]).get_json()
+
+        headers = create_authenticated_headers()
+        update_response = client.put(f"/category/{category['id']}/update", json={"subcategories": [subcategory2["id"]]}, headers=headers)
+        assert update_response.status_code == 201
+
+        assert self._category_subcategory_ids(client, category["id"]) == sorted([subcategory1["id"], subcategory2["id"]])
+
+    def test_update_subcategory_adds_categories_and_products(self, client, create_authenticated_headers, create_category, create_product, create_subcategory):
+        category1 = create_category("UC1").get_json()
+        category2 = create_category("UC2").get_json()
+        product1 = create_product("UP1").get_json()
+        product2 = create_product("UP2").get_json()
+        subcategory = create_subcategory("U_SC").get_json()
+
+        headers = create_authenticated_headers()
+        update_response = client.put(
+            f"/subcategory/{subcategory['id']}/update",
+            json={"categories": [category1["id"], category2["id"]], "products": [product1["id"], product2["id"]]},
+            headers=headers,
+        )
+        assert update_response.status_code == 201
+
+        assert self._subcategory_category_ids(client, subcategory["id"]) == sorted([category1["id"], category2["id"]])
+        assert self._subcategory_product_ids(client, subcategory["id"]) == sorted([product1["id"], product2["id"]])
+
+    def test_update_product_adds_subcategories(self, client, create_authenticated_headers, create_product, create_subcategory):
+        subcategory1 = create_subcategory("UPS1").get_json()
+        subcategory2 = create_subcategory("UPS2").get_json()
+        product = create_product("UP", "desc", subcategories=[subcategory1["id"]]).get_json()
+
+        headers = create_authenticated_headers()
+        update_response = client.put(f"/product/{product['id']}/update", json={"subcategories": [subcategory2["id"]]}, headers=headers)
+        assert update_response.status_code == 201
+
+        assert self._product_subcategory_ids(client, product["id"]) == sorted([subcategory1["id"], subcategory2["id"]])
+
+    def test_get_category_subcategories_empty(self, client, create_category):
+        category = create_category("Cat_NoSC").get_json()
+        assert self._category_subcategory_ids(client, category["id"]) == []
+
+    def test_get_category_subcategories_populated(self, client, create_category, create_subcategory):
+        subcategory1 = create_subcategory("SC1").get_json()
+        subcategory2 = create_subcategory("SC2").get_json()
+        category = create_category("Cat_WithSC", subcategories=[subcategory1["id"], subcategory2["id"]]).get_json()
+
+        assert self._category_subcategory_ids(client, category["id"]) == sorted([subcategory1["id"], subcategory2["id"]])
+
+    def test_get_category_products_empty(self, client, create_category):
+        category = create_category("Cat_NoProd").get_json()
+        assert set(self._category_product_ids_via_subcategories(client, category["id"])) == set()
+
+    def test_get_category_products_populated_with_pagination(self, client, create_category, create_subcategory, create_product):
+        category = create_category("Cat_Prod").get_json()
+        subcategory = create_subcategory("SC_Prod", categories=[category["id"]]).get_json()
+
+        product_ids = set()
+        for index in range(12):
+            product_resp = create_product(f"P{index}", "desc", subcategories=[subcategory["id"]])
+            product_ids.add(product_resp.get_json().get("id"))
+
+        page1 = client.get(f"/category/{category['id']}/products?page=1").get_json()
+        page2 = client.get(f"/category/{category['id']}/products?page=2").get_json()
+        assert len(page1["products"]) == 10
+        assert len(page2["products"]) == 2
+
+        returned_product_ids = set(p["id"] for p in page1["products"] + page2["products"])
+        assert returned_product_ids == product_ids
+
+    def test_get_subcategory_categories_populated(self, client, create_category, create_subcategory):
+        category1 = create_category("C1").get_json()
+        category2 = create_category("C2").get_json()
+        subcategory = create_subcategory("SC_Cats", categories=[category1["id"], category2["id"]]).get_json()
+
+        assert self._subcategory_category_ids(client, subcategory["id"]) == sorted([category1["id"], category2["id"]])
+
+    def test_get_subcategory_products_populated_with_pagination(self, client, create_subcategory, create_product):
+        subcategory = create_subcategory("SC_Pag").get_json()
+
+        product_ids = set()
+        for index in range(11):
+            product_resp = create_product(f"SP{index}", "desc", subcategories=[subcategory["id"]])
+            product_ids.add(product_resp.get_json().get("id"))
+
+        page1 = client.get(f"/subcategory/{subcategory['id']}/products?page=1").get_json()
+        page2 = client.get(f"/subcategory/{subcategory['id']}/products?page=2").get_json()
+        assert len(page1["products"]) == 10
+        assert len(page2["products"]) == 1
+
+        returned_product_ids = set(p["id"] for p in page1["products"] + page2["products"])
+        assert returned_product_ids == product_ids
+
+    def test_get_product_subcategories_populated(self, client, create_product, create_subcategory):
+        subcategory1 = create_subcategory("S1").get_json()
+        subcategory2 = create_subcategory("S2").get_json()
+        product = create_product("Prod_SC", "desc", subcategories=[subcategory1["id"], subcategory2["id"]]).get_json()
+
+        assert self._product_subcategory_ids(client, product["id"]) == sorted([subcategory1["id"], subcategory2["id"]])
+
+    @pytest.mark.parametrize(
+        "path",
+        [
+            "/category/999999/subcategories",
+            "/category/999999/products",
+            "/subcategory/999999/categories",
+            "/subcategory/999999/products",
+            "/product/999999/subcategories",
+        ],
+    )
+    def test_relationship_getters_404(self, client, path):
+        resp = client.get(path)
+        assert resp.status_code == 404