def test_db_query(reset_db, store_example):
# Tests that a single row object is retrieved
name, start, end_time, time_left, max_time = store_example
result = DataManager.query("SELECT * FROM program_log", single=True)
assert len(result) == 6
assert result["name"] == name
assert result["start_time"] == start
assert result["end_time"] == end_time
assert result["time_left"] == time_left
assert result["max_time"] == max_time
# Tests that an array of row objects is returned
result = DataManager.query("SELECT * FROM program_log;")
assert len(result) == 1
def test_store_new_valid(reset_db, default_init):
# Test when TrackedProgram is valid
DataManager.store_new(default_init)
result = DataManager.query("SELECT * FROM program_log", single=True)
assert result['start_time'] == default_init.start_time
assert result['end_time'] == default_init.end_time
assert result['time_left'] == default_init.time_left
assert result['max_time'] == default_init.max_time
assert result['name'] == default_init.name
assert result['id'] is 1
def test_to_obj_valid(reset_db, store_example):
# Tests that a TrackedProgram object is initialized correctly with valid values
name, start, end_time, time_left, max_time = store_example
result = DataManager.query("SELECT * FROM program_log", single=True)
obj = DataManager.to_obj(result)
assert obj.db_id == 1
assert obj.name == name
assert obj.start_time == start
assert obj.end_time == end_time
assert obj._time_left == time_left
assert obj.max_time == max_time
def test_store_many(reset_db, default_init, row_num):
DataManager.store_many(default_init, default_init)
result = DataManager.query("SELECT * FROM program_log")
result = result[row_num]
assert result['start_time'] == default_init.start_time
assert result['end_time'] == default_init.end_time
assert result['time_left'] == default_init.time_left
assert result['max_time'] == default_init.max_time
assert result['name'] == default_init.name
assert result['id'] is row_num + 1
def test_store_many_none(reset_db, default_init):
# Testing for bug where if no object is given to store_many it shows that an error occured with the db
DataManager.store_many()
result = DataManager.query("SELECT * FROM program_log")
assert result == []