Skip to content

Commit

Permalink
fix(test): no longer require stripping cls to origin type
Browse files Browse the repository at this point in the history
  • Loading branch information
shiftinv committed Dec 25, 2024
1 parent 0eca7b5 commit 309f7e7
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions tests/ui/test_decorators.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,14 +41,14 @@ def test_default(self) -> None:
res = ui.button(custom_id="123")(func)
assert_type(res, ui.item.DecoratedItem[ui.Button[_CustomView]])

assert func.__discord_ui_model_type__ is ui.Button
assert func.__discord_ui_model_type__ is ui.Button[Any]
assert func.__discord_ui_model_kwargs__ == {"custom_id": "123"}

with create_callback(_CustomView, ui.StringSelect[ui.View]) as func:
res = ui.string_select(custom_id="123")(func)
assert_type(res, ui.item.DecoratedItem[ui.StringSelect[_CustomView]])

assert func.__discord_ui_model_type__ is ui.StringSelect
assert func.__discord_ui_model_type__ is ui.StringSelect[Any]
assert func.__discord_ui_model_kwargs__ == {"custom_id": "123"}

# from here on out we're mostly only testing the button decorator,
Expand All @@ -60,8 +60,7 @@ def test_cls(self, cls: Type[_CustomButton[ui.View]]) -> None:
res = ui.button(cls=cls, param=1337)(func)
assert_type(res, ui.item.DecoratedItem[cls])

# should strip to origin type
assert func.__discord_ui_model_type__ is _CustomButton
assert func.__discord_ui_model_type__ is cls
assert func.__discord_ui_model_kwargs__ == {"param": 1337}

# typing-only check
Expand Down

0 comments on commit 309f7e7

Please sign in to comment.