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

This commit is contained in:
Dawid Fatyga 2012-04-24 22:22:38 +02:00
parent 89f703dc85
commit 4bf088ce24
1 changed files with 5 additions and 1 deletions

6
mox.py
View File

@ -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.