Files
textual/tests/listview/test_inherit_listview.py
Dave Pearson 76272bcefe Extend ListView test to test an empty inherited list
For testing permutations of #1588.
2023-01-18 10:42:42 +00:00

46 lines
1.4 KiB
Python

from textual.app import App, ComposeResult
from textual.widgets import ListView, ListItem, Label
class MyListView(ListView):
"""Test child class of a ListView."""
def __init__(self, items: int = 0) -> None:
super().__init__()
self._items = items
def compose(self) -> ComposeResult:
"""Compose the child widgets."""
for n in range(self._items):
yield ListItem(Label(f"This is item {n}"))
class ListViewApp(App[None]):
"""ListView test app."""
def __init__(self, items: int = 0) -> None:
super().__init__()
self._items = items
def compose(self) -> ComposeResult:
"""Compose the child widgets."""
yield MyListView(self._items)
async def test_empty_inherited_list_view() -> None:
"""An empty self-populating inherited ListView should work as expected."""
async with ListViewApp().run_test() as pilot:
await pilot.press("tab")
assert pilot.app.query_one(MyListView).index is None
await pilot.press("down")
assert pilot.app.query_one(MyListView).index is None
async def test_populated_inherited_list_view() -> None:
"""A self-populating inherited ListView should work as normal."""
async with ListViewApp(30).run_test() as pilot:
await pilot.press("tab")
assert pilot.app.query_one(MyListView).index == 0
await pilot.press("down")
assert pilot.app.query_one(MyListView).index == 1