def test_insert_or_update_task_update(tmpdir):
temp = tmpdir.mkdir("tmp")
database.DBNAME = os.path.join(temp, 'task.db')
test_name = "TEST"
test_time = 10
test_time_update = 11
expected = (1, 'TEST', 11)
database.insert_or_update_task(test_name, test_time)
database.insert_or_update_task(test_name, test_time_update)
# Now check database
check_path = os.path.join(temp, "task.db")
check_db = sqlite3.connect(check_path)
check_table = database.table_name()
check_cmd = ("""SELECT id, task_name, task_time FROM """ + check_table)
with check_db:
cursor = check_db.cursor()
cursor.execute(check_cmd)
res = cursor.fetchall()
check_db.close()
assert res[0] == expected
def test_task_exists(tmpdir):
temp = tmpdir.mkdir("tmp")
database.DBNAME = os.path.join(temp, 'task.db')
test_name = "TEST"
test_time = 10
name = database.table_name()
test_db = sqlite3.connect(database.DBNAME)
add_table_cmd = ("""CREATE TABLE """ + name + """ (id INTEGER PRIMARY KEY,
task_name text, task_time, int)""")
add_task_cmd = ("""INSERT INTO """ + name +
"""(task_name, task_time) VALUES(?, ?)""")
with test_db:
cursor = test_db.cursor()
cursor.execute(add_table_cmd)
cursor.execute(add_task_cmd, (test_name, test_time))
test_db.close()
res = database.task_exists(test_name)
assert res
def test_table_exists_false(tmpdir):
temp = tmpdir.mkdir("tmp")
database.DBNAME = os.path.join(temp, 'task.db')
name = database.table_name()
test_db = sqlite3.connect(database.DBNAME)
test_db.close()
assert not database.table_exists(name)
def test_ensure_table_exists(tmpdir):
temp = tmpdir.mkdir("tmp")
database.DBNAME = os.path.join(temp, 'task.db')
name = database.table_name()
database.ensure_table_exists(name)
cmd = "SELECT name FROM sqlite_master WHERE type='table'"
test_db = sqlite3.connect(database.DBNAME)
with test_db:
cursor = test_db.cursor()
cursor.execute(cmd)
res = cursor.fetchall()
test_db.close()
assert len(res) > 0
def test_store_task(tmpdir):
temp = tmpdir.mkdir("tmp")
database.DBNAME = os.path.join(temp, 'task.db')
test_data = database.TaskData('TEST', 10)
expected = (1, 'TEST', 10)
database.store_task(test_data)
# Now check database
check_path = os.path.join(temp, "task.db")
check_db = sqlite3.connect(check_path)
check_table = database.table_name()
check_cmd = ("""SELECT id, task_name, task_time FROM """ + check_table)
with check_db:
cursor = check_db.cursor()
cursor.execute(check_cmd)
res = cursor.fetchall()
check_db.close()
assert res[0] == expected