Добавлены тесты

tests/test_bun.py

@@ -0,0 +1,29 @@
+import pytest
+from praktikum.bun import Bun
+
+
+class TestBun:
+
+    @pytest.mark.parametrize(
+        "name, price",
+        [
+            ("black bun", 100),
+            ("white bun", 200.50),
+            ("red bun", 0),
+        ],
+    )
+    def test_get_name_returns_correct_name(self, name, price):
+        bun = Bun(name, price)
+        assert bun.get_name() == name
+
+    @pytest.mark.parametrize(
+        "name, price",
+        [
+            ("black bun", 100),
+            ("white bun", 200.50),
+            ("red bun", 0),
+        ],
+    )
+    def test_get_price_returns_correct_price(self, name, price):
+        bun = Bun(name, price)
+        assert bun.get_price() == price

tests/test_burger.py

@@ -0,0 +1,67 @@
+from unittest.mock import MagicMock
+import pytest
+from praktikum.burger import Burger
+
+
+class TestBurger:
+    def test_set_buns(self):
+        burger = Burger()
+        mock_bun = MagicMock()
+        burger.set_buns(mock_bun)
+        assert burger.bun == mock_bun
+
+    def test_add_ingredient(self):
+        burger = Burger()
+        mock_ingredient = MagicMock()
+        burger.add_ingredient(mock_ingredient)
+        assert mock_ingredient in burger.ingredients
+        assert len(burger.ingredients) == 1
+
+    def test_remove_ingredient(self):
+        burger = Burger()
+        mock_ing_1 = MagicMock()
+        mock_ing_2 = MagicMock()
+        burger.add_ingredient(mock_ing_1)
+        burger.add_ingredient(mock_ing_2)
+        burger.remove_ingredient(0)
+        assert len(burger.ingredients) == 1
+        assert burger.ingredients[0] == mock_ing_2
+
+    def test_move_ingredient(self):
+        burger = Burger()
+        mock_ing_1 = MagicMock()
+        mock_ing_2 = MagicMock()
+        burger.add_ingredient(mock_ing_1)
+        burger.add_ingredient(mock_ing_2)
+        burger.move_ingredient(0, 1)
+        assert burger.ingredients[0] == mock_ing_2
+        assert burger.ingredients[1] == mock_ing_1
+
+    def test_get_price(self):
+        burger = Burger()
+        mock_bun = MagicMock()
+        mock_bun.get_price.return_value = 100.0
+        mock_ingredient = MagicMock()
+        mock_ingredient.get_price.return_value = 50.0
+
+        burger.set_buns(mock_bun)
+        burger.add_ingredient(mock_ingredient)
+        assert burger.get_price() == 250.0
+
+    def test_get_receipt(self):
+        burger = Burger()
+        mock_bun = MagicMock()
+        mock_bun.get_name.return_value = "black bun"
+        mock_bun.get_price.return_value = 100.0
+        mock_ingredient = MagicMock()
+        mock_ingredient.get_type.return_value = "SAUCE"
+        mock_ingredient.get_name.return_value = "hot sauce"
+        mock_ingredient.get_price.return_value = 50.0
+
+        burger.set_buns(mock_bun)
+        burger.add_ingredient(mock_ingredient)
+        receipt = burger.get_receipt()
+
+        assert "(==== black bun ====)" in receipt
+        assert "= sauce hot sauce =" in receipt
+        assert "Price: 250.0" in receipt

tests/test_database.py

@@ -0,0 +1,14 @@
+from praktikum.database import Database
+
+class TestDatabase:
+    def test_database_buns_count(self):
+        db = Database()
+        buns = db.available_buns()
+        assert len(buns) == 3
+        assert buns[0].get_name() == "black bun"
+
+    def test_database_ingredients_count(self):
+        db = Database()
+        ingredients = db.available_ingredients()
+        assert len(ingredients) == 6
+        assert ingredients[0].get_name() == "hot sauce"

tests/test_ingredient.py

@@ -0,0 +1,39 @@
+import pytest
+from praktikum.ingredient import Ingredient
+from praktikum.ingredient_types import INGREDIENT_TYPE_SAUCE, INGREDIENT_TYPE_FILLING
+
+
+class TestIngredient:
+
+    @pytest.mark.parametrize(
+        "ingredient_type, name, price",
+        [
+            (INGREDIENT_TYPE_SAUCE, "hot sauce", 100),
+            (INGREDIENT_TYPE_FILLING, "cutlet", 200.99),
+        ],
+    )
+    def test_get_type_returns_correct_type(self, ingredient_type, name, price):
+        ingredient = Ingredient(ingredient_type, name, price)
+        assert ingredient.get_type() == ingredient_type
+
+    @pytest.mark.parametrize(
+        "ingredient_type, name, price",
+        [
+            (INGREDIENT_TYPE_SAUCE, "hot sauce", 100),
+            (INGREDIENT_TYPE_FILLING, "cutlet", 200.99),
+        ],
+    )
+    def test_get_name_returns_correct_name(self, ingredient_type, name, price):
+        ingredient = Ingredient(ingredient_type, name, price)
+        assert ingredient.get_name() == name
+
+    @pytest.mark.parametrize(
+        "ingredient_type, name, price",
+        [
+            (INGREDIENT_TYPE_SAUCE, "hot sauce", 100),
+            (INGREDIENT_TYPE_FILLING, "cutlet", 200.99),
+        ],
+    )
+    def test_get_price_returns_correct_price(self, ingredient_type, name, price):
+        ingredient = Ingredient(ingredient_type, name, price)
+        assert ingredient.get_price() == price