From: Tomas Babej Date: Thu, 24 Dec 2015 10:20:47 +0000 (+0100) Subject: tests: Add tests for the LazyUUIDTaskSet X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/commitdiff_plain/da5d98a02980fee41065de903d1ed6403bbd9f4d?hp=238c0ba7d462d67f8b847bee06233a7fd20893c4 tests: Add tests for the LazyUUIDTaskSet --- diff --git a/tasklib/tests.py b/tasklib/tests.py index fa4e5f3..7611536 100644 --- a/tasklib/tests.py +++ b/tasklib/tests.py @@ -12,7 +12,8 @@ import tempfile import unittest from .backends import TaskWarrior -from .task import Task, ReadOnlyDictView, LazyUUIDTask +from .task import (Task, ReadOnlyDictView, LazyUUIDTask, + LazyUUIDTaskSet, TaskQuerySet) from .serializing import DATE_FORMAT, local_zone # http://taskwarrior.org/docs/design/task.html , Section: The Attributes @@ -1134,7 +1135,47 @@ class LazyUUIDTaskTest(TasklibTest): assert type(lazy2) is LazyUUIDTask def test_lazy_in_queryset(self): - tasks = self.tw.tasks.filter(uuid=self.stored['uuid']) + tasks = self.tw.tasks.tasks.filter(uuid=self.stored['uuid']) assert self.lazy in tasks assert type(self.lazy) is LazyUUIDTask + + +class LazyUUIDTaskSetTest(TasklibTest): + + def setUp(self): + super(LazyUUIDTaskTest, self).setUp() + + self.task1 = Task(self.tw, description="task 1") + self.task2 = Task(self.tw, description="task 2") + self.task3 = Task(self.tw, description="task 3") + + self.task1.save() + self.task2.save() + self.task3.save() + + self.uuids = ( + self.task1['uuid'], + self.task2['uuid'], + self.task3['uuid'] + ) + + self.lazy = LazyUUIDTaskSet(self.tw, self.uuids) + + def test_length(self): + assert len(self.lazy) == 3 + assert type(self.lazy) is LazyUUIDTaskSet + + def test_contains(self): + assert self.task1 in self.lazy + assert self.task2 in self.lazy + assert self.task3 in self.lazy + assert type(self.lazy) is LazyUUIDTaskSet + + def test_eq(self): + assert self.lazy == self.tw.tasks.all() + assert type(self.lazy) is LazyUUIDTaskSet + + def test_conversion(self): + assert set(self.lazy) == set(self.tw.tasks.all()) + assert type(self.lazy) is TaskQuerySet