def setUp(self): self.patcher = patch('dissect.modeling.static.driver.Orchestrator') self.orchestrator_mock = Mock() self.orchestrator_class = self.patcher.start() self.orchestrator_class.return_value = self.orchestrator_mock self.file_set = fake('FileSet') self.observer = Mock() self.static_driver = StaticDriver(self.file_set)
class TestDriver(): def setUp(self): self.patcher = patch('dissect.modeling.static.driver.Orchestrator') self.orchestrator_mock = Mock() self.orchestrator_class = self.patcher.start() self.orchestrator_class.return_value = self.orchestrator_mock self.file_set = fake('FileSet') self.observer = Mock() self.static_driver = StaticDriver(self.file_set) def tearDown(self): self.patcher.stop() def test_it_models_data_from_each_file(self): # Arrange self.file_set.__iter__.return_value = ['a.py', 'path/to/b.py'] # Act self.static_driver.run() # Assert self.orchestrator_mock.process.assert_called_once_with(['a.py', 'path/to/b.py'])