1import unittest
 2from unittest.mock import mock_open, patch
 3from itertools import product
 4from secret import validate_recipe, RuinedNikuldenDinnerError
 5
 6
 7def permute(word):
 8    """Permute all case variations of a word."""
 9    letter_options = [(char.lower(), char.upper()) for char in word]
10    permutations = list(map(''.join, product(*letter_options)))
11    return permutations
12
13
14class TestNikuldenValidator(unittest.TestCase):
15    def setUp(self):
16        """Set up valid and invalid mock recipes."""
17        keywords = ['риба', 'рибена', 'шаран', 'сьонга']
18
19        self.mock_valid_recipes = []
20        for keyword in keywords:
21            self.mock_valid_recipes.extend(permute(keyword))
22
23        self.mock_invalid_recipes = list(
24            map(lambda x: 'a' + x + 'a', self.mock_valid_recipes)
25        )
26
27    def test_valid_recipe(self):
28        for mock_recipe in self.mock_valid_recipes:
29            with patch("builtins.open", mock_open(read_data=mock_recipe)):
30                self.assertTrue(validate_recipe("hacknah_vi.txt"))
31
32    def test_invalid_recipe(self):
33        mock_invalid_recipe = (
34            """Всеки ми казва – спри да я чакаш
35            Ако обича те, ще бъде тук
36            Месеци минаха, даже година
37            Може би има си друг
38            Всеки ми казва – не настоявай
39            И да умираш, дишай напук
40            Живея и дишам с тази надежда
41            Някога пак да си тук"""
42        )
43
44        self.mock_invalid_recipes.append(mock_invalid_recipe) #нямаше как
45
46        for mock_recipe in self.mock_invalid_recipes:
47            with patch("builtins.open", mock_open(read_data=mock_recipe)):
48                self.assertFalse(validate_recipe("hacknah_vi.txt"))
49
50    def test_bad_recipe_file(self):
51        with patch("builtins.open", side_effect=OSError("File error!")):
52            with self.assertRaises(RuinedNikuldenDinnerError):
53                validate_recipe("pak_vi_hacknah.txt")
54
55        with patch("builtins.open", side_effect=IOError("File error!")):
56            with self.assertRaises(RuinedNikuldenDinnerError):
57                validate_recipe("pak_vi_hacknah.txt")
58
59
60if __name__ == "__main__":
61    unittest.main()