|
3 | 3 | import os |
4 | 4 | from asyncio import AbstractEventLoop |
5 | 5 | from contextlib import suppress |
6 | | -from datetime import datetime |
| 6 | +from datetime import datetime, timedelta |
7 | 7 |
|
8 | 8 | import pytest |
9 | 9 | import pytest_asyncio |
|
33 | 33 | ParagraphText, |
34 | 34 | Message, |
35 | 35 | GuildVoice, |
| 36 | + Poll, |
| 37 | + PollMedia, |
36 | 38 | ) |
37 | 39 | from interactions.models.discord.asset import Asset |
38 | 40 | from interactions.models.discord.components import ActionRow, Button, StringSelectMenu |
@@ -432,6 +434,95 @@ async def test_components(bot: Client, channel: GuildText) -> None: |
432 | 434 | await thread.delete() |
433 | 435 |
|
434 | 436 |
|
| 437 | +@pytest.mark.asyncio |
| 438 | +async def test_polls(bot: Client, channel: GuildText) -> None: |
| 439 | + msg = await channel.send("Polls Tests") |
| 440 | + thread = await msg.create_thread("Test Thread") |
| 441 | + |
| 442 | + try: |
| 443 | + poll_1 = Poll.create("Test Poll", duration=1, answers=["Answer 1", "Answer 2"]) |
| 444 | + test_data_1 = { |
| 445 | + "question": {"text": "Test Poll"}, |
| 446 | + "layout_type": 1, |
| 447 | + "duration": 1, |
| 448 | + "allow_multiselect": False, |
| 449 | + "answers": [{"poll_media": {"text": "Answer 1"}}, {"poll_media": {"text": "Answer 2"}}], |
| 450 | + } |
| 451 | + poll_1_dict = poll_1.to_dict() |
| 452 | + for key in poll_1_dict.keys(): |
| 453 | + assert poll_1_dict[key] == test_data_1[key] |
| 454 | + |
| 455 | + msg_1 = await thread.send(poll=poll_1) |
| 456 | + |
| 457 | + assert msg_1.poll is not None |
| 458 | + assert msg_1.poll.question.to_dict() == PollMedia(text="Test Poll").to_dict() |
| 459 | + assert msg_1.poll.expiry <= msg_1.created_at + timedelta(hours=1, minutes=1) |
| 460 | + poll_1_answer_medias = [poll_answer.poll_media.to_dict() for poll_answer in msg_1.poll.answers] |
| 461 | + assert poll_1_answer_medias == [ |
| 462 | + PollMedia.create(text="Answer 1").to_dict(), |
| 463 | + PollMedia.create(text="Answer 2").to_dict(), |
| 464 | + ] |
| 465 | + |
| 466 | + poll_2 = Poll.create("Test Poll 2", duration=1, allow_multiselect=True) |
| 467 | + poll_2.add_answer("Answer 1") |
| 468 | + poll_2.add_answer("Answer 2") |
| 469 | + test_data_2 = { |
| 470 | + "question": {"text": "Test Poll 2"}, |
| 471 | + "layout_type": 1, |
| 472 | + "duration": 1, |
| 473 | + "allow_multiselect": True, |
| 474 | + "answers": [{"poll_media": {"text": "Answer 1"}}, {"poll_media": {"text": "Answer 2"}}], |
| 475 | + } |
| 476 | + poll_2_dict = poll_2.to_dict() |
| 477 | + for key in poll_2_dict.keys(): |
| 478 | + assert poll_2_dict[key] == test_data_2[key] |
| 479 | + msg_2 = await thread.send(poll=poll_2) |
| 480 | + |
| 481 | + assert msg_2.poll is not None |
| 482 | + assert msg_2.poll.question.to_dict() == PollMedia(text="Test Poll 2").to_dict() |
| 483 | + assert msg_2.poll.expiry <= msg_2.created_at + timedelta(hours=1, minutes=1) |
| 484 | + assert msg_2.poll.allow_multiselect |
| 485 | + poll_2_answer_medias = [poll_answer.poll_media.to_dict() for poll_answer in msg_2.poll.answers] |
| 486 | + assert poll_2_answer_medias == [ |
| 487 | + PollMedia.create(text="Answer 1").to_dict(), |
| 488 | + PollMedia.create(text="Answer 2").to_dict(), |
| 489 | + ] |
| 490 | + |
| 491 | + poll_3 = Poll.create( |
| 492 | + "Test Poll 3", |
| 493 | + duration=1, |
| 494 | + answers=[PollMedia.create(text="One", emoji="1️⃣"), PollMedia.create(text="Two", emoji="2️⃣")], |
| 495 | + ) |
| 496 | + test_data_3 = { |
| 497 | + "question": {"text": "Test Poll 3"}, |
| 498 | + "layout_type": 1, |
| 499 | + "duration": 1, |
| 500 | + "allow_multiselect": False, |
| 501 | + "answers": [ |
| 502 | + {"poll_media": {"text": "One", "emoji": {"name": "1️⃣", "animated": False}}}, |
| 503 | + {"poll_media": {"text": "Two", "emoji": {"name": "2️⃣", "animated": False}}}, |
| 504 | + ], |
| 505 | + } |
| 506 | + poll_3_dict = poll_3.to_dict() |
| 507 | + for key in poll_3_dict.keys(): |
| 508 | + assert poll_3_dict[key] == test_data_3[key] |
| 509 | + |
| 510 | + msg_3 = await thread.send(poll=poll_3) |
| 511 | + |
| 512 | + assert msg_3.poll is not None |
| 513 | + assert msg_3.poll.question.to_dict() == PollMedia(text="Test Poll 3").to_dict() |
| 514 | + assert msg_3.poll.expiry <= msg_3.created_at + timedelta(hours=1, minutes=1) |
| 515 | + poll_3_answer_medias = [poll_answer.poll_media.to_dict() for poll_answer in msg_3.poll.answers] |
| 516 | + assert poll_3_answer_medias == [ |
| 517 | + PollMedia.create(text="One", emoji="1️⃣").to_dict(), |
| 518 | + PollMedia.create(text="Two", emoji="2️⃣").to_dict(), |
| 519 | + ] |
| 520 | + |
| 521 | + finally: |
| 522 | + with suppress(interactions.errors.NotFound): |
| 523 | + await thread.delete() |
| 524 | + |
| 525 | + |
435 | 526 | @pytest.mark.asyncio |
436 | 527 | async def test_webhooks(bot: Client, guild: Guild, channel: GuildText) -> None: |
437 | 528 | test_thread = await channel.create_thread("Test Thread") |
|
0 commit comments