diff options
-rw-r--r-- | tests/probetests.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/probetests.py b/tests/probetests.py index 7cb2ab2..481eae3 100644 --- a/tests/probetests.py +++ b/tests/probetests.py @@ -504,7 +504,8 @@ class ProbeProxyTest(unittest.TestCase): self.mockObj.MockRet["install"] = address self.probeProxy.install(action, action_installed_cb, error_cb) assert pickle.loads(self.mockObj.MockCall["install"]["args"][0]) == action, "1 argument, the action" - self.mockObj.MockCall["install"]["kwargs"]["reply_handler"](address) + pickled_value = pickle.dumps((address, {})) + self.mockObj.MockCall["install"]["kwargs"]["reply_handler"](pickled_value) #ErrorCase: Update should fail on noninstalled actions self.assertRaises(RuntimeWarning, self.probeProxy.update, action2_address, action2) |