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 == []