new container tests by Lorenzo Gil Sanchez

2 files changed, 58 insertions, 1 deletions

diff --git a/tests/Makefile.am b/tests/Makefile.am
index e95c6d7e..9724d474 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -10,7 +10,8 @@ tests =  \
 	test_pango.py \
 	test_radiobutton.py \
 	test_style.py \
-	test_textview.py
+	test_textview.py \
+	test_container.py
 
 GTK_PY_FILES = __init__.py _lazyutils.py compat.py deprecation.py keysyms.py
 
diff --git a/tests/test_container.py b/tests/test_container.py
new file mode 100644
index 00000000..a970fa44
--- /dev/null
+++ b/tests/test_container.py
@@ -0,0 +1,56 @@
+import unittest
+
+from common import gtk, gobject
+
+class MyContainer(gtk.Container):
+    __gtype_name__ = 'MyContainer'
+    def __init__(self):
+        gtk.Container.__init__(self)
+        self.children = []
+        self.props = {}
+
+    def do_add(self, child):
+        child.set_parent(self)
+        self.children.append(child)
+        self.props[child] = '' # we just have one child property
+
+    def do_remove(self, child):
+        widget_was_visible = child.flags() & gtk.VISIBLE
+        child.unparent()
+        self.children.remove(child)
+        del self.props[child]
+
+        if widget_was_visible:
+            self.queue_resize()
+
+    def do_forall(self, internal, callback, data):
+        for child in self.children:
+            callback(child, data)
+
+    def do_set_child_property(self, child, property_id, value, pspec):
+        if pspec.name == 'dumb-prop':
+            self.props[child] = value
+
+    def do_get_child_property(self, child, property_id, pspec):
+        if pspec.name == 'dumb-prop':
+            return self.props[child]
+
+MyContainer.install_child_property(1,
+                                   ('dumb_prop',
+                                    gobject.TYPE_STRING,
+                                    'Dumb Prop',
+                                    'Dumb Property for testing purposes',
+                                    '', gobject.PARAM_READWRITE))
+    
+class ContainerTest(unittest.TestCase):
+
+    def testChildProperties(self):
+        obj = MyContainer()
+        label = gtk.Label()
+        obj.add(label)
+        v = 'dumb value'
+        obj.child_set_property(label, 'dumb_prop', v)
+        self.assertEqual(v, obj.child_get_property(label, 'dumb_prop'))
+
+if __name__ == '__main__':
+    unittest.main()