diff options
Diffstat (limited to 'migrate/tests/versioning/test_repository.py')
-rw-r--r-- | migrate/tests/versioning/test_repository.py | 65 |
1 files changed, 32 insertions, 33 deletions
diff --git a/migrate/tests/versioning/test_repository.py b/migrate/tests/versioning/test_repository.py index 7065368..6845a0e 100644 --- a/migrate/tests/versioning/test_repository.py +++ b/migrate/tests/versioning/test_repository.py @@ -22,39 +22,39 @@ class TestRepository(fixture.Pathed): repo = Repository.create(path, name) config_path = repo.config.path manage_path = os.path.join(repo.path, 'manage.py') - self.assert_(repo) + self.assertTrue(repo) # Files should actually be created - self.assert_(os.path.exists(path)) - self.assert_(os.path.exists(config_path)) - self.assert_(os.path.exists(manage_path)) + self.assertTrue(os.path.exists(path)) + self.assertTrue(os.path.exists(config_path)) + self.assertTrue(os.path.exists(manage_path)) # Can't create it again: it already exists self.assertRaises(exceptions.PathFoundError, Repository.create, path, name) return path - + def test_load(self): """We should be able to load information about an existing repository""" # Create a repository to load path = self.test_create() repos = Repository(path) - self.assert_(repos) - self.assert_(repos.config) - self.assert_(repos.config.get('db_settings', 'version_table')) + self.assertTrue(repos) + self.assertTrue(repos.config) + self.assertTrue(repos.config.get('db_settings', 'version_table')) # version_table's default isn't none self.assertNotEquals(repos.config.get('db_settings', 'version_table'), 'None') - + def test_load_notfound(self): """Nonexistant repositories shouldn't be loaded""" path = self.tmp_repos() - self.assert_(not os.path.exists(path)) + self.assertTrue(not os.path.exists(path)) self.assertRaises(exceptions.InvalidRepositoryError, Repository, path) def test_load_invalid(self): """Invalid repos shouldn't be loaded""" - # Here, invalid=empty directory. There may be other conditions too, + # Here, invalid=empty directory. There may be other conditions too, # but we shouldn't need to test all of them path = self.tmp_repos() os.mkdir(path) @@ -78,23 +78,23 @@ class TestVersionedRepository(fixture.Pathed): self.assertEqual(repos.latest, 0) # repos.latest isn't an integer, but a VerNum # (so we can't just assume the following tests are correct) - self.assert_(repos.latest >= 0) - self.assert_(repos.latest < 1) + self.assertTrue(repos.latest >= 0) + self.assertTrue(repos.latest < 1) # Create a script and test again repos.create_script('') self.assertEqual(repos.latest, 1) - self.assert_(repos.latest >= 0) - self.assert_(repos.latest >= 1) - self.assert_(repos.latest < 2) + self.assertTrue(repos.latest >= 0) + self.assertTrue(repos.latest >= 1) + self.assertTrue(repos.latest < 2) # Create a new script and test again repos.create_script('') self.assertEqual(repos.latest, 2) - self.assert_(repos.latest >= 0) - self.assert_(repos.latest >= 1) - self.assert_(repos.latest >= 2) - self.assert_(repos.latest < 3) + self.assertTrue(repos.latest >= 0) + self.assertTrue(repos.latest >= 1) + self.assertTrue(repos.latest >= 2) + self.assertTrue(repos.latest < 3) def test_timestmap_numbering_version(self): @@ -105,13 +105,12 @@ class TestVersionedRepository(fixture.Pathed): self.assertEqual(repos.latest, 0) # repos.latest isn't an integer, but a VerNum # (so we can't just assume the following tests are correct) - self.assert_(repos.latest >= 0) - self.assert_(repos.latest < 1) + self.assertTrue(repos.latest >= 0) + self.assertTrue(repos.latest < 1) # Create a script and test again now = int(datetime.utcnow().strftime('%Y%m%d%H%M%S')) repos.create_script('') - print repos.latest self.assertEqual(repos.latest, now) def test_source(self): @@ -134,9 +133,9 @@ class TestVersionedRepository(fixture.Pathed): """Repository.version() (no params) returns the latest version""" repos = Repository(self.path_repos) repos.create_script('') - self.assert_(repos.version(repos.latest) is repos.version()) - self.assert_(repos.version() is not None) - + self.assertTrue(repos.version(repos.latest) is repos.version()) + self.assertTrue(repos.version() is not None) + def test_changeset(self): """Repositories can create changesets properly""" # Create a nonzero-version repository of empty scripts @@ -152,9 +151,9 @@ class TestVersionedRepository(fixture.Pathed): uniq = list() # Changesets are iterable for version, change in changeset: - self.assert_(isinstance(change, BaseScript)) + self.assertTrue(isinstance(change, BaseScript)) # Changes aren't identical - self.assert_(id(change) not in uniq) + self.assertTrue(id(change) not in uniq) uniq.append(id(change)) return changeset @@ -201,17 +200,17 @@ class TestVersionedRepository(fixture.Pathed): self.assertEqual(cs.end, 0) check_changeset((10, 5), 5) check_changeset((5, 0), 5) - + def test_many_versions(self): """Test what happens when lots of versions are created""" repos = Repository(self.path_repos) - for i in range(1001): + for i in range(1001): repos.create_script('') # since we normally create 3 digit ones, let's see if we blow up - self.assert_(os.path.exists('%s/versions/1000.py' % self.path_repos)) - self.assert_(os.path.exists('%s/versions/1001.py' % self.path_repos)) + self.assertTrue(os.path.exists('%s/versions/1000.py' % self.path_repos)) + self.assertTrue(os.path.exists('%s/versions/1001.py' % self.path_repos)) + - # TODO: test manage file # TODO: test changeset |