1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
'''
Task Coach - Your friendly task manager
Copyright (C) 2004-2014 Task Coach developers <developers@taskcoach.org>
Task Coach is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Task Coach is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
'''
from taskcoachlib import command
from taskcoachlib.domain import attachment, task, note, category
from CommandTestCase import CommandTestCase
class AddAttachmentTestsMixin(object):
def addAttachment(self, selectedItems=None):
self.attachment = attachment.FileAttachment('attachment') # pylint: disable=W0201
addAttachmentCommand = command.AddAttachmentCommand(self.container,
selectedItems or [], attachments=[self.attachment])
addAttachmentCommand.do()
def testAddOneAttachmentToOneItem(self):
self.addAttachment([self.item1])
self.assertDoUndoRedo(lambda: self.assertEqual([self.attachment],
self.item1.attachments()), lambda: self.assertEqual([],
self.item1.attachments()))
def testAddOneAttachmentToTwoItems(self):
self.addAttachment([self.item1, self.item2])
self.assertDoUndoRedo(lambda: self.failUnless([self.attachment] == \
self.item1.attachments() == self.item2.attachments()),
lambda: self.failUnless([] == self.item1.attachments() == \
self.item2.attachments()))
class AddAttachmentTestCase(CommandTestCase):
ItemClass = ContainerClass = lambda subject: 'Subclass responsibility'
def setUp(self):
super(AddAttachmentTestCase, self).setUp()
self.item1 = self.ItemClass(subject='item1')
self.item2 = self.ItemClass(subject='item2')
self.container = self.ContainerClass([self.item1, self.item2])
class AddAttachmentCommandWithTasksTest(AddAttachmentTestCase, AddAttachmentTestsMixin):
ItemClass = task.Task
ContainerClass = task.TaskList
class AddAttachmentCommandWithNotesTest(AddAttachmentTestCase, AddAttachmentTestsMixin):
ItemClass = note.Note
ContainerClass = note.NoteContainer
class AddAttachmentCommandWithCategoriesTest(AddAttachmentTestCase, AddAttachmentTestsMixin):
ItemClass = category.Category
ContainerClass = category.CategoryList
|
