Python3 mock doesn't have assert_called_once

Oops. With python 2.x we use pypi's mock module but with python 3.x it's built in (under unittest.mock). I thought Python bundled an exact copy of the upstream module but seems it doesn't include the assert_called_once method... ====================================================================== ERROR: test_event_listing_with_malformed_event ---------------------------------------------------------------------- Traceback (most recent call last): File "/usr/lib/python3.5/unittest/mock.py", line 1157, in patched return func(*args, **keywargs) File "/home/atagar/Desktop/stem/test/unit/control/controller.py", line 688, in test_event_listing_with_malformed_event self.malformed_listener.assert_called_once() File "/usr/lib/python3.5/unittest/mock.py", line 583, in __getattr__ raise AttributeError(name) AttributeError: assert_called_once

Python3 mock doesn't have assert_called_once
e75cf259 · Damian Johnson · c9cc0f55 · e75cf259
Commit e75cf259 authored Aug 14, 2018 by Damian Johnson
--- a/test/unit/control/controller.py
+++ b/test/unit/control/controller.py
@@ -10,6 +10,7 @@ import unittest

 import stem.descriptor.router_status_entry
 import stem.response
+import stem.response.events
 import stem.socket
 import stem.util.system
 import stem.version
@@ -681,11 +682,18 @@ class TestControl(unittest.TestCase):
    is_alive_mock.return_value = True
    self.controller._launch_threads()

+    # When stem.response.convert() encounters malformed content we still recast
+    # the message.
+
+    expected_bad_event = ControlMessage.from_str(BAD_EVENT.raw_content())
+    setattr(expected_bad_event, 'arrived_at', TEST_TIMESTAMP)
+    expected_bad_event.__class__ = stem.response.events.BandwidthEvent
+
    try:
      self._emit_event(BAD_EVENT)
      self.circ_listener.assert_not_called()
      self.bw_listener.assert_not_called()
-      self.malformed_listener.assert_called_once()
+      self.malformed_listener.assert_called_once_with(casted_bad_event)

      self._emit_event(BW_EVENT)
      self.bw_listener.assert_called_once_with(BW_EVENT)