Used assertRaisesMessage() to test MigrationLoader.get_migration_by_prefix()'s error messages.
@@ -179,9 +179,11 @@ class LoaderTests(TestCase):
migration_loader.get_migration_by_prefix("migrations", "0001").name,
"0001_initial",
)
- with self.assertRaises(AmbiguityError):
+ msg = "There is more than one migration for 'migrations' with the prefix '0'"
+ with self.assertRaisesMessage(AmbiguityError, msg):
migration_loader.get_migration_by_prefix("migrations", "0")
- with self.assertRaises(KeyError):
+ msg = "There no migrations for 'migrations' with the prefix 'blarg'"
+ with self.assertRaisesMessage(KeyError, msg):
migration_loader.get_migration_by_prefix("migrations", "blarg")
def test_load_import_error(self):
