From b024d1ba89f4869fac35223ba75f1d5e1d41e26f Mon Sep 17 00:00:00 2001 From: Yarne Coppens Date: Sun, 4 Aug 2024 17:18:19 +0200 Subject: [PATCH] Changed product classes to contain image_filename instead of URL --- src/classes/product_classes.py | 2 +- src/data_connection.py | 6 +++--- tests/test_main.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/classes/product_classes.py b/src/classes/product_classes.py index 0752d78..649562f 100644 --- a/src/classes/product_classes.py +++ b/src/classes/product_classes.py @@ -4,4 +4,4 @@ class Product(BaseModel): name: str price: float barcode: int - image_url: HttpUrl \ No newline at end of file + image_filename: str \ No newline at end of file diff --git a/src/data_connection.py b/src/data_connection.py index eae726a..213ba81 100644 --- a/src/data_connection.py +++ b/src/data_connection.py @@ -1,9 +1,9 @@ from src.classes import product_classes -apple = product_classes.Product(name="Appel", price=1.0, barcode=1000 , image_url="https://i.etsystatic.com/16348658/r/il/380709/1340147432/il_570xN.1340147432_p3pc.jpg") -bubble_tea = product_classes.Product(name="Bubble Tea", price=5.0, barcode=1001 ,image_url="https://uxwing.com/wp-content/themes/uxwing/download/food-and-drinks/bubble-tea-icon.png") +pizza = product_classes.Product(name="Pizza", price=3.0, barcode=1000 , image_filename="pizza.svg") +banana = product_classes.Product(name="Banana", price=1.0, barcode=1001 ,image_filename="banana.svg") -product_list = [apple, bubble_tea] +product_list = [pizza, banana] def get_all_products() -> list[product_classes.Product]: return product_list diff --git a/tests/test_main.py b/tests/test_main.py index 64fc017..4f94626 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -10,7 +10,7 @@ client = TestClient(app) def default_product_test(product: product_classes.Product): assert type(product.name) == str assert type(product.price) == float - assert validators.url(str(product.image_url)) + assert type(product.image_filename) == str