1import unittest
 2
 3import solution
 4
 5
 6class TestSanity(unittest.TestCase):
 7    """Check if all of the classes are present."""
 8
 9    def test_santa(self):
10        self.assertIn('Santa', dir(solution), 'Убеди се, че класът "Santa" е наличен с точно това име.')
11        self.assertTrue(isinstance(solution.Santa, type), 'Убеди се, че "Santa" е клас.')
12
13    def test_interval(self):
14        self.assertIn('Kid', dir(solution), 'Убеди се, че класът "Kid" е наличен с точно това име.')
15        self.assertTrue(isinstance(solution.Kid, type), 'Убеди се, че "Kid" е клас.')
16
17
18if __name__ == '__main__':
19    unittest.main()

  1import unittest
  2from copy import deepcopy
  3from importlib import reload
  4
  5import solution
  6
  7
  8class TestKid(unittest.TestCase):
  9    """Test the Kid metaclass."""
 10
 11    def test_class_from_kid(self):
 12        """Test creating new class from Kid."""
 13        cls = solution.Kid('name', (), {'__call__': print})
 14        self.assertIsInstance(cls, solution.Kid)
 15
 16    def test_class_from_kid_without_call_dunder(self):
 17        """Test creating new class from Kid."""
 18        with self.assertRaises(NotImplementedError):
 19            solution.Kid('name', (), {})
 20
 21
 22class TestSanta(unittest.TestCase):
 23    """Test the Santa metaclass."""
 24
 25    KID_DICT = {
 26        '__init__': lambda self: setattr(self, 'SECRET_PRESENT', None),
 27        '__call__': lambda self, present: setattr(self, 'SECRET_PRESENT', present),
 28        'public_with_error': lambda self: 0 / 0,
 29        'public_without_error': lambda self: 42,
 30        '_private_with_error': lambda self: not_existing,
 31    }
 32
 33    def setUp(self):
 34        """Create isntances to be used by test cases."""
 35        self.KidClass1 = solution.Kid('KidClass1', (), deepcopy(self.KID_DICT))
 36        self.KidClass2 = solution.Kid('KidClass2', (), deepcopy(self.KID_DICT))
 37        self.santa = solution.Santa()
 38
 39    def tearDown(self):
 40        """Reload the classes."""
 41        del self.KidClass1
 42        del self.KidClass2
 43        del self.santa
 44        reload(solution)
 45
 46    def test_singleton(self):
 47        """Ensure Santa is a singleton."""
 48        self.assertIs(solution.Santa(), solution.Santa())
 49
 50    def test_call(self):
 51        """Test sending message via calling."""
 52        kid1 = self.KidClass1()
 53        kid2 = self.KidClass2()
 54        self.santa(kid1, "'toy1'")
 55        self.santa(kid2, "'toy2'")
 56        self.assertEqual(list(self.santa), ['toy1', 'toy2'])
 57
 58    def test_mail(self):
 59        """Test sending message via email."""
 60        kid1 = self.KidClass1()
 61        kid2 = self.KidClass2()
 62        self.santa @ f"'toy1'\n{id(kid1)}"
 63        self.santa @ f"'toy2'\n{id(kid2)}"
 64        self.assertEqual(list(self.santa), ['toy1', 'toy2'])
 65
 66    def test_call_and_mail_same_kid(self):
 67        """Test that calls and mails work for the same kid."""
 68        kid1 = self.KidClass1()
 69        self.santa(kid1, "'toy1'")
 70        self.santa @ f"'toy1'\n{id(kid1)}"
 71        self.assertEqual(list(self.santa), ['toy1'])
 72
 73    def test_present_matching(self):
 74        """Test matching signature in the letter."""
 75        kid1 = self.KidClass1()
 76        kid2 = self.KidClass1()
 77        # Misleading toy definiton
 78        self.santa @ f"""
 79            Това е примерен текст, в който има текст в "кавички" и
 80            'апострофи', но на кирилица.
 81
 82            'toy4' а това е текст след дефиницията на играчка с кавичка и апостроф " '
 83
 84            {id(kid1)}
 85        """
 86        # Complex toy definiton after the signature
 87        self.santa @ f"""
 88            {id(kid2)}
 89
 90            "abcdefgHIJKLMNopQRstUVwxYZ 1 2 3 4567890   "
 91        """
 92        self.assertEqual(list(self.santa), ['toy4', 'abcdefgHIJKLMNopQRstUVwxYZ 1 2 3 4567890   '])
 93
 94    def test_signature_matching(self):
 95        """Test matching present in the letter / call."""
 96        kid1 = self.KidClass1()
 97        kid2 = self.KidClass1()
 98        kid3 = self.KidClass1()
 99        # Whitespaces around signature
