From 0b78f3d931301d9a66b6778b87ebbd4fc40d0777 Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Tue, 30 Mar 2021 08:42:44 +1100 Subject: [PATCH] Add unit testing for migrations --- InvenTree/users/test_migrations.py | 38 ++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 InvenTree/users/test_migrations.py diff --git a/InvenTree/users/test_migrations.py b/InvenTree/users/test_migrations.py new file mode 100644 index 0000000000..1f10d041c2 --- /dev/null +++ b/InvenTree/users/test_migrations.py @@ -0,0 +1,38 @@ +""" +Unit tests for the user model database migrations +""" + +from django_test_migrations.contrib.unittest_case import MigratorTestCase + +from InvenTree import helpers + + +class TestForwardMigrations(MigratorTestCase): + """ + Test entire schema migration sequence for the users app + """ + + migrate_from = ('users', helpers.getOldestMigrationFile('users')) + migrate_to = ('users', helpers.getNewestMigrationFile('users')) + + def prepare(self): + + User = self.old_state.apps.get_model('auth', 'user') + + User.objects.create( + username='fred', + email='fred@fred.com', + password='password' + ) + + User.objects.create( + username='brad', + email='brad@fred.com', + password='password' + ) + + def test_users_exist(self): + + User = self.new_state.apps.get_model('auth', 'user') + + self.assertEqual(User.objects.count(), 2)