Nucypher
/
nucypher
mirror of https://github.com/nucypher/nucypher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Reintroducing _get_mock_client_by_port in order to dehydrate middleware.

pull/730/head
jMyles 2019-02-05 16:53:48 -07:00
parent 6b9a205c8b
commit 77a763c090
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nucypher/utilities/sandbox/middleware.py
View File

@ -35,6 +35,13 @@ class _TestMiddlewareClient(NucypherMiddlewareClient):
port = int(url.split(":")[1])
return self._get_mock_client_by_port(port)
def _get_mock_client_by_port(self, port):
ursula = self._get_ursula_by_port(port)
rest_app = ursula.rest_app
rest_app.testing = True
mock_client = rest_app.test_client()
return mock_client
def _get_ursula_by_port(self, port):
try:
return MOCK_KNOWN_URSULAS_CACHE[port]