def __init__(self, start_pages, config=Config(), work_executor_factory=WorkExecutorFactory()): self.start_pages = start_pages self.config = config self.work_executor_factory = work_executor_factory
class TestWorkExecutorFactory(unittest.TestCase): def setUp(self): self.work_executor_factory = WorkExecutorFactory() @mock.patch('xcrawler.threads.work_executor.WorkExecutor') def test_create_work_executor(self, mock_work_executor_class): mock_config = mock.create_autospec(Config).return_value mock_work_executor = mock.create_autospec(WorkExecutor).return_value mock_work_executor_class.return_value = mock_work_executor result = self.work_executor_factory.create_work_executor(mock_config) self.assertEquals(result, mock_work_executor)
def setUp(self): self.work_executor_factory = WorkExecutorFactory()