''' Task Coach - Your friendly task manager Copyright (C) 2004-2014 Task Coach developers 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 . ''' import wx, StringIO, os import test from taskcoachlib import persistence, gui, config, render from taskcoachlib.domain import task, category, effort, date class HTMLWriterTestCase(test.wxTestCase): treeMode = 'Subclass responsibility' filename = 'Subclass responsibility' def setUp(self): super(HTMLWriterTestCase, self).setUp() task.Task.settings = self.settings = config.Settings(load=False) self.fd = StringIO.StringIO() self.writer = persistence.HTMLWriter(self.fd, self.filename) self.taskFile = persistence.TaskFile() self.task = task.Task('Task subject') self.taskFile.tasks().append(self.task) self.viewer = self.createViewer() def tearDown(self): super(HTMLWriterTestCase, self).tearDown() self.taskFile.close() self.taskFile.stop() cssFilename = self.filename + '.css' if os.path.exists(cssFilename): os.remove(cssFilename) def createViewer(self): raise NotImplementedError # pragma: no cover def __writeAndRead(self, selectionOnly): self.writer.write(self.viewer, self.settings, selectionOnly, True) return self.fd.getvalue() def expectInHTML(self, *htmlFragments, **kwargs): selectionOnly = kwargs.pop('selectionOnly', False) html = self.__writeAndRead(selectionOnly) for htmlFragment in htmlFragments: self.failUnless(htmlFragment in html, '%s not in %s'%(htmlFragment, html)) def expectNotInHTML(self, *htmlFragments, **kwargs): selectionOnly = kwargs.pop('selectionOnly', False) html = self.__writeAndRead(selectionOnly) for htmlFragment in htmlFragments: self.failIf(htmlFragment in html, '%s in %s'%(htmlFragment, html)) def selectItem(self, items): self.viewer.widget.select(items) class CommonTestsMixin(object): def testHTML(self): self.expectInHTML('\n', '\n') def testHeader(self): self.expectInHTML(' \n', ' \n') def testStyle(self): self.expectInHTML(' \n') def testBody(self): self.expectInHTML(' \n', ' \n') class TaskWriterTestCase(HTMLWriterTestCase): def createViewer(self): self.settings.set('taskviewer', 'treemode', self.treeMode) return gui.viewer.TaskViewer(self.frame, self.taskFile, self.settings) class TaskTestsMixin(CommonTestsMixin): def testTaskSubject(self): self.expectInHTML('>Task subject<') def testWriteSelectionOnly(self): self.expectNotInHTML('>Task subject<', selectionOnly=True) def testWriteSelectionOnly_SelectedChild(self): child = task.Task('Child') self.task.addChild(child) self.taskFile.tasks().append(child) self.selectItem([child]) self.expectInHTML('>Task subject<') def testColumnStyle(self): self.expectInHTML(' .subject {text-align: left}\n') def testSortIndicator(self): self.expectInHTML('id="sorted"') if not self.filename: self.expectInHTML('') def testTaskStatusStyle(self): self.expectInHTML(' .completed {color: #00FF00}\n') def testTaskStatusStyleWhenForegroundColorChangedInSettings(self): self.settings.set('fgcolor', 'completedtasks', str(wx.RED)) self.expectInHTML(' .completed {color: #FF0000}\n') def testOverdueTask(self): self.task.setDueDateTime(date.Yesterday()) fragment = '' if self.filename else 'Task subject' self.expectInHTML(fragment) def testCompletedTask(self): self.task.setCompletionDateTime() if self.filename: self.expectInHTML('') else: self.expectInHTML('Task subject') def testTaskDueSoon(self): self.task.setDueDateTime(date.Now() + date.ONE_HOUR) fragment = '' if self.filename else 'Task subject' self.expectInHTML(fragment) def testInactiveTask(self): self.task.setPlannedStartDateTime(date.Tomorrow()) fragment = '' if self.filename else 'Task subject' self.expectInHTML(fragment) def testLateTask(self): self.task.setPlannedStartDateTime(date.Yesterday()) fragment = '' if self.filename else 'Task subject' self.expectInHTML(fragment) def testTaskBackgroundColor(self): self.task.setActualStartDateTime(date.Now()) self.task.setBackgroundColor(wx.RED) fragment = '' if self.filename else '' self.expectInHTML(fragment) def testTaskHasCategoryBackgroundColor(self): self.task.setActualStartDateTime(date.Now()) cat = category.Category('cat', bgColor=wx.RED) self.task.addCategory(cat) fragment = '' if self.filename else '' self.expectInHTML(fragment) def testCategoryBackgroundColorAsTuple(self): self.task.setActualStartDateTime(date.Now()) cat = category.Category('cat', bgColor=(255, 0, 0, 0)) self.task.addCategory(cat) if self.filename: self.expectInHTML('') else: self.expectInHTML('') def testCSSLink(self): if self.filename: self.expectInHTML('') else: self.expectNotInHTML('stylesheet') def testOSErrorWhileWritingCSS(self): def open(*args): # pylint: disable=W0613,W0622 raise IOError self.writer._writeCSS(open=open) # pylint: disable=W0212 class TaskListTestsMixin(object): def testTaskDescription(self): self.task.setDescription('Task description') self.viewer.showColumnByName('description') self.expectInHTML('>Task description<') def testTaskDescriptionWithNewLine(self): self.task.setDescription('Line1\nLine2') self.viewer.showColumnByName('description') self.expectInHTML('>Line1
Line2<') def testCreationDateTime(self): self.viewer.showColumnByName('creationDateTime') self.expectInHTML(render.dateTime(self.task.creationDateTime(), humanReadable=False)) def testMissingCreationDateTime(self): self.viewer.showColumnByName('creationDateTime') self.taskFile.tasks().append(task.Task(creationDateTime=date.DateTime.min)) self.taskFile.tasks().remove(self.task) self.expectNotInHTML('1/1/1') def testModificationDateTime(self): self.task.setModificationDateTime(date.DateTime(2012, 1, 1, 10, 0, 0)) self.viewer.showColumnByName('modificationDateTime') self.expectInHTML(render.dateTime(self.task.modificationDateTime(), humanReadable=False)) def testMissingModificationDateTime(self): self.viewer.showColumnByName('modificationDateTime') self.expectInHTML(render.dateTime(self.task.modificationDateTime(), humanReadable=False)) self.expectNotInHTML('1/1/1') class TaskListExportTest(TaskTestsMixin, TaskListTestsMixin, TaskWriterTestCase): treeMode = 'False' filename = 'filename' class TaskListPrintTest(TaskTestsMixin, TaskListTestsMixin, TaskWriterTestCase): treeMode = 'False' filename = '' class TaskTreeExportTest(TaskTestsMixin, TaskWriterTestCase): treeMode = 'True' filename = 'filename' class TaskTreePrintTest(TaskTestsMixin, TaskWriterTestCase): treeMode = 'True' filename = '' class EffortWriterTestCase(CommonTestsMixin, HTMLWriterTestCase): filename = 'filename' def setUp(self): super(EffortWriterTestCase, self).setUp() now = date.DateTime.now() self.task.addEffort(effort.Effort(self.task, start=now, stop=now + date.ONE_SECOND)) def createViewer(self): return gui.viewer.EffortViewer(self.frame, self.taskFile, self.settings) def testTaskSubject(self): self.expectInHTML('>Task subject<') def testEffortDuration(self): self.expectInHTML('>0:00:01<') def testColumnStyle(self): self.expectInHTML(' .task {text-align: left}\n') class CategoryWriterTestsMixin(CommonTestsMixin): def testCategorySubject(self): self.expectInHTML('>Category<') def testCategoryBackgroundColor(self): self.category.setBackgroundColor(wx.RED) if self.filename: self.expectInHTML('') else: self.expectInHTML('') class CategoryWriterTestCase(HTMLWriterTestCase): def setUp(self): super(CategoryWriterTestCase, self).setUp() self.category = category.Category('Category') self.taskFile.categories().append(self.category) def createViewer(self): return gui.viewer.CategoryViewer(self.frame, self.taskFile, self.settings) class CategoryWriterExportTest(CategoryWriterTestsMixin, CategoryWriterTestCase): filename = 'filename' class CategoryWriterPrintTest(CategoryWriterTestsMixin, CategoryWriterTestCase): filename = ''