def test_all_migrations(self):
app = migration.get_app("fakeapp")
self.assertEqual(
{app: {
"0001_spam": migration.get_migration(app, "0001_spam"),
"0002_eggs": migration.get_migration(app, "0002_eggs"),
}},
migration.all_migrations(),
)
def test_get_migration(self):
app = self.create_test_app()
# Can't use vanilla import, modules beginning with numbers aren't in grammar
M1 = __import__("fakeapp.migrations.0001_spam", {}, {}, ['Migration']).Migration
M2 = __import__("fakeapp.migrations.0002_eggs", {}, {}, ['Migration']).Migration
self.assertEqual(M1, migration.get_migration(app, "0001_spam"))
self.assertEqual(M2, migration.get_migration(app, "0002_eggs"))
self.assertRaises(ValueError, migration.get_migration, app, "0001_jam")
def test_all_migrations(self):
app = migration.get_app("fakeapp")
self.assertEqual(
{app: {
"0001_spam": migration.get_migration(app, "0001_spam"),
"0002_eggs": migration.get_migration(app, "0002_eggs"),
"0003_alter_spam": migration.get_migration(app, "0003_alter_spam"),
}},
migration.all_migrations(),
)
def test_get_migration(self):
app = self.create_test_app()
# Can't use vanilla import, modules beginning with numbers aren't in grammar
M1 = __import__("fakeapp.migrations.0001_spam", {}, {}, ['Migration']).Migration
M2 = __import__("fakeapp.migrations.0002_eggs", {}, {}, ['Migration']).Migration
self.assertEqual(M1, migration.get_migration(app, "0001_spam"))
self.assertEqual(M2, migration.get_migration(app, "0002_eggs"))
# Temporarily redirect sys.stdout during this, it whinges.
stdout, sys.stdout = sys.stdout, StringIO.StringIO()
try:
self.assertRaises((ImportError, ValueError), migration.get_migration, app, "0001_jam")
finally:
sys.stdout = stdout
def test_all_migrations(self):
app = migration.get_app("fakeapp")
otherapp = migration.get_app("otherfakeapp")
self.assertEqual(
{
app: {
"0001_spam":
migration.get_migration(app, "0001_spam"),
"0002_eggs":
migration.get_migration(app, "0002_eggs"),
"0003_alter_spam":
migration.get_migration(app, "0003_alter_spam"),
},
otherapp: {
"0001_first": migration.get_migration(
otherapp, "0001_first"),
"0002_second": migration.get_migration(
otherapp, "0002_second"),
"0003_third": migration.get_migration(
otherapp, "0003_third"),
},
},
migration.all_migrations(),
)
def test_all_migrations(self):
app = migration.get_app("fakeapp")
otherapp = migration.get_app("otherfakeapp")
self.assertEqual({
app: {
"0001_spam": migration.get_migration(app, "0001_spam"),
"0002_eggs": migration.get_migration(app, "0002_eggs"),
"0003_alter_spam": migration.get_migration(app, "0003_alter_spam"),
},
otherapp: {
"0001_first": migration.get_migration(otherapp, "0001_first"),
"0002_second": migration.get_migration(otherapp, "0002_second"),
"0003_third": migration.get_migration(otherapp, "0003_third"),
},
},
migration.all_migrations(),
)
