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)
