Mock network stuff to call reencrypt.

pull/133/head
jMyles 2017-12-01 16:02:05 -08:00
parent 3bda4b4a3c
commit df9f24e612
2 changed files with 17 additions and 1 deletions

View File

@ -295,7 +295,8 @@ class Bob(Character):
return generated_work_orders
def get_reencrypted_c_frag(self, networky_stuff, work_order):
networky_stuff.reencrypt(work_order)
c_frag = networky_stuff.reencrypt(work_order)
return c_frag
class Ursula(Character):

View File

@ -1,5 +1,8 @@
import asyncio
import msgpack
import pytest
from apistar.test import TestClient
from sqlalchemy.engine import create_engine
@ -70,3 +73,15 @@ class MockNetworkyStuff(NetworkyStuff):
mock_client = TestClient(ursula.rest_app)
response = mock_client.post('http://localhost/kFrag/{}'.format(hrac.hex()), payload)
return True, ursula.interface_dht_key()
def reencrypt(self, work_order):
for _ursula in self.ursulas:
if _ursula.interface_dht_key() == work_order.ursula_id:
ursula = _ursula
break
else:
pytest.fail("No Ursula with ID {}".format(work_order.ursula_id))
mock_client = TestClient(ursula.rest_app)
payload = work_order.payload()
response = mock_client.post('http://localhost/kFrag/{}/reencrypt'.format(work_order.kfrag_hrac.hex()), payload)
self