Fixed a bug when dir() called on MockObject was not returning attributes of a mocked class (Python3).

1 files changed, 5 insertions, 1 deletions

diff --git a/mox.py b/mox.py
index fd68fdf..b2c671f 100755
--- a/mox.py
+++ b/mox.py
@@ -576,7 +576,7 @@ class MockObject(MockAnything):
     self._known_vars = set()
     self._class_to_mock = class_to_mock
 
-    if inspect.isclass(self._class_to_mock):
+    if inspect.isclass(class_to_mock):
       self._class_to_bind = self._class_to_mock
     else:
       self._class_to_bind = class_to_bind
@@ -817,6 +817,10 @@ class MockObject(MockAnything):
     """Return the name that is being mocked."""
     return self._description
 
+  def __dir__(self):
+    """Return only attributes of a class to mock """
+    return dir(self._class_to_mock)
+
 
 class _MockObjectFactory(MockObject):
   """A MockObjectFactory creates mocks and verifies __init__ params.