From 84c1cd7c52e7a07f2e0605433f255d946fe2737a Mon Sep 17 00:00:00 2001 From: tilly-Q Date: Wed, 31 Jul 2013 20:50:38 -0400 Subject: [PATCH] I added a few more unitests in this commit. It now confirms that even after mi- -gration, there is only one of each Foundation object. --- mediagoblin/tests/test_sql_migrations.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/mediagoblin/tests/test_sql_migrations.py b/mediagoblin/tests/test_sql_migrations.py index a004aa43..3d67fdf6 100644 --- a/mediagoblin/tests/test_sql_migrations.py +++ b/mediagoblin/tests/test_sql_migrations.py @@ -59,7 +59,8 @@ class Level1(Base1): SET1_MODELS = [Creature1, Level1] FOUNDATIONS = {Creature1:[{'name':u'goblin','num_legs':2,'is_demon':False}, - {'name':u'cerberus','num_legs':4,'is_demon':True}]} + {'name':u'cerberus','num_legs':4,'is_demon':True}] + } SET1_MIGRATIONS = {} @@ -790,6 +791,15 @@ def test_set1_to_set3(): # Now check to see if stuff seems to be in there. session = Session() + + + # Start with making sure that the foundations did not run again + assert session.query(Creature3).filter_by( + name=u'goblin').count() == 1 + assert session.query(Creature3).filter_by( + name=u'cerberus').count() == 1 + + # Then make sure the models have been migrated correctly creature = session.query(Creature3).filter_by( name=u'centipede').one() assert creature.num_limbs == 100.0 -- 2.25.1