Checking method call parameters

Siddharta Govindaraj · Siddharta Govindaraj · commit b48fe76bcf6e · 2015-03-23T14:54:40.000+05:30
diff --git a/stock_alerter/tests/test_event.py b/stock_alerter/tests/test_event.py
@@ -4,15 +4,6 @@
 from ..event import Event
 
 
-class Mock:
-    called = False
-    params = ()
-
-    def __call__(self, *args, **kwargs):
-        self.called = True
-        self.params = (args, kwargs)
-
-
 class EventTest(unittest.TestCase):
     def test_a_listener_is_notified_when_an_event_is_raised(self):
         listener = mock.Mock()
@@ -22,8 +13,8 @@ def test_a_listener_is_notified_when_an_event_is_raised(self):
         self.assertTrue(listener.called)
 
     def test_a_listener_is_passed_right_parameters(self):
-        listener = Mock()
+        listener = mock.Mock()
         event = Event()
         event.connect(listener)
         event.fire(5, shape="square")
-        self.assertEqual(((5, ), {"shape": "square"}), listener.params)
+        listener.assert_called_with(5, shape="square")
