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'])
