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