class TestVersionedSqlite(unittest.TestCase):
def setUp(self):
self._db = VersionedSqlite(":memory:")
def tearDown(self):
self._db.close()
del (self._db)
def test_migrations(self):
self.assertTrue(self._db.get_schema_version("main") == -1)
migrations = (self.schema0, self.schema1)
self._db.setup("main", migrations)
self.assertTrue(self._db.get_schema_version("main") == len(migrations) - 1)
def schema0(self, curr):
curr.execute(
"""
CREATE TABLE main (
col1 VARCHAR
)"""
)
return True
def schema1(self, curr):
curr.execute(
"""
ALTER TABLE main ADD COLUMN col2 INT
"""
)
return True
def setUp(self):
self._db = VersionedSqlite(":memory:")
