SanctuaryCommander/test/db_interface_test.py

30 lines
791 B
Python

from utilities.database.sqlite import SQLite
from pprint import pprint
test_db = SQLite("testdb.db")
test_db.query("CREATE TABLE vegetables (veggie_id INTEGER PRIMARY KEY AUTOINCREMENT, veggie_name TEXT NOT NULL)")
test_db.insert("vegetables", ["veggie_name"], ["Green Beans"])
test_db.insert("vegetables", ["veggie_name"], ["Peas"])
test_db.insert("vegetables", ["veggie_name"], ["Corn"])
test_db.insert("vegetables", ["veggie_name"], ["Stephen Hawking"])
pprint(test_db.select("vegetables"))
pprint(test_db.update("vegetables", ["veggie_name"], ["Snow Peas", 2], [
{
"name": "veggie_id",
"compare": "="
}
]))
pprint(test_db.delete("vegetables", [3], [
{
"name": "veggie_id",
"compare": ">="
}
]))
pprint(test_db.select("vegetables"))