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