diff -r 000000000000 -r b97547f5f1fa common/test/unittest_migration.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/common/test/unittest_migration.py Wed Nov 05 15:52:50 2008 +0100 @@ -0,0 +1,126 @@ +"""cubicweb.common.migration unit tests""" + +from os.path import abspath +from logilab.common.testlib import TestCase, unittest_main + +from cubicweb.devtools import TestServerConfiguration +from cubicweb.devtools.apptest import TestEnvironment + +from cubicweb.cwconfig import CubicWebConfiguration +from cubicweb.common.migration import migration_files, filter_scripts + + +class Schema(dict): + def has_entity(self, e_type): + return self.has_key(e_type) + +SMIGRDIR = abspath('data/server_migration') + '/' +TMIGRDIR = abspath('data/migration') + '/' + +class MigrTestConfig(TestServerConfiguration): + verbosity = 0 + def migration_scripts_dir(cls): + return SMIGRDIR + + def cube_migration_scripts_dir(cls, cube): + return TMIGRDIR + +class MigrationToolsTC(TestCase): + def setUp(self): + self.config = MigrTestConfig('data') + from yams.schema import Schema + self.config.load_schema = lambda expand_cubes=False: Schema('test') + + def test_migration_files_base(self): + self.assertListEquals(migration_files(self.config, [('cubicweb', (2,3,0), (2,4,0)), + ('TEMPLATE', (0,0,2), (0,0,3))]), + [SMIGRDIR+'bootstrapmigration_repository.py', + TMIGRDIR+'0.0.3_Any.py']) + self.assertListEquals(migration_files(self.config, [('cubicweb', (2,4,0), (2,5,0)), + ('TEMPLATE', (0,0,2), (0,0,3))]), + [SMIGRDIR+'bootstrapmigration_repository.py', + SMIGRDIR+'2.5.0_Any.sql', + TMIGRDIR+'0.0.3_Any.py']) + self.assertListEquals(migration_files(self.config, [('cubicweb', (2,5,0), (2,6,0)), + ('TEMPLATE', (0,0,3), (0,0,4))]), + [SMIGRDIR+'bootstrapmigration_repository.py', + SMIGRDIR+'2.6.0_Any.sql', + TMIGRDIR+'0.0.4_Any.py']) + +## def test_migration_files_overlap(self): +## self.assertListEquals(migration_files(self.config, (2,4,0), (2,10,2), +## (0,0,2), (0,1,2)), +## [SMIGRDIR+'bootstrapmigration_repository.py', +## TMIGRDIR+'0.0.3_Any.py', +## TMIGRDIR+'0.0.4_Any.py', +## SMIGRDIR+'2.4.0_2.5.0_Any.sql', +## SMIGRDIR+'2.5.1_2.6.0_Any.sql', +## TMIGRDIR+'0.1.0_Any.py', +## TMIGRDIR+'0.1.0_common.py', +## TMIGRDIR+'0.1.0_repository.py', +## TMIGRDIR+'0.1.2_Any.py', +## SMIGRDIR+'2.10.1_2.10.2_Any.sql']) + + def test_migration_files_for_mode(self): + from cubicweb.server.migractions import ServerMigrationHelper + self.assertIsInstance(self.config.migration_handler(), ServerMigrationHelper) + from cubicweb.common.migration import MigrationHelper + config = CubicWebConfiguration('data') + config.verbosity = 0 + self.assert_(not isinstance(config.migration_handler(), ServerMigrationHelper)) + self.assertIsInstance(config.migration_handler(), MigrationHelper) + config = self.config + config.__class__.name = 'twisted' + self.assertListEquals(migration_files(config, [('TEMPLATE', (0,0,4), (0,1,0))]), + [TMIGRDIR+'0.1.0_common.py', + TMIGRDIR+'0.1.0_web.py']) + config.__class__.name = 'repository' + self.assertListEquals(migration_files(config, [('TEMPLATE', (0,0,4), (0,1,0))]), + [SMIGRDIR+'bootstrapmigration_repository.py', + TMIGRDIR+'0.1.0_Any.py', + TMIGRDIR+'0.1.0_common.py', + TMIGRDIR+'0.1.0_repository.py']) + config.__class__.name = 'all-in-one' + self.assertListEquals(migration_files(config, [('TEMPLATE', (0,0,4), (0,1,0))]), + [SMIGRDIR+'bootstrapmigration_repository.py', + TMIGRDIR+'0.1.0_Any.py', + TMIGRDIR+'0.1.0_common.py', + TMIGRDIR+'0.1.0_repository.py', + TMIGRDIR+'0.1.0_web.py']) + config.__class__.name = 'repository' + + def test_filter_scripts(self): + self.assertListEquals(filter_scripts(self.config, SMIGRDIR, (2,4,0), (2,5,0)), + [((2, 5, 0), SMIGRDIR+'2.5.0_Any.sql')]) + self.assertListEquals(filter_scripts(self.config, SMIGRDIR, (2,4,0), (2,6,0)), + [((2, 5, 0), SMIGRDIR+'2.5.0_Any.sql'), + ((2, 6, 0), SMIGRDIR+'2.6.0_Any.sql')]) + self.assertListEquals(filter_scripts(self.config, SMIGRDIR, (2,5,0), (2,5,1)), + []) + self.assertListEquals(filter_scripts(self.config, SMIGRDIR, (2,5,0), (2,6,0)), + [((2, 6, 0), SMIGRDIR+'2.6.0_Any.sql')]) + self.assertListEquals(filter_scripts(self.config, SMIGRDIR, (2,5,0), (2,10,2)), + [((2, 6, 0), SMIGRDIR+'2.6.0_Any.sql'), + ((2, 10, 2), SMIGRDIR+'2.10.2_Any.sql')]) + self.assertListEquals(filter_scripts(self.config, SMIGRDIR, (2,5,1), (2,6,0)), + [((2, 6, 0), SMIGRDIR+'2.6.0_Any.sql')]) + self.assertListEquals(filter_scripts(self.config, SMIGRDIR, (2,5,1), (2,10,2)), + [((2, 6, 0), SMIGRDIR+'2.6.0_Any.sql'), + ((2, 10, 2), SMIGRDIR+'2.10.2_Any.sql')]) + + +from cubicweb.devtools import ApptestConfiguration, init_test_database, cleanup_sqlite + +class BaseCreationTC(TestCase): + + def test_db_creation(self): + """make sure database can be created""" + config = ApptestConfiguration('data') + source = config.sources()['system'] + self.assertEquals(source['db-driver'], 'sqlite') + cleanup_sqlite(source['db-name'], removecube=True) + init_test_database(driver=source['db-driver'], config=config) + + +if __name__ == '__main__': + unittest_main()