def setup(self):
"""Initialise the class."""
self.host = get_host()
self.port = get_unused_tcp_port()
self.identity = Identity("", address="some string")
self.path = "/webhooks/topic/{topic}/"
self.loop = asyncio.get_event_loop()
configuration = ConnectionConfig(
webhook_address=self.host,
webhook_port=self.port,
webhook_url_path=self.path,
connection_id=WebhookConnection.connection_id,
)
self.webhook_connection = WebhookConnection(
configuration=configuration, data_dir=MagicMock(), identity=self.identity,
)
self.dialogues = HttpDialogues(self.identity.address)
class TestWebhookConnection:
"""Tests the webhook connection's 'connect' functionality."""
def setup(self):
"""Initialise the class."""
self.host = get_host()
self.port = get_unused_tcp_port()
self.identity = Identity("", address="some string")
self.path = "/webhooks/topic/{topic}/"
self.loop = asyncio.get_event_loop()
configuration = ConnectionConfig(
webhook_address=self.host,
webhook_port=self.port,
webhook_url_path=self.path,
connection_id=WebhookConnection.connection_id,
)
self.webhook_connection = WebhookConnection(
configuration=configuration, identity=self.identity,
)
self.webhook_connection.loop = self.loop
self.dialogues = HttpDialogues(self.identity.address)
async def test_initialization(self):
"""Test the initialisation of the class."""
assert self.webhook_connection.address == self.identity.address
@pytest.mark.asyncio
async def test_connection(self):
"""Test the connect functionality of the webhook connection."""
await self.webhook_connection.connect()
assert self.webhook_connection.is_connected is True
@pytest.mark.asyncio
async def test_disconnect(self):
"""Test the disconnect functionality of the webhook connection."""
await self.webhook_connection.connect()
assert self.webhook_connection.is_connected is True
await self.webhook_connection.disconnect()
assert self.webhook_connection.is_connected is False
def teardown(self):
"""Close connection after testing."""
try:
self.loop.run_until_complete(self.webhook_connection.disconnect())
except Exception:
print_exc()
raise
@pytest.mark.asyncio
async def test_receive_post_ok(self):
"""Test the connect functionality of the webhook connection."""
await self.webhook_connection.connect()
assert self.webhook_connection.is_connected is True
payload = {"hello": "world"}
call_task = self.loop.create_task(self.call_webhook("test_topic", json=payload))
envelope = await asyncio.wait_for(self.webhook_connection.receive(), timeout=10)
assert envelope
orig_message = cast(HttpMessage, envelope.message)
message = copy.copy(orig_message)
message.counterparty = orig_message.sender
message.is_incoming = True
dialogue = self.dialogues.update(message)
assert dialogue is not None
assert message.method.upper() == "POST"
assert message.bodyy.decode("utf-8") == json.dumps(payload)
await call_task
@pytest.mark.asyncio
async def test_send(self):
"""Test the connect functionality of the webhook connection."""
await self.webhook_connection.connect()
assert self.webhook_connection.is_connected is True
http_message = HttpMessage(
dialogue_reference=("", ""),
target=0,
message_id=1,
performative=HttpMessage.Performative.REQUEST,
method="get",
url="/",
headers="",
bodyy="",
version="",
)
envelope = Envelope(
to="addr",
sender="my_id",
protocol_id=PublicId.from_str("fetchai/http:0.4.0"),
message=http_message,
)
with patch.object(self.webhook_connection.logger, "warning") as mock_logger:
await self.webhook_connection.send(envelope)
await asyncio.sleep(0.01)
mock_logger.assert_any_call(
RegexComparator(
"Dropping envelope=.* as sending via the webhook is not possible!"
)
)
async def call_webhook(self, topic: str, **kwargs) -> ClientResponse:
"""
Make a http request to a webhook.
:param topic: topic to use
:params **kwargs: data or json for payload
:return: http response
"""
path = self.path.format(topic=topic)
method = kwargs.get("method", "post")
url = f"http://{self.host}:{self.port}{path}"
try:
async with aiohttp.ClientSession() as session:
async with session.request(method, url, **kwargs) as resp:
await resp.read()
return resp
except Exception:
print_exc()
raise