From 4eaa17e0cec8fd9d417604337e62df0b24c2e2b3 Mon Sep 17 00:00:00 2001 From: Yarninator Date: Mon, 5 Aug 2024 13:49:17 +0200 Subject: [PATCH] Wrote a icon retrieval test --- tests/test_main.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/test_main.py b/tests/test_main.py index 4f94626..e2957d6 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -4,6 +4,7 @@ import validators from src.main import app from src.classes import product_classes +from shop_validators import image_validator client = TestClient(app) @@ -27,3 +28,11 @@ def test_retrieve_products(): returned_product = product_classes.Product(**response.json()[0]) default_product_test(returned_product) + +def test_retrieve_icon(): + response = client.get("/icons/cart") + assert response.status_code == 200 + + returned_icon = response.text + + assert image_validator.is_valid_svg(file_content=returned_icon)