def testSetup(self, mock_add_error):
"""Tests that no paths specified in setup will generate an error."""
test_state = state.DFTimewolfState()
filesystem_collector = filesystem.FilesystemCollector(test_state)
filesystem_collector.setup(paths=None)
mock_add_error.assert_called_with(
'No `paths` argument provided in recipe, bailing', critical=True)
self.assertIsNone(filesystem_collector._paths) # pylint: disable=protected-access
def testOutput(self, mock_exists):
"""Tests that the module ouput is consistent with the input."""
test_state = state.DFTimewolfState(config.Config)
filesystem_collector = filesystem.FilesystemCollector(test_state)
fake_paths = '/fake/path/1,/fake/path/2'
filesystem_collector.setup(paths=fake_paths)
mock_exists.return_value = True
filesystem_collector.process()
expected_output = [('1', '/fake/path/1'), ('2', '/fake/path/2')]
self.assertEqual(test_state.output, expected_output)
def testSetup(self):
"""Tests that no paths specified in setup will generate an error."""
test_state = state.DFTimewolfState(config.Config)
filesystem_collector = filesystem.FilesystemCollector(test_state)
with self.assertRaises(errors.DFTimewolfError) as error:
filesystem_collector.SetUp(paths=None)
self.assertEqual(
'No `paths` argument provided in recipe, bailing',
error.exception.message)
self.assertIsNone(filesystem_collector._paths) # pylint: disable=protected-access
def testOutput(self, mock_exists): """Tests that the module output is consistent with the input.""" test_state = state.DFTimewolfState(config.Config) filesystem_collector = filesystem.FilesystemCollector(test_state) fake_paths = '/fake/path/1,/fake/path/2' filesystem_collector.SetUp(paths=fake_paths) mock_exists.return_value = True filesystem_collector.Process() files = test_state.GetContainers(containers.File) self.assertEqual(files[0].path, '/fake/path/1') self.assertEqual(files[0].name, '1') self.assertEqual(files[1].path, '/fake/path/2') self.assertEqual(files[1].name, '2')
def testInitialization(self):
"""Tests that the collector can be initialized."""
test_state = state.DFTimewolfState()
filesystem_collector = filesystem.FilesystemCollector(test_state)
self.assertIsNotNone(filesystem_collector)
