simple function to create an arbitrary mock object

2 files changed, 17 insertions, 0 deletions

diff --git a/test/unittest_testlib.py b/test/unittest_testlib.py
index 09c50e2..a4b68a7 100644
--- a/test/unittest_testlib.py
+++ b/test/unittest_testlib.py
@@ -20,6 +20,13 @@ class MockTestCase(testlib.TestCase):
     def fail(self, msg):
         raise AssertionError(msg)
 
+class UtilTC(testlib.TestCase):
+
+    def test_mockobject(self):
+        obj = testlib.mock_object(foo='bar', baz='bam')
+        self.assertEquals(obj.foo, 'bar')
+        self.assertEquals(obj.baz, 'bam')
+
 class TestlibTC(testlib.TestCase):
 
     def setUp(self):
diff --git a/testlib.py b/testlib.py
index 40a345b..2f6bf7d 100644
--- a/testlib.py
+++ b/testlib.py
@@ -626,3 +626,13 @@ class MockConnection:
 
 MockConnexion = class_renamed('MockConnexion', MockConnection)
 
+def mock_object(**params):
+    """creates an object using params to set attributes
+    >>> option = mock_object(verbose=False, index=range(5))
+    >>> option.verbose
+    False
+    >>> option.index
+    [0, 1, 2, 3, 4]
+    """
+    return type('Mock', (), params)()
+