def setUp(self): self.test_state = state.DFTimewolfState(config.Config) self.grr_file_collector = grr_hosts.GRRFileCollector(self.test_state) self.grr_file_collector.setup(hosts='tomchop,tomchop2', files='/etc/passwd', use_tsk=True, reason='random reason', grr_server_url='http://fake/endpoint', grr_username='******', grr_password='******', approvers='approver1,approver2')
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_file_collector = grr_hosts.GRRFileCollector(self.test_state) self.grr_file_collector.SetUp(hosts='tomchop', files='/etc/passwd', use_tsk=True, reason='random reason', grr_server_url='http://fake/endpoint', grr_username='******', grr_password='******', approvers='approver1,approver2', action='stat')
def testInitialization(self): """Tests that the collector can be initialized.""" test_state = state.DFTimewolfState() grr_file_collector = grr_hosts.GRRFileCollector(test_state) self.assertIsNotNone(grr_file_collector)