Initial tables for polymorphic associations tests.

twistar/tests/mysql_config.py

@@ -16,6 +16,10 @@ def runInitTxn(txn):
         txn.execute("""CREATE TABLE favorite_colors (id INT AUTO_INCREMENT, name VARCHAR(255), PRIMARY KEY (id))""")
         txn.execute("""CREATE TABLE favorite_colors_users (favorite_color_id INT, user_id INT)""")
         txn.execute("""CREATE TABLE coltests (id INT AUTO_INCREMENT, `select` VARCHAR(255), `where` VARCHAR(255), PRIMARY KEY (id))""")
+        # poly tables
+        txn.execute("""CREATE TABLE child (id INT AUTO_INCREMENT, `name` VARCHAR(255), parent_id INT, parent_type VARCHAR(32))""")
+        txn.execute("""CREATE TABLE mother (id INT AUTO_INCREMENT, name VARCHAR(255))""")
+        txn.execute("""CREATE TABLE father (id INT AUTO_INCREMENT, name VARCHAR(255))""")
     return CONNECTION.runInteraction(runInitTxn)
 
 
@@ -27,5 +31,9 @@ def runTearDownDB(txn):
         txn.execute("DROP TABLE favorite_colors")
         txn.execute("DROP TABLE favorite_colors_users")
         txn.execute("DROP TABLE coltests")
+        # poly tables
+        txn.execute("DROP TABLE child")
+        txn.execute("DROP TABLE mother")
+        txn.execute("DROP TABLE father")
     return CONNECTION.runInteraction(runTearDownDB)
 

twistar/tests/utils.py

@@ -28,5 +28,15 @@ class FakeObject(DBObject):
 class Coltest(DBObject):
     pass
 
+class Child(DBObject):
+    BELONGSTO = ['parent']
+
+class Mother(DBObject):
+    HAS_MANY = ['childs']
+
+class Father(DBObject):
+    HAS_MANY = ['childs']
+
 Registry.register(Picture, User, Avatar, FakeObject, FavoriteColor)
+Registry.register(Pen, Mother, Father)