Try and fix tests

Signed-off-by: jamshale <jamiehalebc@gmail.com>
openwallet-foundation · Feb 19, 2024 · 5d5beee · 5d5beee
1 parent c002bf5
commit 5d5beee
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/aries_cloudagent/multitenant/admin/tests/test_routes.py b/aries_cloudagent/multitenant/admin/tests/test_routes.py
@@ -688,12 +688,13 @@ async def test_wallet_remove_managed(self):
         ):
             self.mock_multitenant_mgr.remove_wallet = mock.CoroutineMock()
 
-            await test_module.wallet_remove(self.request)
+            result = await test_module.wallet_remove(self.request)
 
             self.mock_multitenant_mgr.remove_wallet.assert_called_once_with(
                 "dummy", None
             )
             mock_response.assert_called_once_with({})
+            assert result == mock_response.return_value
 
     async def test_wallet_remove_unmanaged(self):
         self.request.match_info = {"wallet_id": "dummy"}
@@ -706,12 +707,13 @@ async def test_wallet_remove_unmanaged(self):
         ):
             self.mock_multitenant_mgr.remove_wallet = mock.CoroutineMock()
 
-            await test_module.wallet_remove(self.request)
+            result = await test_module.wallet_remove(self.request)
 
             self.mock_multitenant_mgr.remove_wallet.assert_called_once_with(
                 "dummy", "dummy_key"
             )
             mock_response.assert_called_once_with({})
+            assert result == mock_response.return_value
 
     async def test_wallet_remove_managed_wallet_key_provided_throws(self):
         self.request.match_info = {"wallet_id": "dummy"}