def test_evidence_list(self): thread = self._start_server() client = EngineClient(thread.server._transport) case_uuid = str(uuid.uuid4()) one = client.evidence_add(case_uuid=case_uuid, container_format='raw', container_size=1024, name='one') two = client.evidence_add(case_uuid=case_uuid, container_format='raw', container_size=1024, name='two') self.assertValidId(one) self.assertValidId(two) items = client.evidence_list(case_uuid) self._stop_server(client, thread) found = 0 for item in items: if item['uuid'] == one['uuid']: found += 1 self.assertEquals(item['name'], 'one') elif item['uuid'] == two['uuid']: found += 1 self.assertEquals(item['name'], 'two') self.assertEquals(found, 2)
def __init__(self, host, pipeline_driver=None, topic=None, allow_stop=False): topic = topic or 'process' self.host = host self.topic = 'gate.%s' % topic self._pipeline_driver = pipeline_driver if not self._pipeline_driver: self._pipeline_driver = get_pipeline_driver() self._transport = messaging.get_transport(cfg.CONF) self._target = messaging.Target(topic=self.topic, server=self.host) self._engine_client = EngineClient(self._transport) self._endpoints = [ProcessAPI(self, engine_client=self._engine_client)] if allow_stop: self._endpoints.append(self) self._server = messaging.get_rpc_server(self._transport, self._target, self._endpoints)
def test_case_create(self): thread = self._start_server() client = EngineClient(thread.server._transport) account_uuid = str(uuid.uuid4()) result = client.case_create(account_uuid=account_uuid, name='testname') self.assertValidId(result) result2 = client.case_get(result['uuid']) self._stop_server(client, thread) self.assertEquals(result2['uuid'], result['uuid']) self.assertEquals(result2['name'], 'testname')
def test_evidence_add(self): thread = self._start_server() client = EngineClient(thread.server._transport) case_uuid = str(uuid.uuid4()) result = client.evidence_add(case_uuid=case_uuid, container_format='raw', container_size=1024, name='testname') self.assertValidId(result) result2 = client.evidence_get(result['uuid']) self._stop_server(client, thread) self.assertEquals(result2['uuid'], result['uuid']) self.assertEquals(result2['name'], 'testname')
def test_case_list(self): thread = self._start_server() client = EngineClient(thread.server._transport) account_uuid = str(uuid.uuid4()) one = client.case_create(account_uuid=account_uuid, name='one') two = client.case_create(account_uuid=account_uuid, name='two') self.assertValidId(one) self.assertValidId(two) items = client.case_list(account_uuid) self._stop_server(client, thread) found = 0 for item in items: if item['uuid'] == one['uuid']: found += 1 self.assertEquals(item['name'], 'one') elif item['uuid'] == two['uuid']: found += 1 self.assertEquals(item['name'], 'two') self.assertEquals(found, 2)
def test_custom_topic(self): thread = self._start_server(topic='custom') client = EngineClient(thread.server._transport, topic='custom') self.assertEqual(thread.server.topic, client.topic) self._stop_server(client, thread)
def test_default_topic(self): thread = self._start_server() client = EngineClient(thread.server._transport) self.assertEqual(thread.server.topic, client.topic) self._stop_server(client, thread)