Skip to content

Commit 46e1fa7

Browse files
committed
tests: add test_forward_message
1 parent 39ad76d commit 46e1fa7

File tree

1 file changed

+18
-3
lines changed

1 file changed

+18
-3
lines changed

tests/hikari/impl/test_rest.py

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2130,9 +2130,7 @@ class StubResponse:
21302130
await rest_client._request(route)
21312131

21322132
@hikari_test_helpers.timeout()
2133-
async def test_request_when_status_in_retry_codes_will_retry_until_exhausted(
2134-
self, rest_client, exit_exception
2135-
):
2133+
async def test_request_when_status_in_retry_codes_will_retry_until_exhausted(self, rest_client, exit_exception):
21362134
class StubResponse:
21372135
status = http.HTTPStatus.INTERNAL_SERVER_ERROR
21382136

@@ -2936,6 +2934,23 @@ async def test_create_voice_message(self, rest_client):
29362934
rest_client._request.assert_awaited_once_with(expected_route, form_builder=mock_form)
29372935
rest_client._entity_factory.deserialize_message.assert_called_once_with({"message_id": 123})
29382936

2937+
async def test_forward_message(self, rest_client):
2938+
rest_client._request = mock.AsyncMock(return_value={"message_id": 1239})
2939+
expected_route = routes.POST_CHANNEL_MESSAGES.compile(channel=1234)
2940+
m = await rest_client.forward_message(channel_to=1234, message=123, channel_from=12345)
2941+
assert m is rest_client._entity_factory.deserialize_message.return_value
2942+
rest_client._request.assert_awaited_once_with(
2943+
expected_route,
2944+
json={
2945+
"message_reference": {
2946+
"message_id": "123",
2947+
"channel_id": "12345",
2948+
"type": message_models.MessageReferenceType.FORWARD,
2949+
}
2950+
},
2951+
)
2952+
rest_client._entity_factory.deserialize_message.assert_called_once_with({"message_id": 1239})
2953+
29392954
async def test_create_voice_message_no_flags(self, rest_client):
29402955
rest_client._request = mock.AsyncMock(return_value={"message_id": 123})
29412956
returned = await rest_client.create_voice_message(

0 commit comments

Comments
 (0)