def setUp(self):
if sys.platform == "win32":
self.schema_file1 = path.join(TESTDIR, "test_schema_vsimple.xml")
assert (os_file_exists(self.schema_file1))
self.schema_file2 = path.join(TESTDIR, "test_schema_vsimple2.xml")
assert (os_file_exists(self.schema_file2))
else:
self.schema_file1 = os.path.join(
os.environ["PYTHONPATH"], "test_misc/test_schema_vsimple.xml")
self.schema_file2 = os.path.join(
os.environ["PYTHONPATH"], "test_mis/test_schema_vsimple2.xml")
schema_execute(self.schema_file1)
datarows = {'tbl_col_name': [], 'tbl_rows': []}
schema_data_get(self.schema_file1, 'workout', datarows)
self.database1 = Database('fitness')
with self.database1:
tbl_rows_insert(self.database1, 'workout',
datarows['tbl_col_name'], datarows['tbl_rows'])
schema_execute(self.schema_file2)
schema_data_get(self.schema_file2, 'food', datarows)
self.database2 = Database('diet')
with self.database2:
tbl_rows_insert(self.database2, 'food', datarows['tbl_col_name'],
datarows['tbl_rows'])
def test_db_create_from_schema(self):
schema_execute(self.schema_file)
database = Database('fitness', True)
with database:
self.assertTrue(database.tbl_exists('workout'))
database = Database('diet')
with database:
self.assertTrue(database.tbl_exists('food'))
database = Database('diet', True)
with database:
self.assertTrue(database.tbl_exists('meals'))
def test_tbl_cols_get(self):
schema_execute(self.schema_file)
database = Database('fitness', remove_flag=True)
with database:
self.assertListEqual([('date', 'datetime'), ('type', 'text')],
tbl_cols_get(database, 'workout'))
database = Database('diet', remove_flag=True)
with database:
self.assertListEqual([('name', 'text'), ('calories', 'integer')],
tbl_cols_get(database, 'food'))
self.assertListEqual([('type', 'text'), ('time', 'datetime')],
tbl_cols_get(database, 'meals'))
def test_tbl_col_add(self):
schema_execute(self.schema_file)
database = Database('fitness')
with database:
tbl_col_add(database, 'workout', 'foobar', 'text')
database = Database('fitness', remove_flag=True)
with database:
self.assertListEqual([('date', 'datetime'), ('type', 'text'),
('foobar', 'text')],
tbl_cols_get(database, 'workout'))
# this is there to force the delete of the 2nd db created but does
database = Database('diet', remove_flag=True)
with database:
pass
def test_tbl_rows_insert_from_schema(self):
schema_execute(self.schema_file)
database = Database('fitness')
with database:
tbl_rows_insert_from_schema(database, self.schema_file, 'workout')
database = Database('fitness', remove_flag=True)
with database:
tbl_col_name, tbl_rows, _ = tbl_rows_get(database, 'workout')
self.assertListEqual(tbl_rows,
[[250772, 'cycling'], [260772, 'rowing']])
# this is there to force the delete of the 2nd db created but does
database = Database('diet', True)
with database:
pass
def setUp(self):
self.schema_file = path.join(TESTDIR, "test_schema_simple.xml")
assert (os_file_exists(self.schema_file))
schema_execute(self.schema_file, insert=True)