Convert all python code to use four-space indents instead of eight-space tabs.
Signed-off-by: John Admanski <[email protected]>
git-svn-id: http://test.kernel.org/svn/autotest/trunk@1658 592f7852-d20e-0410-864c-8624ca9c26a4
diff --git a/migrate/migrate_unittest.py b/migrate/migrate_unittest.py
index ec4ece1..d33dccd 100644
--- a/migrate/migrate_unittest.py
+++ b/migrate/migrate_unittest.py
@@ -13,140 +13,140 @@
NUM_MIGRATIONS = 3
class DummyMigration(object):
- """\
- Dummy migration class that records all migrations done in a class
- varaible.
- """
+ """\
+ Dummy migration class that records all migrations done in a class
+ varaible.
+ """
- migrations_done = []
+ migrations_done = []
- def __init__(self, version):
- self.version = version
- self.name = '%03d_test' % version
+ def __init__(self, version):
+ self.version = version
+ self.name = '%03d_test' % version
- @classmethod
- def get_migrations_done(cls):
- return cls.migrations_done
+ @classmethod
+ def get_migrations_done(cls):
+ return cls.migrations_done
- @classmethod
- def clear_migrations_done(cls):
- cls.migrations_done = []
+ @classmethod
+ def clear_migrations_done(cls):
+ cls.migrations_done = []
- @classmethod
- def do_migration(cls, version, direction):
- cls.migrations_done.append((version, direction))
+ @classmethod
+ def do_migration(cls, version, direction):
+ cls.migrations_done.append((version, direction))
- def migrate_up(self, manager):
- self.do_migration(self.version, 'up')
- if self.version == 1:
- manager.create_migrate_table()
+ def migrate_up(self, manager):
+ self.do_migration(self.version, 'up')
+ if self.version == 1:
+ manager.create_migrate_table()
- def migrate_down(self, manager):
- self.do_migration(self.version, 'down')
+ def migrate_down(self, manager):
+ self.do_migration(self.version, 'down')
MIGRATIONS = [DummyMigration(n) for n in xrange(1, NUM_MIGRATIONS + 1)]
class TestableMigrationManager(migrate.MigrationManager):
- def __init__(self, database, migrations_dir=None):
- self.database = database
- self.migrations_dir = migrations_dir
- self.db_host = None
- self.db_name = None
- self.username = None
- self.password = None
+ def __init__(self, database, migrations_dir=None):
+ self.database = database
+ self.migrations_dir = migrations_dir
+ self.db_host = None
+ self.db_name = None
+ self.username = None
+ self.password = None
- def read_db_info(self):
- migrate.MigrationManager.read_db_info(self)
- self.db_name = 'test_' + self.db_name
+ def read_db_info(self):
+ migrate.MigrationManager.read_db_info(self)
+ self.db_name = 'test_' + self.db_name
- def get_migrations(self, minimum_version=None, maximum_version=None):
- minimum_version = minimum_version or 1
- maximum_version = maximum_version or len(MIGRATIONS)
- return MIGRATIONS[minimum_version-1:maximum_version]
+ def get_migrations(self, minimum_version=None, maximum_version=None):
+ minimum_version = minimum_version or 1
+ maximum_version = maximum_version or len(MIGRATIONS)
+ return MIGRATIONS[minimum_version-1:maximum_version]
class MigrateManagerTest(unittest.TestCase):
- config = global_config.global_config
- host = config.get_config_value(CONFIG_DB, 'host')
- db_name = 'test_' + config.get_config_value(CONFIG_DB, 'database')
- user = config.get_config_value(CONFIG_DB, 'user')
- password = config.get_config_value(CONFIG_DB, 'password')
+ config = global_config.global_config
+ host = config.get_config_value(CONFIG_DB, 'host')
+ db_name = 'test_' + config.get_config_value(CONFIG_DB, 'database')
+ user = config.get_config_value(CONFIG_DB, 'user')
+ password = config.get_config_value(CONFIG_DB, 'password')
- def do_sql(self, sql):
- self.con = MySQLdb.connect(host=self.host, user=self.user,
- passwd=self.password)
- self.con.autocommit(True)
- self.cur = self.con.cursor()
- try:
- self.cur.execute(sql)
- finally:
- self.con.close()
+ def do_sql(self, sql):
+ self.con = MySQLdb.connect(host=self.host, user=self.user,
+ passwd=self.password)
+ self.con.autocommit(True)
+ self.cur = self.con.cursor()
+ try:
+ self.cur.execute(sql)
+ finally:
+ self.con.close()
- def remove_db(self):
- self.do_sql('DROP DATABASE ' + self.db_name)
+ def remove_db(self):
+ self.do_sql('DROP DATABASE ' + self.db_name)
- def setUp(self):
- self.do_sql('CREATE DATABASE ' + self.db_name)
- try:
- self.manager = TestableMigrationManager(CONFIG_DB)
- except MySQLdb.OperationalError:
- self.remove_db()
- raise
- DummyMigration.clear_migrations_done()
+ def setUp(self):
+ self.do_sql('CREATE DATABASE ' + self.db_name)
+ try:
+ self.manager = TestableMigrationManager(CONFIG_DB)
+ except MySQLdb.OperationalError:
+ self.remove_db()
+ raise
+ DummyMigration.clear_migrations_done()
- def tearDown(self):
- self.remove_db()
+ def tearDown(self):
+ self.remove_db()
- def test_sync(self):
- self.manager.do_sync_db()
- self.assertEquals(self.manager.get_db_version(), NUM_MIGRATIONS)
- self.assertEquals(DummyMigration.get_migrations_done(),
- [(1, 'up'), (2, 'up'), (3, 'up')])
+ def test_sync(self):
+ self.manager.do_sync_db()
+ self.assertEquals(self.manager.get_db_version(), NUM_MIGRATIONS)
+ self.assertEquals(DummyMigration.get_migrations_done(),
+ [(1, 'up'), (2, 'up'), (3, 'up')])
- DummyMigration.clear_migrations_done()
- self.manager.do_sync_db(0)
- self.assertEquals(self.manager.get_db_version(), 0)
- self.assertEquals(DummyMigration.get_migrations_done(),
- [(3, 'down'), (2, 'down'), (1, 'down')])
+ DummyMigration.clear_migrations_done()
+ self.manager.do_sync_db(0)
+ self.assertEquals(self.manager.get_db_version(), 0)
+ self.assertEquals(DummyMigration.get_migrations_done(),
+ [(3, 'down'), (2, 'down'), (1, 'down')])
- def test_sync_one_by_one(self):
- for version in xrange(1, NUM_MIGRATIONS + 1):
- self.manager.do_sync_db(version)
- self.assertEquals(self.manager.get_db_version(),
- version)
- self.assertEquals(
- DummyMigration.get_migrations_done()[-1],
- (version, 'up'))
+ def test_sync_one_by_one(self):
+ for version in xrange(1, NUM_MIGRATIONS + 1):
+ self.manager.do_sync_db(version)
+ self.assertEquals(self.manager.get_db_version(),
+ version)
+ self.assertEquals(
+ DummyMigration.get_migrations_done()[-1],
+ (version, 'up'))
- for version in xrange(NUM_MIGRATIONS - 1, -1, -1):
- self.manager.do_sync_db(version)
- self.assertEquals(self.manager.get_db_version(),
- version)
- self.assertEquals(
- DummyMigration.get_migrations_done()[-1],
- (version + 1, 'down'))
+ for version in xrange(NUM_MIGRATIONS - 1, -1, -1):
+ self.manager.do_sync_db(version)
+ self.assertEquals(self.manager.get_db_version(),
+ version)
+ self.assertEquals(
+ DummyMigration.get_migrations_done()[-1],
+ (version + 1, 'down'))
- def test_null_sync(self):
- self.manager.do_sync_db()
- DummyMigration.clear_migrations_done()
- self.manager.do_sync_db()
- self.assertEquals(DummyMigration.get_migrations_done(), [])
+ def test_null_sync(self):
+ self.manager.do_sync_db()
+ DummyMigration.clear_migrations_done()
+ self.manager.do_sync_db()
+ self.assertEquals(DummyMigration.get_migrations_done(), [])
if __name__ == '__main__':
- unittest.main()
+ unittest.main()
