call_function_test: Fix assumption that we run as a script

2018-04-01 01:23:13 +01:00 · 2018-04-01 01:23:13 +01:00 · a22294dda9
parent 0dcaeb21a2
commit a22294dda9
1 changed files with 4 additions and 1 deletions
--- a/tests/call_function_test.py
+++ b/tests/call_function_test.py
@ -64,7 +64,10 @@ class CallFunctionTest(testlib.RouterMixin, testlib.TestCase):
    def test_bad_return_value(self):
        exc = self.assertRaises(mitogen.core.StreamError,
            lambda: self.local.call(func_with_bad_return_value))
-        self.assertEquals(exc[0], "cannot unpickle '__main__'/'CrazyType'")
+        self.assertEquals(
+                exc[0],
+                "cannot unpickle '%s'/'CrazyType'" % (__name__,),
+        )

    def test_returns_dead(self):
        self.assertEqual(mitogen.core._DEAD, self.local.call(func_returns_dead))