fix import from new unit tests

Signed-off-by: Andrew Whitehead <[email protected]>
openwallet-foundation · Nov 26, 2019 · 5842f7a · 5842f7a
1 parent 625a0b0
commit 5842f7a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/aries_cloudagent/protocols/problem_report/tests/test_handler.py b/aries_cloudagent/protocols/problem_report/tests/test_handler.py
@@ -1,9 +1,9 @@
 import pytest
 
 from ....messaging.base_handler import HandlerException
-from ....messaging.message_delivery import MessageDelivery
 from ....messaging.request_context import RequestContext
 from ....messaging.responder import MockResponder
+from ....transport.inbound.receipt import MessageReceipt
 
 from ..handler import ProblemReportHandler
 from ..message import ProblemReport
@@ -18,7 +18,7 @@ def request_context() -> RequestContext:
 class TestPingHandler:
     @pytest.mark.asyncio
     async def test_problem_report(self, request_context):
-        request_context.message_delivery = MessageDelivery()
+        request_context.message_receipt = MessageReceipt()
         request_context.message = ProblemReport()
         request_context.connection_ready = True
         handler = ProblemReportHandler()

diff --git a/aries_cloudagent/protocols/trustping/handlers/tests/test_ping_handler.py b/aries_cloudagent/protocols/trustping/handlers/tests/test_ping_handler.py
@@ -1,9 +1,9 @@
 import pytest
 
 from .....messaging.base_handler import HandlerException
-from .....messaging.message_delivery import MessageDelivery
 from .....messaging.request_context import RequestContext
 from .....messaging.responder import MockResponder
+from .....transport.inbound.receipt import MessageReceipt
 
 from ...handlers.ping_handler import PingHandler
 from ...messages.ping import Ping
@@ -19,7 +19,7 @@ def request_context() -> RequestContext:
 class TestPingHandler:
     @pytest.mark.asyncio
     async def test_ping(self, request_context):
-        request_context.message_delivery = MessageDelivery()
+        request_context.message_receipt = MessageReceipt()
         request_context.message = Ping(response_requested=False)
         request_context.connection_ready = True
         handler = PingHandler()
@@ -30,7 +30,7 @@ async def test_ping(self, request_context):
 
     @pytest.mark.asyncio
     async def test_ping_response(self, request_context):
-        request_context.message_delivery = MessageDelivery()
+        request_context.message_receipt = MessageReceipt()
         request_context.message = Ping(response_requested=True)
         request_context.connection_ready = True
         handler = PingHandler()

diff --git a/aries_cloudagent/protocols/trustping/handlers/tests/test_ping_response_handler.py b/aries_cloudagent/protocols/trustping/handlers/tests/test_ping_response_handler.py
@@ -1,9 +1,9 @@
 import pytest
 
 from .....messaging.base_handler import HandlerException
-from .....messaging.message_delivery import MessageDelivery
 from .....messaging.request_context import RequestContext
 from .....messaging.responder import MockResponder
+from .....transport.inbound.receipt import MessageReceipt
 
 from ...handlers.ping_response_handler import PingResponseHandler
 from ...messages.ping_response import PingResponse
@@ -18,7 +18,7 @@ def request_context() -> RequestContext:
 class TestPingResponseHandler:
     @pytest.mark.asyncio
     async def test_ping_response(self, request_context):
-        request_context.message_delivery = MessageDelivery()
+        request_context.message_receipt = MessageReceipt()
         request_context.message = PingResponse()
         request_context.connection_ready = True
         handler = PingResponseHandler()