def test_deleting_lists(self): added_data = self.initialize_list() added_data.pop(3) added_data.pop(1) db_functions.delete_list(2, db_name=self.db_name) db_functions.delete_list(4, db_name=self.db_name) retrieved_data = db_functions.get_all_lists(self.db_name) self.check_lists_equal(added_data, retrieved_data)
def get_lists(): todo_lists = db_functions.get_all_lists() xml_todo_lists = ET.Element('todo_lists') for todo_list in todo_lists: xml_todo_list = ET.SubElement(xml_todo_lists, 'todo_list') for field, value in todo_list.items(): ET.SubElement(xml_todo_list, field).text = str(value) if value is not None else '' return xml_todo_lists
def get_todo_list(): todo_lists = db_functions.get_all_lists() xml_todo_lists = ET.Element('todo_lists') for todo_list in todo_lists: xml_todo_list = ET.SubElement(xml_todo_lists, 'todo_list') for field, value in todo_list.items(): ET.SubElement(xml_todo_list, field).text = str(value) if value is not None else '' xml_todo_lists = ET.ElementTree(xml_todo_lists) temp_file = BytesIO() xml_todo_lists.write(temp_file, encoding='UTF-8', xml_declaration=True) response.content_type = 'text/xml' return temp_file.getvalue()
def test_updating_lists(self): added_data = self.initialize_list() updated_data = [ ('Updated header 1', None, None, None), (None, 'Updated name 2', 0, None), (None, None, None, None), ('Updated header 4', 'Updated name 4', 1, None), ] for i, updated in enumerate(updated_data): db_functions.update_list(i + 1, header=updated[0], author_name=updated[1], type=updated[2], db_name=self.db_name) for i in range(len(updated_data)): updated = () for j in range(len(updated_data[i])): updated += updated_data[i][j] if updated_data[i][ j] is not None else added_data[i][j], updated_data[i] = updated retrieved_data = db_functions.get_all_lists(self.db_name) self.check_lists_equal(updated_data, retrieved_data)
def test_adding_and_retrieving_lists(self): added_data = self.initialize_list() retrieved_data = db_functions.get_all_lists(self.db_name) self.check_lists_equal(added_data, retrieved_data)
def get_lists(): todo_lists = db_functions.get_all_lists() for i in range(len(todo_lists)): todo_lists[i]['created'] = todo_lists[i]['created'].isoformat() return todo_lists