From 4bf088ce24e8eb0c8c3a34595eda71267b03d797 Mon Sep 17 00:00:00 2001
From: Dawid Fatyga <dawid.fatyga@gmail.com>
Date: Tue, 24 Apr 2012 22:22:38 +0200
Subject: [PATCH] Fixed a bug when dir() called on MockObject was not returning
 attributes of a mocked class (Python3).

---
 mox.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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.