Add more tests for colour, embeds, files, buttons and modals

tests/test_ui_buttons.py

@@ -0,0 +1,167 @@
+"""
+The MIT License (MIT)
+
+Copyright (c) 2015-present Rapptz
+
+Permission is hereby granted, free of charge, to any person obtaining a
+copy of this software and associated documentation files (the "Software"),
+to deal in the Software without restriction, including without limitation
+the rights to use, copy, modify, merge, publish, distribute, sublicense,
+and/or sell copies of the Software, and to permit persons to whom the
+Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+DEALINGS IN THE SOFTWARE.
+"""
+
+from __future__ import annotations
+
+import discord
+import pytest
+
+
+def test_button_init():
+    button = discord.ui.Button(
+        label="Click me!",
+    )
+    assert button.label == "Click me!"
+    assert button.style == discord.ButtonStyle.secondary
+    assert button.disabled == False
+    assert button.url == None
+    assert button.emoji == None
+    assert button.sku_id == None
+
+
+def test_button_with_sku_id():
+    button = discord.ui.Button(
+        label="Click me!",
+        sku_id=1234567890,
+    )
+    assert button.label == "Click me!"
+    assert button.style == discord.ButtonStyle.premium
+    assert button.sku_id == 1234567890
+
+
+def test_button_with_url():
+    button = discord.ui.Button(
+        label="Click me!",
+        url="https://example.com",
+    )
+    assert button.label == "Click me!"
+    assert button.style == discord.ButtonStyle.link
+    assert button.url == "https://example.com"
+
+
+def test_mix_both_custom_id_and_url():
+    with pytest.raises(TypeError):
+        discord.ui.Button(
+            label="Click me!",
+            url="https://example.com",
+            custom_id="test",
+        )
+
+
+def test_mix_both_custom_id_and_sku_id():
+    with pytest.raises(TypeError):
+        discord.ui.Button(
+            label="Click me!",
+            sku_id=1234567890,
+            custom_id="test",
+        )
+
+
+def test_mix_both_url_and_sku_id():
+    with pytest.raises(TypeError):
+        discord.ui.Button(
+            label="Click me!",
+            url="https://example.com",
+            sku_id=1234567890,
+        )
+
+
+def test_invalid_url():
+    button = discord.ui.Button(
+        label="Click me!",
+    )
+    with pytest.raises(TypeError):
+        button.url = 1234567890  # type: ignore
+
+
+def test_invalid_custom_id():
+    with pytest.raises(TypeError):
+        discord.ui.Button(
+            label="Click me!",
+            custom_id=1234567890,  # type: ignore
+        )
+
+    button = discord.ui.Button(
+        label="Click me!",
+    )
+    with pytest.raises(TypeError):
+        button.custom_id = 1234567890  # type: ignore
+
+
+def test_button_with_partial_emoji():
+    button = discord.ui.Button(
+        label="Click me!",
+        emoji="👍",
+    )
+    assert button.label == "Click me!"
+    assert button.emoji is not None and button.emoji.name == "👍"
+
+
+def test_button_with_str_emoji():
+    emoji = discord.PartialEmoji(name="👍")
+    button = discord.ui.Button(
+        label="Click me!",
+        emoji=emoji,
+    )
+    assert button.label == "Click me!"
+    assert button.emoji == emoji
+
+
+def test_button_with_invalid_emoji():
+    with pytest.raises(TypeError):
+        discord.ui.Button(
+            label="Click me!",
+            emoji=-0.53,  # type: ignore
+        )
+
+    button = discord.ui.Button(
+        label="Click me!",
+    )
+    with pytest.raises(TypeError):
+        button.emoji = -0.53  # type: ignore
+
+
+def test_button_setter():
+    button = discord.ui.Button()
+
+    button.label = "Click me!"
+    assert button.label == "Click me!"
+
+    button.style = discord.ButtonStyle.primary
+    assert button.style == discord.ButtonStyle.primary
+
+    button.disabled = True
+    assert button.disabled == True
+
+    button.url = "https://example.com"
+    assert button.url == "https://example.com"
+
+    button.emoji = "👍"
+    assert button.emoji is not None and button.emoji.name == "👍"  # type: ignore
+
+    button.custom_id = "test"
+    assert button.custom_id == "test"
+
+    button.sku_id = 1234567890
+    assert button.sku_id == 1234567890