mirror of
https://github.com/Textualize/textual.git
synced 2025-10-17 02:38:12 +03:00
0a9b9603a0
* Add tests for errors when removing things that don't exist * Add tests for errors when toggling enabled/disabled on invalid options
102 lines
3.9 KiB
Python
102 lines
3.9 KiB
Python
"""Test removing options from an option list."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from textual.app import App, ComposeResult
|
|
from textual.widgets import OptionList
|
|
from textual.widgets.option_list import Option, OptionDoesNotExist
|
|
|
|
|
|
class OptionListApp(App[None]):
|
|
"""Test option list application."""
|
|
|
|
def compose(self) -> ComposeResult:
|
|
yield OptionList(
|
|
Option("0", id="0"),
|
|
Option("1", id="1"),
|
|
)
|
|
|
|
|
|
async def test_remove_first_option_via_index() -> None:
|
|
"""It should be possible to remove the first option of an option list, via index."""
|
|
async with OptionListApp().run_test() as pilot:
|
|
option_list = pilot.app.query_one(OptionList)
|
|
assert option_list.option_count == 2
|
|
assert option_list.highlighted == 0
|
|
option_list.remove_option_at_index(0)
|
|
assert option_list.option_count == 1
|
|
assert option_list.highlighted == 0
|
|
|
|
|
|
async def test_remove_first_option_via_id() -> None:
|
|
"""It should be possible to remove the first option of an option list, via ID."""
|
|
async with OptionListApp().run_test() as pilot:
|
|
option_list = pilot.app.query_one(OptionList)
|
|
assert option_list.option_count == 2
|
|
assert option_list.highlighted == 0
|
|
option_list.remove_option("0")
|
|
assert option_list.option_count == 1
|
|
assert option_list.highlighted == 0
|
|
|
|
|
|
async def test_remove_last_option_via_index() -> None:
|
|
"""It should be possible to remove the last option of an option list, via index."""
|
|
async with OptionListApp().run_test() as pilot:
|
|
option_list = pilot.app.query_one(OptionList)
|
|
assert option_list.option_count == 2
|
|
assert option_list.highlighted == 0
|
|
option_list.remove_option_at_index(1)
|
|
assert option_list.option_count == 1
|
|
assert option_list.highlighted == 0
|
|
|
|
|
|
async def test_remove_last_option_via_id() -> None:
|
|
"""It should be possible to remove the last option of an option list, via ID."""
|
|
async with OptionListApp().run_test() as pilot:
|
|
option_list = pilot.app.query_one(OptionList)
|
|
assert option_list.option_count == 2
|
|
assert option_list.highlighted == 0
|
|
option_list.remove_option("1")
|
|
assert option_list.option_count == 1
|
|
assert option_list.highlighted == 0
|
|
|
|
|
|
async def test_remove_all_options_via_index() -> None:
|
|
"""It should be possible to remove all options via index."""
|
|
async with OptionListApp().run_test() as pilot:
|
|
option_list = pilot.app.query_one(OptionList)
|
|
assert option_list.option_count == 2
|
|
assert option_list.highlighted == 0
|
|
option_list.remove_option_at_index(0)
|
|
option_list.remove_option_at_index(0)
|
|
assert option_list.option_count == 0
|
|
assert option_list.highlighted is None
|
|
|
|
|
|
async def test_remove_all_options_via_id() -> None:
|
|
"""It should be possible to remove all options via ID."""
|
|
async with OptionListApp().run_test() as pilot:
|
|
option_list = pilot.app.query_one(OptionList)
|
|
assert option_list.option_count == 2
|
|
assert option_list.highlighted == 0
|
|
option_list.remove_option("0")
|
|
option_list.remove_option("1")
|
|
assert option_list.option_count == 0
|
|
assert option_list.highlighted is None
|
|
|
|
|
|
async def test_remove_invalid_id() -> None:
|
|
"""Attempting to remove an option ID that doesn't exist should raise an exception."""
|
|
async with OptionListApp().run_test() as pilot:
|
|
with pytest.raises(OptionDoesNotExist):
|
|
pilot.app.query_one(OptionList).remove_option("does-not-exist")
|
|
|
|
|
|
async def test_remove_invalid_index() -> None:
|
|
"""Attempting to remove an option index that doesn't exist should raise an exception."""
|
|
async with OptionListApp().run_test() as pilot:
|
|
with pytest.raises(OptionDoesNotExist):
|
|
pilot.app.query_one(OptionList).remove_option_at_index(23)
|