56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
import os
|
|
|
|
from Config import config
|
|
from Db import Db
|
|
|
|
|
|
class TestDb:
|
|
def testCheckTables(self):
|
|
db_path = "%s/zeronet.db" % config.data_dir
|
|
schema = {
|
|
"db_name": "TestDb",
|
|
"db_file": "%s/zeronet.db" % config.data_dir,
|
|
"map": {
|
|
"data.json": {
|
|
"to_table": {
|
|
"test": "test"
|
|
}
|
|
}
|
|
},
|
|
"tables": {
|
|
"test": {
|
|
"cols": [
|
|
["test_id", "INTEGER"],
|
|
["title", "TEXT"],
|
|
],
|
|
"indexes": ["CREATE UNIQUE INDEX test_id ON test(test_id)"],
|
|
"schema_changed": 1426195822
|
|
}
|
|
}
|
|
}
|
|
|
|
if os.path.isfile(db_path):
|
|
os.unlink(db_path)
|
|
db = Db(schema, db_path)
|
|
db.checkTables()
|
|
db.close()
|
|
|
|
# Verify tables
|
|
assert os.path.isfile(db_path)
|
|
db = Db(schema, db_path)
|
|
|
|
tables = [row["name"] for row in db.execute("SELECT name FROM sqlite_master WHERE type='table'")]
|
|
assert "keyvalue" in tables # To store simple key -> value
|
|
assert "json" in tables # Json file path registry
|
|
assert "test" in tables # The table defined in dbschema.json
|
|
|
|
# Verify test table
|
|
cols = [col["name"] for col in db.execute("PRAGMA table_info(test)")]
|
|
assert "test_id" in cols
|
|
assert "title" in cols
|
|
|
|
db.close()
|
|
|
|
# Cleanup
|
|
os.unlink(db_path)
|