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
|
'''
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/>.
'''
import os
import test, mock
class SaveTest(test.TestCase):
def setUp(self):
self.filename = 'SaveTest.tsk'
self.filename2 = 'SaveTest2.tsk'
self.mockApp = mock.App()
self.mockApp.addTasks()
def tearDown(self):
self.mockApp.iocontroller.save()
self.mockApp.quitApplication()
for filename in [self.filename, self.filename2]:
if os.path.isfile(filename):
os.remove(filename)
mock.App.deleteInstance()
super(SaveTest, self).tearDown()
def assertTasksLoaded(self, nrTasks):
self.assertEqual(nrTasks, len(self.mockApp.taskFile.tasks()))
def testSave(self):
self.mockApp.iocontroller.saveas(self.filename)
self.mockApp.iocontroller.open(self.filename)
self.assertTasksLoaded(2)
def testSaveSelection_Child(self):
self.mockApp.iocontroller.saveas(self.filename)
self.mockApp.iocontroller.saveselection([self.mockApp.child], self.filename2)
self.mockApp.iocontroller.close()
self.mockApp.iocontroller.open(self.filename2)
self.assertTasksLoaded(1)
def testSaveSelection_Parent(self):
self.mockApp.iocontroller.saveas(self.filename)
self.mockApp.iocontroller.saveselection([self.mockApp.parent], self.filename2)
self.mockApp.iocontroller.close()
self.mockApp.iocontroller.open(self.filename2)
self.assertTasksLoaded(2)
def testSaveAndMerge(self):
mockApp2 = mock.App()
mockApp2.addTasks()
mockApp2.iocontroller.saveas(self.filename2)
self.mockApp.iocontroller.merge(self.filename2)
self.assertTasksLoaded(4)
self.mockApp.iocontroller.saveas(self.filename)
mockApp2.quitApplication()
|