def testProcessSpecificArtifacts(self,
mock_SearchClients,
mock_ArtifactCollectorFlowArgs,
mock_DownloadFiles,
mock_CreateFlow):
"""Tests that artifacts defined during setup are searched for."""
mock_DownloadFiles.return_value = '/tmp/tmpRandom/tomchop'
mock_SearchClients.return_value = mock_grr_hosts.MOCK_CLIENT_LIST
mock_CreateFlow.return_value = mock_grr_hosts.MOCK_FLOW
self.grr_artifact_collector = grr_hosts.GRRArtifactCollector(
self.test_state)
self.grr_artifact_collector.SetUp(
hosts='tomchop,tomchop2',
artifacts='RandomArtifact',
extra_artifacts='AnotherArtifact',
use_tsk=True,
reason='Random reason',
grr_server_url='http://fake/endpoint',
grr_username='******',
grr_password='******',
approvers='approver1,approver2',
verify=False
)
self.grr_artifact_collector.Process()
kwargs = mock_ArtifactCollectorFlowArgs.call_args[1]
# raise ValueError(str(kwargs[1]))
self.assertFalse(kwargs['apply_parsers']) # default argument
self.assertTrue(kwargs['ignore_interpolation_errors']) # default argument
self.assertTrue(kwargs['use_tsk'])
sorted_artifacts = sorted(['AnotherArtifact', 'RandomArtifact'])
self.assertEqual(sorted(kwargs['artifact_list']), sorted_artifacts)
def setUp(self):
self.test_state = state.DFTimewolfState(config.Config)
self.grr_artifact_collector = grr_hosts.GRRArtifactCollector(
self.test_state)
self.grr_artifact_collector.setup(
hosts='tomchop,tomchop2',
artifacts=None,
extra_artifacts=None,
use_tsk=True,
reason='Random reason',
grr_server_url='http://fake/endpoint',
grr_username='******',
grr_password='******',
approvers='approver1,approver2',
verify=False)
def setUp(self, mock_InitHttp):
self.mock_grr_api = mock.Mock()
mock_InitHttp.return_value = self.mock_grr_api
self.test_state = state.DFTimewolfState(config.Config)
self.grr_artifact_collector = grr_hosts.GRRArtifactCollector(
self.test_state)
self.grr_artifact_collector.SetUp(
hosts='tomchop',
artifacts=None,
extra_artifacts=None,
use_tsk=True,
reason='Random reason',
grr_server_url='http://fake/endpoint',
grr_username='******',
grr_password='******',
approvers='approver1,approver2',
verify=False)
def testInitialization(self):
"""Tests that the collector can be initialized."""
test_state = state.DFTimewolfState()
grr_artifact_collector = grr_hosts.GRRArtifactCollector(test_state)
self.assertIsNotNone(grr_artifact_collector)