def get_taskdb(request, filename):
if filename == 'pending.data':
taskstr = Task.serialize('pending')
elif filename == 'completed.data':
taskstr = Task.serialize('completed')
elif filename == 'undo.data':
taskstr = Undo.serialize()
else:
return HttpResponseNotFound()
response = HttpResponse(taskstr, mimetype='text/plain')
response['Content-Length'] = len(taskstr)
return response
def test_taskdb_PUT_completed(self):
self._create_user_and_login()
data = open(os.path.join(TASK_DATA, 'completed.data'), 'r').read()
response = self.client.put('/taskdb/completed.data',
content_type='text/plain',
data=data)
self.assertEqual(response.status_code, 200)
self.assertEqual(list(Undo.objects.all()), [])
self.assertEqual(Task.serialize('completed'), data)
def test_taskdb_PUT_pending(self):
self._create_user_and_login()
data = open(os.path.expanduser('~/.task/pending.data'), 'r').read()
response = self.client.put('/taskdb/pending.data',
content_type='text/plain',
data=data)
self.assertEqual(response.status_code, 200)
self.assertEqual(list(Undo.objects.all()), [])
self.assertEqual(Task.serialize('pending'), data)
def test_taskdb_PUT_completed(self):
self._create_user_and_login()
data = open(os.path.join(TASK_DATA, 'completed.data'), 'r').read()
response = self.client.put('/taskdb/completed.data',
content_type='text/plain',
data=data)
self.assertEqual(response.status_code, 200)
self.assertEqual(list(Undo.objects.all()), [])
self.assertEqual(Task.serialize('completed'), data)
