[SQL-CVS] r2174 - SQLObject/trunk/sqlobject/versioning/test

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

Author: novalis
Date: 2007-01-04 13:39:49 -0700 (Thu, 04 Jan 2007)
New Revision: 2174

Modified:
   SQLObject/trunk/sqlobject/versioning/test/test_version.py
Log:
clean up before tests

Modified: SQLObject/trunk/sqlobject/versioning/test/test_version.py
===================================================================
--- SQLObject/trunk/sqlobject/versioning/test/test_version.py	2007-01-04 19:44:56 UTC (rev 2173)
+++ SQLObject/trunk/sqlobject/versioning/test/test_version.py	2007-01-04 20:39:49 UTC (rev 2174)
@@ -8,33 +8,30 @@
 
 
 def setup():
-    setupClass(MyClass)
-    setupClass(Base)
-    setupClass(Child)
-    setupClass(Government)
-    setupClass(Monarchy)
-    setupClass(VChild)
-    setupClass(HasForeign)
-
+    for cls in [MyClass, Base, Child, Government, Monarchy, VChild, HasForeign]:
+        setupClass(cls)
+        if hasattr(cls, 'versions'):
+            for version in cls.versions.versionClass.select():
+                version.destroySelf()
+    
 class MyClass(SQLObject):
-     name = StringCol()
-     versions = Versioning()
+    name = StringCol()
+    versions = Versioning()
 
 class Base(InheritableSQLObject):
-     name = StringCol()
-     value = IntCol(default=0)
-     versions = Versioning()
+    name = StringCol()
+    value = IntCol(default=0)
+    versions = Versioning()
 
 class Child(Base):
-     toy = StringCol()
+    toy = StringCol()
 
-
 class Government(InheritableSQLObject):
-     name = StringCol()
+    name = StringCol()
 
 class Monarchy(Government):
-     monarch = StringCol()
-     versions = Versioning()
+    monarch = StringCol()
+    versions = Versioning()
 
 class VChild(Base):
     weapon = StringCol()