1import unittest
  2from unittest.mock import mock_open, patch
  3from secret import validate_recipe, RuinedNikuldenDinnerError
  4
  5def memoization(func): # Decorator to cache results of expensive function calls
  6    data = {}
  7
  8    def wrapper(*args):
  9        if args in data:
 10            return data[args]
 11        result = func(*args)
 12        data[args] = result
 13        return result
 14
 15    return wrapper
 16
 17@memoization
 18def generate_all_variations(word):
 19    size = len(word)
 20    result = []
 21    for i in range(1 << size): # Loop through all possible bitmasks for the word  
 22        current = []
 23        for j in range(size): # Loop through all letters (their binary representation)
 24            if i & (1 << j):  # Check if the j-th bit is 1 (is upper)
 25                current.append(word[j].upper())
 26            else:
 27                current.append(word[j].lower())
 28        result.append(''.join(current))
 29    return result
 30
 31@memoization
 32def generate_all_insertions_for_keyword(word, special_word):
 33    # Generate all insertions of special_word into the given word
 34    return [word[:i] + special_word + word[i:] for i in range(len(word) + 1)]
 35
 36@memoization
 37def generate_all_insertions_for_random(word, special_word):
 38    # Generate all insertions of word into the given special_word
 39    return [special_word[:i] + word + special_word[i:] for i in range(len(special_word) + 1)]
 40
 41
 42class TestNikuldenValidator(unittest.TestCase):
 43
 44    def setUp(self):
 45        self.keywords = ["риба", "рибена", "шаран", "сьонга"]
 46        self.special_word = "тр"
 47        self.valid_keywords = set()
 48        self.invalid_keywords = set()
 49        self.some_edge_cases = ["", "\t", "\n", "123", "!", "?", ".", ","]
 50
 51        self.templates = [
 52            "Днес ще ям {keyword}, защото е Никулден ;Д.",
 53            "Тази рецепта включва {keyword}.",
 54            "{keyword} е подходящо ястие за Никулден.",
 55            "Нямам против да приготвя {keyword}, знаейки че иначе ще бъде жена ми ;(.",
 56            "{keyword} е доста екзотично и полезно ястие.",
 57            "Практически {keyword} е клас {keyword} нали???",
 58            "{keyword}",
 59            "{keyword} ",
 60            " {keyword}",
 61            " {keyword} "
 62        ]
 63
 64        for current_keyword in self.keywords:
 65            variations = generate_all_variations(current_keyword)
 66            self.valid_keywords.update(variations)
 67
 68    def test_valid_recipe(self):
 69        valid_contents = [
 70            template.format(keyword=current_keyword)
 71            for current_keyword in self.valid_keywords
 72            for template in self.templates
 73        ]
 74
 75        for content in valid_contents:
 76            with self.subTest(content=content):
 77                m = mock_open(read_data=content)
 78                with patch("builtins.open", m):
 79                    result = validate_recipe("dummy_path.txt")
 80                    self.assertTrue(result, "Error")
 81
 82    def test_invalid_recipe(self):
 83        for current_keyword in self.valid_keywords:
 84            variations_keyword = generate_all_insertions_for_keyword(current_keyword, self.special_word)
 85            variations_random = generate_all_insertions_for_random(current_keyword, self.special_word)
 86            self.invalid_keywords.update(variations_keyword)
 87            self.invalid_keywords.update(variations_random)
 88
 89        invalid_contents = self.some_edge_cases.copy() # self.some_edge_cases is mutable
 90
 91        invalid_contents.extend(
 92            template.format(keyword=current_keyword)
 93            for current_keyword in self.invalid_keywords
 94            for template in self.templates
 95        )
 96
 97        for content in invalid_contents:
 98            with self.subTest(content=content):
 99                m = mock_open(read_data=content)
100                with patch("builtins.open", m):
101                    result = validate_recipe("dummy_path.txt")
102                    self.assertFalse(result, "Error")
103
104    def test_bad_recipe_file(self):
105        error_cases = [OSError, IOError]
106
107        for error in error_cases:
108            with self.subTest(error=error):
109                with patch("builtins.open", side_effect=error):
110                    with self.assertRaises(RuinedNikuldenDinnerError):
111                        validate_recipe("missing_file.txt")
112
113if __name__ == "__main__":
114    unittest.main()