Fixed SpecChecker to not break with callables not supported by

getargspec (e.g. builtin functions).

1 files changed, 23 insertions, 0 deletions

diff --git a/test.py b/test.py
index 19b132b..2d74217 100755
--- a/test.py
+++ b/test.py
@@ -211,6 +211,18 @@ class IntegrationTest(unittest.TestCase):
         self.assertEquals(list(mock), [1, 2, 3])
         self.mocker.verify()
 
+    def test_replace_builtin_function(self):
+        """
+        Inspection doesn't work on builtin functions, but proxying should
+        work even then (without spec enforcement in these cases).
+        """
+        from zlib import adler32
+        mock = self.mocker.proxy(adler32)
+        mock()
+        self.mocker.result(42)
+        self.mocker.replay()
+        self.assertEquals(mock(), 42)
+
 
 class ExpectTest(unittest.TestCase):
 
@@ -2644,6 +2656,17 @@ class SpecCheckerTest(unittest.TestCase):
         else:
             self.fail("AssertionError not raised")
 
+    def test_unsupported_object_for_getargspec(self):
+        from zlib import adler32
+        # If that fails, this test has to change because either adler32 has
+        # changed, or the implementation of getargspec has changed.
+        self.assertRaises(TypeError, inspect.getargspec, adler32)
+        try:
+            task = SpecChecker(adler32)
+            task.run(self.path("asd"))
+        except TypeError, e:
+            self.fail("TypeError: %s" % str(e))
+
     def test_recorder(self):
         self.mocker.add_recorder(spec_checker_recorder)
         obj = self.mocker.mock(spec=self.cls)