def test_migrator(transactional_db):
"""We only need this test for coverage."""
migrator = Migrator()
old_state = migrator.before(('main_app', None))
new_state = migrator.after(('main_app', '0001_initial'))
assert isinstance(old_state, ProjectState)
assert isinstance(new_state, ProjectState)
assert migrator.reset() is None
class MigratorTestCase(TransactionTestCase):
"""Used when using raw ``unitest`` library for test."""
database_name: ClassVar[Optional[str]] = None
old_state: ProjectState
new_state: ProjectState
#: Part of the end-user API. Used to tell what migrations we are using.
migrate_from: ClassVar[_Migration]
migrate_to: ClassVar[_Migration]
def __init__(self, *args, **kwargs) -> None:
"""Initializes our :class:`Migrator` instance."""
super().__init__(*args, **kwargs)
def setUp(self) -> None:
"""
Regular ``unittest`` styled setup case.
What it does?
- It starts with defining the initial migration state
- Then it allows to run custom method
to prepare some data before the migration will happen
- Then it applies the migration and saves all states
"""
super().setUp()
self._pre_setup()
self._migrator = Migrator(self.database_name)
self.old_state = self._migrator.before(self.migrate_from)
self.prepare()
self.new_state = self._migrator.after(self.migrate_to)
def prepare(self) -> None:
"""
Part of the end-user API.
Used to prepare some data before the migration process.
"""
def tearDown(self) -> None:
"""Used to clean mess up after each test."""
self._migrator.reset()
super().tearDown()