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 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_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 __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_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_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 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_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)
