def setUp(self):
self.test_state = state.DFTimewolfState(config.Config)
self.grr_flow_collector = grr_hosts.GRRFlowCollector(self.test_state)
self.grr_flow_collector.setup(host='tomchop',
flow_id='F:12345',
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_flow_collector = grr_hosts.GRRFlowCollector(self.test_state)
self.grr_flow_collector.SetUp(host='tomchop',
flow_id='F:12345',
reason='random reason',
grr_server_url='http://fake/endpoint',
grr_username='******',
grr_password='******',
approvers='approver1,approver2')
def testInitialization(self):
"""Tests that the collector can be initialized."""
test_state = state.DFTimewolfState()
grr_flow_collector = grr_hosts.GRRFlowCollector(test_state)
self.assertIsNotNone(grr_flow_collector)