unit tests for burger class

conftest.py

@@ -0,0 +1,12 @@
+from praktikum.burger import Burger
+from praktikum.bun import Bun
+from data import bun_data
+import pytest
+
+@pytest.fixture
+def burger():
+    return Burger()
+
+@pytest.fixture
+def bun():
+    return Bun(**bun_data)

data.py

@@ -0,0 +1,13 @@
+from praktikum.ingredient_types import INGREDIENT_TYPE_FILLING, INGREDIENT_TYPE_SAUCE
+
+bun_data = {
+    'name': 'Кунжутная булка',
+    'price': 150  
+}
+
+ingredient_data = [
+    [INGREDIENT_TYPE_FILLING, 'Котлета говяжья', 320],
+    [INGREDIENT_TYPE_SAUCE, 'Чесночный соус', 80]
+]
+
+exception_receipt = '(==== Кунжутная булка ====)\n= filling Котлета говяжья =\n= sauce Чесночный соус =\n(==== Кунжутная булка ====)\n\nPrice: 700'

burger.py → praktikum/burger.py

@@ -3,7 +3,6 @@
 from praktikum.bun import Bun
 from praktikum.ingredient import Ingredient
 
-
 class Burger:
     """
     Модель бургера.

tests/test_burger.py

@@ -0,0 +1,48 @@
+import pytest
+from praktikum.ingredient import Ingredient
+from data import ingredient_data, exception_receipt
+
+class TestBurger:
+    def test_set_buns_success(self, burger, bun):
+        burger.set_buns(bun)
+        assert burger.bun == bun
+
+    @pytest.mark.parametrize('ingredient_type, name, price', ingredient_data) 
+    def test_add_ingredients_success(self, burger, ingredient_type, name, price):
+        ingredient = Ingredient(ingredient_type, name, price)
+        burger.add_ingredient(ingredient)
+        assert ingredient in burger.ingredients
+
+    def test_remove_ingredient_success(self, burger):
+        ingredient1 = Ingredient(*ingredient_data[0])
+        burger.add_ingredient(ingredient1)
+        ingredient2 = Ingredient(*ingredient_data[1])
+        burger.add_ingredient(ingredient2)
+        burger.remove_ingredient(1)
+        assert ingredient2 not in burger.ingredients
+
+    def test_move_ingredient_success(self, burger):
+        ingredient1 = Ingredient(*ingredient_data[0])
+        burger.add_ingredient(ingredient1)
+        ingredient2 = Ingredient(*ingredient_data[1])
+        burger.add_ingredient(ingredient2)
+        burger.move_ingredient(0,1)
+        assert burger.ingredients[1] == ingredient1
+
+    def test_get_price_success(self, burger, bun):
+        burger.set_buns(bun)
+        ingredient1 = Ingredient(*ingredient_data[0])
+        burger.add_ingredient(ingredient1)
+        ingredient2 = Ingredient(*ingredient_data[1])
+        burger.add_ingredient(ingredient2)
+        buns_price = bun.get_price() * 2
+        ingredients_price = ingredient_data[0][2] + ingredient_data[1][2]
+        assert burger.get_price() == buns_price + ingredients_price
+
+    def test_get_receipt(self, burger, bun):
+        burger.set_buns(bun)
+        ingredient1 = Ingredient(*ingredient_data[0])
+        burger.add_ingredient(ingredient1)
+        ingredient2 = Ingredient(*ingredient_data[1])
+        burger.add_ingredient(ingredient2)
+        assert burger.get_receipt() == exception_receipt