From 87b47012dabbd784ac5c808c4080030f316043e1 Mon Sep 17 00:00:00 2001 From: Tomas Babej Date: Tue, 20 Jan 2015 23:36:07 +0100 Subject: [PATCH] Tests: Add tests for serializing annotations --- tasklib/tests.py | 21 +++++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) diff --git a/tasklib/tests.py b/tasklib/tests.py index 6407221..686e086 100644 --- a/tasklib/tests.py +++ b/tasklib/tests.py @@ -2,6 +2,7 @@ import datetime import itertools +import json import six import shutil import tempfile @@ -30,7 +31,7 @@ TASK_STANDARD_ATTRS = ( 'priority', 'depends', 'tags', - 'annotation', + 'annotations', ) class TasklibTest(unittest.TestCase): @@ -494,7 +495,6 @@ class TaskTest(TasklibTest): self.assertTrue(deserializer('') in (None, [], set())) - class TaskFromHookTest(TasklibTest): input_add_data = six.StringIO( @@ -587,6 +587,23 @@ class AnnotationTest(TasklibTest): task.save() self.assertEqual(task['project'], 'test') + def test_serialize_annotations(self): + # Test that serializing annotations is possible + t = Task(self.tw, description="test") + t.save() + + t.add_annotation("annotation1") + t.add_annotation("annotation2") + + data = json.loads(t._serialize('annotations', t._data['annotations'])) + + self.assertEqual(len(data), 2) + self.assertEqual(type(data[0]), dict) + self.assertEqual(type(data[1]), dict) + + self.assertEqual(data[0]['description'], "annotation1") + self.assertEqual(data[1]['description'], "annotation2") + class UnicodeTest(TasklibTest): -- 2.39.5