From 4bf088ce24e8eb0c8c3a34595eda71267b03d797 Mon Sep 17 00:00:00 2001 From: Dawid Fatyga 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.