100        self.santa @ f"""
101            'toy1'
102
103            {id(kid1)}
104        """
105        # Misleading signature 1
106        self.santa @ f"""
107            'toy2'
108            invalid signature row 12345678
109            {id(kid2)}
110        """
111        # Misleading signature 2
112        self.santa @ f"""
113            'toy3'
114            " 12345678
115            {id(kid3)}
116        """
117        self.santa.xmas()
118        self.assertEqual(kid1.SECRET_PRESENT, 'toy1')
119        self.assertEqual(kid2.SECRET_PRESENT, 'toy2')
120        self.assertEqual(kid3.SECRET_PRESENT, 'toy3')
121
122    def test_iterable(self):
123        """Ensure Santa can be iterated multiple times including overwriting presents."""
124        kid1 = self.KidClass1()
125        kid2 = self.KidClass2()
126        self.santa(kid1, '"something"')
127        self.santa(kid2, '"something not used"')
128        self.santa(kid2, '"something else"')
129        self.assertEqual(list(self.santa), ['something', 'something else'])
130        self.assertEqual(list(self.santa), ['something', 'something else'])
131
132    def test_xmass(self):
133        """Test a simple Christmas case."""
134        kid1 = self.KidClass1()
135        kid2 = self.KidClass1()
136        kid3 = self.KidClass2()
137        self.santa(kid1, '"toy1"')
138        self.santa(kid2, '"toy2"')
139        self.santa @ f"'toy3'\n{id(kid3)}"
140        self.santa.xmas()
141        self.assertEqual(kid1.SECRET_PRESENT, 'toy1')
142        self.assertEqual(kid2.SECRET_PRESENT, 'toy2')
143        self.assertEqual(kid3.SECRET_PRESENT, 'toy3')
144
145    def test_xmass_no_wishes(self):
146        """Test a Christmas with no wishes."""
147        kid1 = self.KidClass1()
148        kid2 = self.KidClass1()
149        kid3 = self.KidClass2()
150        self.santa.xmas()
151        self.assertEqual(kid1.SECRET_PRESENT, None)
152        self.assertEqual(kid2.SECRET_PRESENT, None)
153        self.assertEqual(kid3.SECRET_PRESENT, None)
154
155    def test_xmass_no_wishes_but_naughty_kids(self):
156        """Test a Christmas with no wishes, but naughty kids present."""
157        kid1 = self.KidClass1()
158        try:
159            kid1.public_with_error() # Don't care if it's decorated properly
160        except:
161            pass
162        self.santa.xmas()
163        self.assertEqual(kid1.SECRET_PRESENT, None)
164
165    def test_xmass_kid_without_a_wish(self):
166        """Test a Christmas with a kids that hasn't sent a wish."""
167        kid1 = self.KidClass1()
168        kid2 = self.KidClass1()
169        kid3 = self.KidClass2()
170        kid4 = self.KidClass2()
171        self.santa(kid1, '"toy1"')
172        self.santa(kid2, '"toy1"')
173        self.santa @ f"'toy2'\n{id(kid3)}"
174        self.santa.xmas()
175        self.assertEqual(kid1.SECRET_PRESENT, 'toy1')
176        self.assertEqual(kid2.SECRET_PRESENT, 'toy1')
177        self.assertEqual(kid3.SECRET_PRESENT, 'toy2')
178        self.assertEqual(kid4.SECRET_PRESENT, 'toy1')
179
180    def test_xmass_kid_with_multiple_wishes(self):
181        """Test a Christmas with a kid who sends multiple wishes."""
182        kid1 = self.KidClass1()
183        self.santa(kid1, '"toy1"')
184        self.santa(kid1, '"toy2"')
185        self.santa(kid1, '"toy3"')
186        self.santa.xmas()
187        self.assertEqual(kid1.SECRET_PRESENT, 'toy3')
188
189    def test_xmass_years_under_5(self):
190        """Test with passing years with a kid under 5 years old."""
191        kid1 = self.KidClass1()
192        self.santa(kid1, '"toy1"')
193        self.santa.xmas()  # Christmas 1
194        self.assertEqual(kid1.SECRET_PRESENT, 'toy1')
195        kid1.SECRET_PRESENT = None
196        self.santa.xmas()  # Christmas 2
197        self.santa.xmas()  # Christmas 3
198        self.santa.xmas()  # Christmas 4
199        self.santa(kid1, '"toy2"')
200        self.santa.xmas()  # Christmas 5
201        self.assertEqual(kid1.SECRET_PRESENT, 'toy2')
202
203    def test_xmass_years_5_and_over(self):
204        """Test with passing years with kid aged 5 and over."""
205        kid1 = self.KidClass1()
206        self.santa.xmas()  # Christmas 1
207        self.santa.xmas()  # Christmas 2
208        self.santa.xmas()  # Christmas 3
209        self.santa.xmas()  # Christmas 4
210        self.santa.xmas()  # Christmas 5
211        self.santa(kid1, '"toy"')
212        self.santa.xmas()  # Christmas 6
213        self.assertEqual(kid1.SECRET_PRESENT, None)
214
215    def test_xmass_naughty(self):
216        """Test a Christmas with naughty kids."""
217        kid1 = self.KidClass1()
218        kid2 = self.KidClass1()
219        kid3 = self.KidClass2()
220        self.santa(kid1, "'sirenie'")
221        self.santa(kid2, "'sirenie'")
222        with self.assertRaises(ZeroDivisionError):
223            kid1.public_with_error()
224        with self.assertRaises(ZeroDivisionError):
225            kid2.public_with_error()
226        self.santa.xmas()
227        self.assertEqual(kid1.SECRET_PRESENT, 'coal')
228        self.assertEqual(kid2.SECRET_PRESENT, 'coal')
229        self.assertEqual(kid3.SECRET_PRESENT, 'sirenie')
230
231    def test_xmass_private_with_error(self):
232        """Test a Christmas with not-so-naughty kids."""
233        kid1 = self.KidClass1()
234        self.santa(kid1, "'sirenie'")
235        try:
236            kid1._private_with_error()
237        except:
238            pass
239        self.santa.xmas()
240        self.assertEqual(kid1.SECRET_PRESENT, 'sirenie')
241
242    def test_xmass_public_with_no_error(self):
243        """Test a Christmas with not-so-naughty kids."""
244        kid1 = self.KidClass1()
245        self.santa(kid1, "'sirenie'")
246        self.assertEqual(kid1.public_without_error(), 42)
247        self.santa.xmas()
248        self.assertEqual(kid1.SECRET_PRESENT, 'sirenie')
249
250    def test_santa_gift_order(self):
251        """Test ordering of the Santa iterator."""
252        kid1 = self.KidClass1()
253        kid2 = self.KidClass1()
254        kid3 = self.KidClass2()
255        self.santa(kid2, '"toy2"')
256        self.santa @ f"'toy3'\n{id(kid3)}"
257        self.santa(kid1, '"toy1"')
258        self.santa(kid2, '"toy2v2"')
259        # Kid order must remain kid2, kid3, kid1, as the order of initial requests
260        self.assertEqual(list(self.santa), ["toy2v2", "toy3", "toy1"])
261
262
263if __name__ == '__main__':
264    unittest.main()