Merge pull request #116 from kaste/unstub-str

Author: kaste

mockito/mockito.py

@@ -360,6 +360,8 @@ def unstub(*objs):
 
     if objs:
         for obj in objs:
+            if isinstance(obj, str):
+                obj = get_obj(obj)
             mock_registry.unstub(obj)
     else:
         mock_registry.unstub_all()

tests/modulefunctions_test.py

@@ -36,6 +36,15 @@ def testUnstubs(self):
         unstub()
         self.assertEqual(False, os.path.exists("test"))
 
+    def testUnstubsByDottedPath(self):
+        when("os.path").exists("test").thenReturn(True)
+
+        self.assertEqual(True, os.path.exists("test"))
+
+        unstub("os.path")
+
+        self.assertEqual(False, os.path.exists("test"))
+
     def testStubs(self):
         when(os.path).exists("test").thenReturn(True)