From 690b51faa7aa8723bb08639f21b60dacd5696fcd Mon Sep 17 00:00:00 2001
From: Christopher Allan Webber <cwebber@dustycloud.org>
Date: Sun, 29 Jan 2012 13:14:56 -0600
Subject: [PATCH] Closer to the end of this migration test...

---
 mediagoblin/tests/test_sql_migrations.py | 72 +++++++++++++++++++++++-
 1 file changed, 70 insertions(+), 2 deletions(-)

diff --git a/mediagoblin/tests/test_sql_migrations.py b/mediagoblin/tests/test_sql_migrations.py
index 57a83ec5..33580432 100644
--- a/mediagoblin/tests/test_sql_migrations.py
+++ b/mediagoblin/tests/test_sql_migrations.py
@@ -660,7 +660,7 @@ def test_set1_to_set3():
         'creature', metadata,
         autoload=True, autoload_with=db_conn.engine)
     assert set(creature_table.c.keys()) == set(
-        ['id', 'name', 'num_legs'])
+        ['id', 'name', 'num_limbs'])
     assert_col_type(creature_table.c.id, Integer)
     assert_col_type(creature_table.c.name, Unicode)
     assert creature_table.c.name.nullable is False
@@ -680,7 +680,7 @@ def test_set1_to_set3():
     assert creature_power_table.c.creature.nullable is False
     assert_col_type(creature_power_table.c.name, Unicode)
     assert_col_type(creature_power_table.c.description, Unicode)
-    assert_col_type(creature_power_table.c.hitpower, Integer)
+    assert_col_type(creature_power_table.c.hitpower, Float)
     assert creature_power_table.c.hitpower.nullable is False
 
     # Check the structure of the level table
@@ -704,10 +704,26 @@ def test_set1_to_set3():
     assert_col_type(level_exit_table.c.name, Unicode)
     assert_col_type(level_exit_table.c.from_level, Unicode)
     assert level_exit_table.c.from_level.nullable is False
+    assert level_exit_table.c.from_level.indexed is True
     assert_col_type(level_exit_table.c.to_level, Unicode)
     assert level_exit_table.c.to_level.nullable is False
+    assert level_exit_table.c.to_level.indexed is True
 
     # Now check to see if stuff seems to be in there.
+    creature = session.query(Creature1).filter_by(
+        name=u'centipede').one()
+    assert creature.num_legs == 100.0
+    assert creature.creature_powers == []
+
+    creature = session.query(Creature1).filter_by(
+        name=u'wolf').one()
+    assert creature.num_legs == 4.0
+    assert creature.creature_powers == []
+
+    creature = session.query(Creature1).filter_by(
+        name=u'wizardsnake').one()
+    assert creature.num_legs == 0.0
+    assert creature.creature_powers == []
 
     pass
 
@@ -741,4 +757,56 @@ def test_set1_to_set2_to_set3():
     # Migrate again
     # Make sure version matches expected again
     # Check all things in database match expected again
+
+    ##### Set2
+    # creature_table = Table(
+    #     'creature', metadata,
+    #     autoload=True, autoload_with=db_conn.engine)
+    # assert set(creature_table.c.keys()) == set(
+    #     ['id', 'name', 'num_legs'])
+    # assert_col_type(creature_table.c.id, Integer)
+    # assert_col_type(creature_table.c.name, Unicode)
+    # assert creature_table.c.name.nullable is False
+    # assert creature_table.c.name.index is True
+    # assert creature_table.c.name.unique is True
+    # assert_col_type(creature_table.c.num_legs, Integer)
+    # assert creature_table.c.num_legs.nullable is False
+
+    # # Check the CreaturePower table
+    # creature_power_table = Table(
+    #     'creature_power', metadata,
+    #     autoload=True, autoload_with=db_conn.engine)
+    # assert set(creature_power_table.c.keys()) == set(
+    #     ['id', 'creature', 'name', 'description', 'hitpower'])
+    # assert_col_type(creature_power_table.c.id, Integer)
+    # assert_col_type(creature_power_table.c.creature, Integer)
+    # assert creature_power_table.c.creature.nullable is False
+    # assert_col_type(creature_power_table.c.name, Unicode)
+    # assert_col_type(creature_power_table.c.description, Unicode)
+    # assert_col_type(creature_power_table.c.hitpower, Integer)
+    # assert creature_power_table.c.hitpower.nullable is False
+
+    # # Check the structure of the level table
+    # level_table = Table(
+    #     'level', metadata,
+    #     autoload=True, autoload_with=db_conn.engine)
+    # assert set(level_table.c.keys()) == set(
+    #     ['id', 'name', 'description'])
+    # assert_col_type(level_table.c.id, Unicode)
+    # assert level_table.c.id.primary_key is True
+    # assert_col_type(level_table.c.name, Unicode)
+    # assert_col_type(level_table.c.description, Unicode)
+
+    # # Check the structure of the level_exits table
+    # level_exit_table = Table(
+    #     'level_exit', metadata,
+    #     autoload=True, autoload_with=db_conn.engine)
+    # assert set(level_exit_table.c.keys()) == set(
+    #     ['id', 'name', 'from_level', 'to_level'])
+    # assert_col_type(level_exit_table.c.id, Integer)
+    # assert_col_type(level_exit_table.c.name, Unicode)
+    # assert_col_type(level_exit_table.c.from_level, Unicode)
+    # assert level_exit_table.c.from_level.nullable is False
+    # assert_col_type(level_exit_table.c.to_level, Unicode)
+
     pass
-- 
2.25.1