From ca2e84eb2ae6993c2b4c0fd69c788b5b4f00a448 Mon Sep 17 00:00:00 2001 From: jMyles Date: Wed, 24 Jun 2020 19:28:22 -0700 Subject: [PATCH] Only simulate hardcoded domain nodes for TEMPORARY_DOMAIN, not nonstandard test domains. --- tests/utils/middleware.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/utils/middleware.py b/tests/utils/middleware.py index da6d730da..6e22cb73e 100644 --- a/tests/utils/middleware.py +++ b/tests/utils/middleware.py @@ -23,6 +23,7 @@ from constant_sorrow.constants import CERTIFICATE_NOT_SAVED from flask import Response from nucypher.characters.lawful import Ursula +from nucypher.config.constants import TEMPORARY_DOMAIN from nucypher.network.middleware import NucypherMiddlewareClient, RestMiddleware from tests.utils.ursula import MOCK_KNOWN_URSULAS_CACHE @@ -93,7 +94,11 @@ class MockRestMiddleware(RestMiddleware): @classmethod def get(_cls, item, default): - return tuple(u.rest_url() for u in MOCK_KNOWN_URSULAS_CACHE.values()) + if item is TEMPORARY_DOMAIN: + nodes = tuple(u.rest_url() for u in MOCK_KNOWN_URSULAS_CACHE.values()) + else: + nodes = tuple() + return nodes def get_certificate(self, host, port, timeout=3, retry_attempts: int = 3, retry_rate: int = 2, current_attempt: int = 0):