class ScanManagerTest(TestCase): """Tests against the Manager class and functions.""" scan_manager = None def setUp(self): """Create test case setup.""" self.scan_manager = Manager() self.scan_manager.start() def tearDown(self): """Cleanup test case setup.""" if self.scan_manager.is_alive(): self.scan_manager.running = False self.scan_manager.join() def test_put(self): """Test the put feature of the manager.""" task = MockTask() self.scan_manager.put(task) self.assertEqual(len(self.scan_manager.scan_queue), 1) def test_work(self): """Test the work function.""" task = MockTask() self.scan_manager.put(task) self.scan_manager.work() self.assertIsNone(self.scan_manager.current_task) def test_kill_missing(self): """Test kill on missing id.""" task = MockTask() killed = self.scan_manager.kill(task) self.assertFalse(killed)