]> git.madduck.net Git - etc/taskwarrior.git/blob - tasklib/tests.py

madduck's git repository

Every one of the projects in this repository is available at the canonical URL git://git.madduck.net/madduck/pub/<projectpath> — see each project's metadata for the exact URL.

All patches and comments are welcome. Please squash your changes to logical commits before using git-format-patch and git-send-email to patches@git.madduck.net. If you'd read over the Git project's submission guidelines and adhered to them, I'd be especially grateful.

SSH access, as well as push access can be individually arranged.

If you use my repositories frequently, consider adding the following snippet to ~/.gitconfig and using the third clone URL listed for each project:

[url "git://git.madduck.net/madduck/"]
  insteadOf = madduck:

Merge pull request #9 from tbabej/fix_tw1452
[etc/taskwarrior.git] / tasklib / tests.py
1 # coding=utf-8
2
3 import shutil
4 import tempfile
5 import unittest
6
7 from .task import TaskWarrior, Task
8
9
10 class TasklibTest(unittest.TestCase):
11
12     def setUp(self):
13         self.tmp = tempfile.mkdtemp(dir='.')
14         self.tw = TaskWarrior(data_location=self.tmp)
15
16     def tearDown(self):
17         shutil.rmtree(self.tmp)
18
19
20 class TaskFilterTest(TasklibTest):
21
22     def test_all_empty(self):
23         self.assertEqual(len(self.tw.tasks.all()), 0)
24
25     def test_all_non_empty(self):
26         Task(self.tw, description="test task").save()
27         self.assertEqual(len(self.tw.tasks.all()), 1)
28         self.assertEqual(self.tw.tasks.all()[0]['description'], 'test task')
29         self.assertEqual(self.tw.tasks.all()[0]['status'], 'pending')
30
31     def test_pending_non_empty(self):
32         Task(self.tw, description="test task").save()
33         self.assertEqual(len(self.tw.tasks.pending()), 1)
34         self.assertEqual(self.tw.tasks.pending()[0]['description'],
35                          'test task')
36         self.assertEqual(self.tw.tasks.pending()[0]['status'], 'pending')
37
38     def test_completed_empty(self):
39         Task(self.tw, description="test task").save()
40         self.assertEqual(len(self.tw.tasks.completed()), 0)
41
42     def test_completed_non_empty(self):
43         Task(self.tw, description="test task").save()
44         self.assertEqual(len(self.tw.tasks.completed()), 0)
45         self.tw.tasks.all()[0].done()
46         self.assertEqual(len(self.tw.tasks.completed()), 1)
47
48     def test_filtering_by_attribute(self):
49         Task(self.tw, description="no priority task").save()
50         Task(self.tw, priority="H", description="high priority task").save()
51         self.assertEqual(len(self.tw.tasks.all()), 2)
52
53         # Assert that the correct number of tasks is returned
54         self.assertEqual(len(self.tw.tasks.filter(priority="H")), 1)
55
56         # Assert that the correct tasks are returned
57         high_priority_task = self.tw.tasks.get(priority="H")
58         self.assertEqual(high_priority_task['description'], "high priority task")
59
60     def test_filtering_by_empty_attribute(self):
61         Task(self.tw, description="no priority task").save()
62         Task(self.tw, priority="H", description="high priority task").save()
63         self.assertEqual(len(self.tw.tasks.all()), 2)
64
65         # Assert that the correct number of tasks is returned
66         self.assertEqual(len(self.tw.tasks.filter(priority=None)), 1)
67
68         # Assert that the correct tasks are returned
69         no_priority_task = self.tw.tasks.get(priority=None)
70         self.assertEqual(no_priority_task['description'], "no priority task")
71
72
73 class TaskTest(TasklibTest):
74
75     def test_create_unsaved_task(self):
76         # Make sure a new task is not saved unless explicitly called for
77         t = Task(self.tw, description="test task")
78         self.assertEqual(len(self.tw.tasks.all()), 0)
79
80     # TODO: once python 2.6 compatiblity is over, use context managers here
81     #       and in all subsequent tests for assertRaises
82
83     def test_delete_unsaved_task(self):
84         t = Task(self.tw, description="test task")
85         self.assertRaises(Task.NotSaved, t.delete)
86
87     def test_complete_unsaved_task(self):
88         t = Task(self.tw, description="test task")
89         self.assertRaises(Task.NotSaved, t.done)
90
91     def test_refresh_unsaved_task(self):
92         t = Task(self.tw, description="test task")
93         self.assertRaises(Task.NotSaved, t.refresh)
94
95     def test_delete_deleted_task(self):
96         t = Task(self.tw, description="test task")
97         t.save()
98         t.delete()
99
100         self.assertRaises(Task.DeletedTask, t.delete)
101
102     def test_complete_completed_task(self):
103         t = Task(self.tw, description="test task")
104         t.save()
105         t.done()
106
107         self.assertRaises(Task.CompletedTask, t.done)
108
109     def test_complete_deleted_task(self):
110         t = Task(self.tw, description="test task")
111         t.save()
112         t.delete()
113
114         self.assertRaises(Task.DeletedTask, t.done)
115
116
117 class AnnotationTest(TasklibTest):
118
119     def setUp(self):
120         super(AnnotationTest, self).setUp()
121         Task(self.tw, description="test task").save()
122
123     def test_adding_annotation(self):
124         task = self.tw.tasks.get()
125         task.add_annotation('test annotation')
126         self.assertEqual(len(task['annotations']), 1)
127         ann = task['annotations'][0]
128         self.assertEqual(ann['description'], 'test annotation')
129
130     def test_removing_annotation(self):
131         task = self.tw.tasks.get()
132         task.add_annotation('test annotation')
133         ann = task['annotations'][0]
134         ann.remove()
135         self.assertEqual(len(task['annotations']), 0)
136
137     def test_removing_annotation_by_description(self):
138         task = self.tw.tasks.get()
139         task.add_annotation('test annotation')
140         task.remove_annotation('test annotation')
141         self.assertEqual(len(task['annotations']), 0)
142
143     def test_removing_annotation_by_obj(self):
144         task = self.tw.tasks.get()
145         task.add_annotation('test annotation')
146         ann = task['annotations'][0]
147         task.remove_annotation(ann)
148         self.assertEqual(len(task['annotations']), 0)
149
150
151 class UnicodeTest(TasklibTest):
152
153     def test_unicode_task(self):
154         Task(self.tw, description="†åßk").save()
155         self.tw.tasks.get()
156
157     def test_non_unicode_task(self):
158         Task(self.tw, description="test task").save()
159         self.tw.tasks.get()