def test_up_to_last(self):
self.assertEqual(
[
MigrationFile(2, '0002_test.py'),
MigrationFile(3, '0003_test.py')
],
self.migrations.get_migrations_to_up())
def test_down_to_2(self):
result = self.migrations.get_migrations_to_down(2)
self.assertEqual(
[
MigrationFile(3, '0003_test.py'),
MigrationFile(2, '0002_test.py')
],
result)
def test_get_migration_files(self, mock_listdir):
mock_listdir.return_value = [
'0002_test.py', 'xxx', 'some.py', '0003_wrong name.py',
'0003_test.pyc', '0001_first.py']
result = MockMigrations().get_migration_files()
self.assertEqual(
[
MigrationFile(1, '0001_first.py'),
MigrationFile(2, '0002_test.py')
],
result)
def get_migrations(from_id, to_id=None):
if not to_id:
to_id = from_id + 1
return [
MigrationFile(id=m, filename='{:04}_test.py'.format(m))
for m in range(from_id, to_id + 1)
]
def test_get_unregistered_migrations(self):
migrations = MockMigrations()
migrations.get_migration_files = MagicMock(
return_value=get_migrations(1, 2)
)
mock_find_one = lambda m: m['filename'].startswith('0001')
migrations.collection.find_one.side_effect = mock_find_one
result = migrations.get_unregistered_migrations()
self.assertEqual(
[
MigrationFile(2, '0002_test.py')
],
result)
def test_str(self):
self.assertTrue(str(MigrationFile(1, '0001_test.py')).startswith('1'))
def test_as_dict(self):
self.assertEqual(
{
'id': 1,
'filename': '0001_test.py'
}, MigrationFile(1, '0001_test.py').as_dict())
def test_id_validate(self):
self.assertEqual(1, MigrationFile.validate_id('0001'))
self.assertEqual(None, MigrationFile.validate_id('a'))
def test_name_normalize(self):
self.assertEqual('test_it_', MigrationFile.normalize_name('test it!'))
def test_up_to_2(self):
self.assertEqual(
[
MigrationFile(2, '0002_test.py')
],
self.migrations.get_migrations_to_up(2))