Fake tkinter Canvas.create_polygon now a Mock object.

Author: tmontes

tests/fake_tkinter.py

@@ -21,20 +21,13 @@ def __init__(self):
         self.coords = mock.Mock()
         self.update = mock.Mock()
         self.delete = mock.Mock()
-        self.create_polygon_coords = None
-        self.create_polygon_kwargs = None
+        self.create_polygon = mock.Mock(return_value=42)
         self.create_image = mock.Mock(return_value=24)
         self.create_line = mock.Mock(return_value=42)
         self.itemconfig = mock.Mock()
         self.tag_lower = mock.Mock()
         self.tag_raise = mock.Mock()
 
-    def create_polygon(self, coords, **kwargs):
-        self.create_polygon_coords = coords
-        self.create_polygon_kwargs = kwargs
-        # Return some numeric id.
-        return 42
-
 
 
 class FakeWindow:

tests/test_canvas.py

@@ -96,9 +96,12 @@ def test_create_polygon_inverts_y_coordinates(self):
         coords = [0, 0, 1, 1, 2, -2]
         _item_id = c.create_polygon(coords, fill='fill', outline='outline', width=42)
 
-        args = c._canvas.create_polygon_coords
-
-        self.assertEqual(args, [0, 0, 1, -1, 2, 2])
+        c._canvas.create_polygon.assert_called_once_with(
+            [0, 0, 1, -1, 2, 2],
+            fill=mock.ANY,
+            outline=mock.ANY,
+            width=mock.ANY,
+        )
 
 
     def test_create_polygon_passes_args_to_Canvas_create_polygon(self):
@@ -107,8 +110,12 @@ def test_create_polygon_passes_args_to_Canvas_create_polygon(self):
 
         _item_id = c.create_polygon([], fill='fill', outline='outline', width=42)
 
-        kwargs = c._canvas.create_polygon_kwargs
-        self.assertEqual(kwargs, dict(fill='fill', outline='outline', width=42))
+        c._canvas.create_polygon.assert_called_once_with(
+            mock.ANY,
+            fill='fill',
+            outline='outline',
+            width=42,
+        )
 
 
     def test_create_image_returns_integer_item_id(self):

tests/test_vector_sprites.py

@@ -5,6 +5,8 @@
 # See LICENSE for details.
 # ----------------------------------------------------------------------------
 
+from unittest import mock
+
 from aturtle import sprites, shapes
 
 from . import base
@@ -53,24 +55,36 @@ def test_shape_fill_color_passed_to_create_polygon_fill_kwarg(self):
 
         _sprite = sprites.VectorSprite(self.canvas, UnitSquare(fill_color='#009fff'))
 
-        create_polygon_kwargs = self.canvas.create_polygon_kwargs
-        self.assertEqual(create_polygon_kwargs['fill'], '#009fff')
+        self.canvas.create_polygon.assert_called_once_with(
+            mock.ANY,
+            fill='#009fff',
+            outline=mock.ANY,
+            width=mock.ANY,
+        )
 
 
     def test_shape_line_color_passed_to_create_polygon_outline_kwarg(self):
 
         _sprite = sprites.VectorSprite(self.canvas, UnitSquare(line_color='black'))
 
-        create_polygon_kwargs = self.canvas.create_polygon_kwargs
-        self.assertEqual(create_polygon_kwargs['outline'], 'black')
+        self.canvas.create_polygon.assert_called_once_with(
+            mock.ANY,
+            fill=mock.ANY,
+            outline='black',
+            width=mock.ANY,
+        )
 
 
     def test_shape_line_width_passed_to_create_polygon_width_kwarg(self):
 
         _sprite = sprites.VectorSprite(self.canvas, UnitSquare(line_width=2))
 
-        create_polygon_kwargs = self.canvas.create_polygon_kwargs
-        self.assertEqual(create_polygon_kwargs['width'], 2)
+        self.canvas.create_polygon.assert_called_once_with(
+            mock.ANY,
+            fill=mock.ANY,
+            outline=mock.ANY,
+            width=2,
+        )
 
 
     def test_shape_coords(self):