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)