Fixed protocols route tests

Signed-off-by: Emiliano Suñé <emiliano.sune@gmail.com>
esune · Apr 2, 2024 · 0ea3923 · 0ea3923
1 parent ae5d0d8
commit 0ea3923
Show file tree

Hide file tree

Showing 17 changed files with 172 additions and 64 deletions.
diff --git a/aries_cloudagent/protocols/actionmenu/v1_0/tests/test_routes.py b/aries_cloudagent/protocols/actionmenu/v1_0/tests/test_routes.py
@@ -1,16 +1,22 @@
 from unittest import IsolatedAsyncioTestCase
+
 from aries_cloudagent.tests import mock
 
 from .....admin.request_context import AdminRequestContext
+from .....core.in_memory import InMemoryProfile
 from .....storage.error import StorageNotFoundError
-
 from .. import routes as test_module
 
 
 class TestActionMenuRoutes(IsolatedAsyncioTestCase):
     def setUp(self):
         self.session_inject = {}
-        self.context = AdminRequestContext.test_context(self.session_inject)
+        profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
+        self.context = AdminRequestContext.test_context(self.session_inject, profile)
         self.request_dict = {
             "context": self.context,
             "outbound_message_router": mock.CoroutineMock(),
@@ -20,6 +26,7 @@ def setUp(self):
             match_info={},
             query={},
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
 
     async def test_actionmenu_close(self):

diff --git a/aries_cloudagent/protocols/basicmessage/v1_0/tests/test_routes.py b/aries_cloudagent/protocols/basicmessage/v1_0/tests/test_routes.py
@@ -1,16 +1,22 @@
 from unittest import IsolatedAsyncioTestCase
+
 from aries_cloudagent.tests import mock
 
 from .....admin.request_context import AdminRequestContext
+from .....core.in_memory import InMemoryProfile
 from .....storage.error import StorageNotFoundError
-
 from .. import routes as test_module
 
 
 class TestBasicMessageRoutes(IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
         self.session_inject = {}
-        self.context = AdminRequestContext.test_context(self.session_inject)
+        profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
+        self.context = AdminRequestContext.test_context(self.session_inject, profile)
         self.request_dict = {
             "context": self.context,
             "outbound_message_router": mock.CoroutineMock(),
@@ -20,6 +26,7 @@ async def asyncSetUp(self):
             match_info={},
             query={},
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
         self.test_conn_id = "connection-id"
 

diff --git a/aries_cloudagent/protocols/connections/v1_0/tests/test_routes.py b/aries_cloudagent/protocols/connections/v1_0/tests/test_routes.py
@@ -1,22 +1,27 @@
 import json
-
-from unittest.mock import ANY
 from unittest import IsolatedAsyncioTestCase
+from unittest.mock import ANY
+
 from aries_cloudagent.tests import mock
 
 from .....admin.request_context import AdminRequestContext
 from .....cache.base import BaseCache
 from .....cache.in_memory import InMemoryCache
 from .....connections.models.conn_record import ConnRecord
+from .....core.in_memory import InMemoryProfile
 from .....storage.error import StorageNotFoundError
-
 from .. import routes as test_module
 
 
 class TestConnectionRoutes(IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
         self.session_inject = {}
-        self.context = AdminRequestContext.test_context(self.session_inject)
+        profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
+        self.context = AdminRequestContext.test_context(self.session_inject, profile)
         self.request_dict = {
             "context": self.context,
             "outbound_message_router": mock.CoroutineMock(),
@@ -26,6 +31,7 @@ async def asyncSetUp(self):
             match_info={},
             query={},
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
 
     async def test_connections_list(self):

diff --git a/aries_cloudagent/protocols/coordinate_mediation/v1_0/tests/test_routes.py b/aries_cloudagent/protocols/coordinate_mediation/v1_0/tests/test_routes.py
@@ -1,18 +1,23 @@
-from aries_cloudagent.tests import mock
 from unittest import IsolatedAsyncioTestCase
 
-from .. import routes as test_module
+from aries_cloudagent.tests import mock
+
 from .....admin.request_context import AdminRequestContext
 from .....core.in_memory import InMemoryProfile
 from .....storage.error import StorageError, StorageNotFoundError
+from .....wallet.did_method import DIDMethods
+from .. import routes as test_module
 from ..models.mediation_record import MediationRecord
 from ..route_manager import RouteManager
-from .....wallet.did_method import DIDMethods
 
 
 class TestCoordinateMediationRoutes(IsolatedAsyncioTestCase):
     def setUp(self):
-        self.profile = InMemoryProfile.test_profile()
+        self.profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
         self.profile.context.injector.bind_instance(DIDMethods, DIDMethods())
         self.context = AdminRequestContext.test_context(profile=self.profile)
         self.outbound_message_router = mock.CoroutineMock()
@@ -28,6 +33,7 @@ def setUp(self):
             query={},
             json=mock.CoroutineMock(return_value={}),
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
         serialized = {
             "mediation_id": "fake_id",

diff --git a/aries_cloudagent/protocols/did_rotate/v1_0/tests/test_routes.py b/aries_cloudagent/protocols/did_rotate/v1_0/tests/test_routes.py
@@ -2,12 +2,13 @@
 from unittest import IsolatedAsyncioTestCase
 
 from .....admin.request_context import AdminRequestContext
+from .....core.in_memory import InMemoryProfile
 from .....protocols.didcomm_prefix import DIDCommPrefix
 from .....storage.error import StorageNotFoundError
 from .....tests import mock
-from ..messages import Hangup, Rotate
 from .. import message_types as test_message_types
 from .. import routes as test_module
+from ..messages import Hangup, Rotate
 from ..tests import MockConnRecord, test_conn_id
 
 test_valid_rotate_request = {
@@ -28,8 +29,12 @@ def generate_mock_rotate_message():
 class TestDIDRotateRoutes(IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
         self.session_inject = {}
-
-        self.context = AdminRequestContext.test_context(self.session_inject)
+        profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
+        self.context = AdminRequestContext.test_context(self.session_inject, profile)
         self.request_dict = {
             "context": self.context,
             "outbound_message_router": mock.CoroutineMock(),
@@ -39,6 +44,7 @@ async def asyncSetUp(self):
             match_info={},
             query={},
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
 
     @mock.patch.object(

diff --git a/aries_cloudagent/protocols/didexchange/v1_0/tests/test_routes.py b/aries_cloudagent/protocols/didexchange/v1_0/tests/test_routes.py
@@ -1,16 +1,23 @@
 from unittest import IsolatedAsyncioTestCase
+
 from aries_cloudagent.tests import mock
 
-from .. import routes as test_module
 from .....admin.request_context import AdminRequestContext
+from .....core.in_memory import InMemoryProfile
 from .....storage.error import StorageNotFoundError
 from ....coordinate_mediation.v1_0.route_manager import RouteManager
+from .. import routes as test_module
 
 
 class TestDIDExchangeConnRoutes(IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
-        self.session_inject = {}
-        self.context = AdminRequestContext.test_context(self.session_inject)
+        self.session_inject = {}        
+        profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
+        self.context = AdminRequestContext.test_context(self.session_inject, profile)
         self.profile = self.context.profile
         self.request_dict = {
             "context": self.context,
@@ -21,6 +28,7 @@ async def asyncSetUp(self):
             match_info={},
             query={},
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
         self.profile.context.injector.bind_instance(RouteManager, mock.MagicMock())
 

diff --git a/aries_cloudagent/protocols/discovery/v1_0/tests/test_routes.py b/aries_cloudagent/protocols/discovery/v1_0/tests/test_routes.py
@@ -1,10 +1,10 @@
 from unittest import IsolatedAsyncioTestCase
-from aries_cloudagent.tests import mock
 
+from aries_cloudagent.tests import mock
 
 from .....admin.request_context import AdminRequestContext
+from .....core.in_memory import InMemoryProfile
 from .....storage.error import StorageError
-
 from .. import routes as test_module
 from ..manager import V10DiscoveryMgr
 from ..messages.query import Query
@@ -14,7 +14,12 @@
 class TestDiscoveryRoutes(IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
         self.session_inject = {}
-        self.context = AdminRequestContext.test_context(self.session_inject)
+        profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
+        self.context = AdminRequestContext.test_context(self.session_inject, profile)
         self.profile = self.context.profile
         self.request_dict = {
             "context": self.context,
@@ -25,6 +30,7 @@ async def asyncSetUp(self):
             match_info={},
             query={},
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
 
     async def test_query_features(self):

diff --git a/aries_cloudagent/protocols/discovery/v2_0/tests/test_routes.py b/aries_cloudagent/protocols/discovery/v2_0/tests/test_routes.py
@@ -1,10 +1,10 @@
 from unittest import IsolatedAsyncioTestCase
-from aries_cloudagent.tests import mock
 
+from aries_cloudagent.tests import mock
 
 from .....admin.request_context import AdminRequestContext
+from .....core.in_memory import InMemoryProfile
 from .....storage.error import StorageError
-
 from .. import routes as test_module
 from ..manager import V20DiscoveryMgr
 from ..messages.queries import Queries, QueryItem
@@ -14,7 +14,12 @@
 class TestDiscoveryRoutes(IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
         self.session_inject = {}
-        self.context = AdminRequestContext.test_context(self.session_inject)
+        profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
+        self.context = AdminRequestContext.test_context(self.session_inject, profile)
         self.profile = self.context.profile
         self.request_dict = {
             "context": self.context,
@@ -25,6 +30,7 @@ async def asyncSetUp(self):
             match_info={},
             query={},
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
 
     async def test_query_features(self):

diff --git a/aries_cloudagent/protocols/endorse_transaction/v1_0/tests/test_routes.py b/aries_cloudagent/protocols/endorse_transaction/v1_0/tests/test_routes.py
@@ -1,7 +1,7 @@
 import asyncio
 import json
-
 from unittest import IsolatedAsyncioTestCase
+
 from aries_cloudagent.tests import mock
 
 from .....connections.models.conn_record import ConnRecord
@@ -23,7 +23,11 @@
 
 class TestEndorseTransactionRoutes(IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
-        self.profile = InMemoryProfile.test_profile()
+        self.profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
         self.context = self.profile.context
         setattr(self.context, "profile", self.profile)
         self.session = await self.profile.session()
@@ -67,6 +71,7 @@ async def asyncSetUp(self):
             match_info={},
             query={},
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
 
         self.test_did = "sample-did"

diff --git a/aries_cloudagent/protocols/introduction/v0_1/tests/test_routes.py b/aries_cloudagent/protocols/introduction/v0_1/tests/test_routes.py
@@ -1,15 +1,21 @@
-from aries_cloudagent.tests import mock
 from unittest import IsolatedAsyncioTestCase
 
-from .....admin.request_context import AdminRequestContext
+from aries_cloudagent.tests import mock
 
+from .....admin.request_context import AdminRequestContext
+from .....core.in_memory import InMemoryProfile
 from .. import routes as test_module
 
 
 class TestIntroductionRoutes(IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
         self.session_inject = {}
-        self.context = AdminRequestContext.test_context(self.session_inject)
+        profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
+        self.context = AdminRequestContext.test_context(self.session_inject, profile)
         self.request_dict = {
             "context": self.context,
             "outbound_message_router": mock.CoroutineMock(),
@@ -19,6 +25,7 @@ async def asyncSetUp(self):
             match_info={},
             query={},
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
 
     async def test_introduction_start_no_service(self):

diff --git a/aries_cloudagent/protocols/issue_credential/v1_0/tests/test_routes.py b/aries_cloudagent/protocols/issue_credential/v1_0/tests/test_routes.py
@@ -1,18 +1,23 @@
-from aries_cloudagent.tests import mock
 from unittest import IsolatedAsyncioTestCase
 
+from aries_cloudagent.tests import mock
+
 from .....admin.request_context import AdminRequestContext
+from .....core.in_memory import InMemoryProfile
 from .....wallet.base import BaseWallet
-
 from .. import routes as test_module
-
 from . import CRED_DEF_ID
 
 
 class TestCredentialRoutes(IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
         self.session_inject = {}
-        self.context = AdminRequestContext.test_context(self.session_inject)
+        profile = InMemoryProfile.test_profile(
+            settings={
+                "admin.admin_api_key": "secret-key",
+            }
+        )
+        self.context = AdminRequestContext.test_context(self.session_inject, profile)
         self.request_dict = {
             "context": self.context,
             "outbound_message_router": mock.CoroutineMock(),
@@ -22,6 +27,7 @@ async def asyncSetUp(self):
             match_info={},
             query={},
             __getitem__=lambda _, k: self.request_dict[k],
+            headers={"x-api-key": "secret-key"},
         )
 
     async def test_credential_exchange_list(self):