class TestSleepTask(object): def setup(self): self.root = RootTask(should_stop=Event(), should_pause=Event()) self.task = SleepTask(task_name='Test') self.root.children_task.append(self.task) def test_check1(self): # Simply test that everything is ok if time > 0. self.task.time = 1.0 test, traceback = self.task.check() assert_true(test) assert_false(traceback) def test_check2(self): # Test handling a wrong message. self.task.time = -1.0 test, traceback = self.task.check() assert_false(test) assert_equal(len(traceback), 1) assert_in('root/Test', traceback) def test_perform(self): # Test performing when condition is True. self.root.task_database.prepare_for_running() self.task.perform()