diff options
-rwxr-xr-x | mox.py | 6 |
1 files changed, 5 insertions, 1 deletions
@@ -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. |