Task_1 (unit tests)

burger.py

@@ -1,7 +1,7 @@
 from typing import List
 
-from praktikum.bun import Bun
-from praktikum.ingredient import Ingredient
+from bun import Bun
+from ingredient import Ingredient
 
 
 class Burger:

database.py

@@ -1,8 +1,8 @@
 from typing import List
 
-from praktikum.bun import Bun
-from praktikum.ingredient import Ingredient
-from praktikum.ingredient_types import INGREDIENT_TYPE_SAUCE, INGREDIENT_TYPE_FILLING
+from bun import Bun
+from ingredient import Ingredient
+from ingredient_types import INGREDIENT_TYPE_SAUCE, INGREDIENT_TYPE_FILLING
 
 
 class Database:

requirements.txt

@@ -0,0 +1,7 @@
+coverage==7.13.4
+iniconfig==2.3.0
+packaging==26.0
+pluggy==1.6.0
+Pygments==2.19.2
+pytest==9.0.2
+pytest-cov==7.0.0

tests/conftest.py

@@ -0,0 +1,26 @@
+import pytest
+from unittest.mock import Mock
+
+
+@pytest.fixture(params=["SAUCE", "FILLING"])
+def mock_ingredient(request):
+    type_param = request.param
+    mock_ingredient = Mock()
+    mock_ingredient.get_price.return_value = 12.99
+    mock_ingredient.get_name.return_value = f"test{type_param.lower().capitalize()}"
+    mock_ingredient.get_type.return_value = type_param
+    return mock_ingredient
+
+
+@pytest.fixture
+def mock_bun():
+    mock_bun = Mock()
+    mock_bun.get_price.return_value = 99.5
+    mock_bun.get_name.return_value = "testBun"
+    return mock_bun
+
+
+@pytest.fixture
+def burger():
+    from burger import Burger
+    return Burger()

tests/test_bun.py

@@ -0,0 +1,18 @@
+import sys
+import os
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from bun import Bun
+
+class TestBun:
+
+    bun = Bun("testBun", "99.5")
+
+    def test_bun_init(self):
+        assert self.bun.name == "testBun"
+        assert self.bun.price == "99.5"
+
+    def test_bun_get_name(self):
+        assert self.bun.get_name() == "testBun"
+
+    def test_bun_get_price(self):
+        assert self.bun.get_price() == "99.5"

tests/test_burger.py

@@ -0,0 +1,52 @@
+import sys
+import os
+from unittest.mock import Mock
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from burger import Burger
+
+class TestBurger:
+
+    def test_burger_init(self, burger):
+        assert burger.bun == None
+        assert burger.ingredients == []
+
+    def test_burger_set_buns(self, burger, mock_bun):
+        burger.set_buns(mock_bun)
+        assert burger.bun == mock_bun
+
+    def test_burger_add_ingredient(self, burger, mock_ingredient):
+        burger.add_ingredient(mock_ingredient)
+        assert burger.ingredients == [mock_ingredient]
+
+    def test_burger_remove_ingredient(self, burger, mock_ingredient):
+        burger.add_ingredient(mock_ingredient)
+        burger.remove_ingredient(0)
+        assert burger.ingredients == []
+
+    def test_burger_move_ingredient(self, burger, mock_ingredient):
+        mock_ingredient2 = Mock()
+        mock_ingredient2.get_price.return_value = 15.0
+        mock_ingredient2.get_name.return_value = f"new{mock_ingredient.get_type().lower().capitalize()}"
+        mock_ingredient2.get_type.return_value = mock_ingredient.get_type()
+
+        burger.add_ingredient(mock_ingredient)
+        burger.add_ingredient(mock_ingredient2)
+        burger.move_ingredient(1, 0)
+        assert burger.ingredients == [mock_ingredient2, mock_ingredient]
+
+    def test_burger_get_price(self, burger, mock_bun, mock_ingredient):
+        burger.set_buns(mock_bun)
+        burger.add_ingredient(mock_ingredient)
+        # Expected price: (99.5 * 2) + 12.99 = 211.99
+        assert burger.get_price() == 211.99
+
+    def test_burger_get_receipt(self, burger, mock_bun, mock_ingredient):
+        burger.set_buns(mock_bun)
+        burger.add_ingredient(mock_ingredient)
+        # Expected receipt:
+        ingredient_type = mock_ingredient.get_type().lower()
+        ingredient_name = mock_ingredient.get_name()
+        bun_name = mock_bun.get_name()
+        price = burger.get_price()
+        expected_receipt = f"(==== {bun_name} ====)\n= {ingredient_type} {ingredient_name} =\n(==== testBun ====)\n\nPrice: {price}"
+        assert burger.get_receipt() == expected_receipt

tests/test_database.py

@@ -0,0 +1,60 @@
+import sys
+import os
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from database import Database
+
+
+class TestDatabase:
+
+    def test_database_init(self):
+        db = Database()
+
+        # Test buns initialization
+        assert len(db.buns) == 3
+        assert db.buns[0].name == "black bun"
+        assert db.buns[0].price == 100
+        assert db.buns[1].name == "white bun"
+        assert db.buns[1].price == 200
+        assert db.buns[2].name == "red bun"
+        assert db.buns[2].price == 300
+
+
+        assert len(db.ingredients) == 6
+
+
+        sauces = [ing for ing in db.ingredients if ing.type == "SAUCE"]
+        assert len(sauces) == 3
+        assert sauces[0].name == "hot sauce"
+        assert sauces[0].price == 100
+        assert sauces[1].name == "sour cream"
+        assert sauces[1].price == 200
+        assert sauces[2].name == "chili sauce"
+        assert sauces[2].price == 300
+
+
+        fillings = [ing for ing in db.ingredients if ing.type == "FILLING"]
+        assert len(fillings) == 3
+        assert fillings[0].name == "cutlet"
+        assert fillings[0].price == 100
+        assert fillings[1].name == "dinosaur"
+        assert fillings[1].price == 200
+        assert fillings[2].name == "sausage"
+        assert fillings[2].price == 300
+
+    def test_database_available_buns(self):
+        db = Database()
+        buns = db.available_buns()
+        assert len(buns) == 3
+        assert buns[0].name == "black bun"
+        assert buns[1].name == "white bun"
+        assert buns[2].name == "red bun"
+
+    def test_database_available_ingredients(self):
+        db = Database()
+        ingredients = db.available_ingredients()
+        assert len(ingredients) == 6
+
+        sauce_count = len([ing for ing in ingredients if ing.type == "SAUCE"])
+        filling_count = len([ing for ing in ingredients if ing.type == "FILLING"])
+        assert sauce_count == 3
+        assert filling_count == 3

tests/test_ingredient.py

@@ -0,0 +1,26 @@
+import sys
+import os
+import pytest
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from ingredient import Ingredient
+
+@pytest.mark.parametrize("type", ["SAUCE", "FILLING"])
+class TestIngredient:
+
+    def test_ingredient_init(self, type):
+        ingredient = Ingredient(type, "testIngredient", 12.99)
+        assert ingredient.type == type
+        assert ingredient.name == "testIngredient"
+        assert ingredient.price == 12.99
+
+    def test_ingredient_get_name(self, type):
+        ingredient = Ingredient(type, "testIngredient", 12.99)
+        assert ingredient.get_name() == "testIngredient"
+
+    def test_ingredient_get_price(self, type):
+        ingredient = Ingredient(type, "testIngredient", 12.99)
+        assert ingredient.get_price() == 12.99
+
+    def test_ingredient_get_type(self, type):
+        ingredient = Ingredient(type, "testIngredient", 12.99)
+        assert ingredient.get_type() == type