[SQL-CVS] r2237 - SQLObject/branches/0.8/sqlobject/inheritance/tests

[<sub...@co...>]

Author: phd
Date: 2007-01-23 10:11:12 -0700 (Tue, 23 Jan 2007)
New Revision: 2237

Added:
   SQLObject/branches/0.8/sqlobject/inheritance/tests/test_inheritance_tree.py
Log:
Added test_inheritance_tree.py.


Added: SQLObject/branches/0.8/sqlobject/inheritance/tests/test_inheritance_tree.py
===================================================================
--- SQLObject/branches/0.8/sqlobject/inheritance/tests/test_inheritance_tree.py	                        (rev 0)
+++ SQLObject/branches/0.8/sqlobject/inheritance/tests/test_inheritance_tree.py	2007-01-23 17:11:12 UTC (rev 2237)
@@ -0,0 +1,37 @@
+from sqlobject import *
+from sqlobject.inheritance import *
+from sqlobject.tests.dbtest import *
+
+########################################
+## Inheritance Tree
+########################################
+
+class Tree1(InheritableSQLObject):
+    aprop = StringCol(length=10)
+
+class Tree2(Tree1):
+    bprop = StringCol(length=10)
+
+class Tree3(Tree1):
+    cprop = StringCol(length=10)
+
+class Tree4(Tree2):
+    dprop = StringCol(length=10)
+
+class Tree5(Tree2):
+    eprop = StringCol(length=10)
+
+def test_tree():
+    setupClass([Tree1, Tree2, Tree3, Tree4, Tree5])
+
+    t1 = Tree1(aprop='t1')
+    t2 = Tree2(aprop='t2', bprop='t2')
+    t3 = Tree3(aprop='t3', cprop='t3')
+    t4 = Tree4(aprop='t4', bprop='t4', dprop='t4')
+    t5 = Tree5(aprop='t5', bprop='t5', eprop='t5')
+
+    # find just the t5 out of childs from Tree2
+    assert t5 == Tree1.select(Tree2.q.childName == 'Tree5')[0]
+
+    # t2,t4,t5 are all subclasses of Tree1 with t1 childName of 'Tree2'
+    assert list(Tree1.select(Tree1.q.childName == 'Tree2', orderBy="aprop")) == [t2, t4, t5]