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)