Tidy up the disabled snapshot test

This commit is contained in:
Dave Pearson
2023-02-14 16:20:52 +00:00
parent b4f8a6b778
commit 26df6aeb00

View File

@@ -1,13 +1,26 @@
from textual.app import App, ComposeResult from textual.app import App, ComposeResult
from textual.containers import Vertical from textual.containers import Vertical, Horizontal
from textual.widgets import ( from textual.widgets import (
Header, Footer, Button, DataTable, Input, ListView, ListItem, Label, Tree, TextLog Header,
Footer,
Button,
DataTable,
Input,
ListView,
ListItem,
Label,
Tree,
TextLog,
) )
class WidgetDisableTestApp( App[ None ] ):
class WidgetDisableTestApp(App[None]):
CSS = """ CSS = """
DataTable, ListView, Tree { Horizontal {
height: auto;
}
DataTable, ListView, Tree, TextLog {
height: 2; height: 2;
} }
""" """
@@ -15,30 +28,34 @@ class WidgetDisableTestApp( App[ None ] ):
@property @property
def data_table(self) -> DataTable: def data_table(self) -> DataTable:
data_table = DataTable() data_table = DataTable()
data_table.add_columns( "Column 1", "Column 2", "Column 3", "Column 4" ) data_table.add_columns("Column 1", "Column 2", "Column 3", "Column 4")
data_table.add_rows( data_table.add_rows(
[ ( str( n ), str( n * 10 ), str( n * 100 ), str( n * 1000 ) ) for n in range( 100 ) ] [(str(n), str(n * 10), str(n * 100), str(n * 1000)) for n in range(100)]
) )
return data_table return data_table
@property @property
def list_view( self ) -> ListView: def list_view(self) -> ListView:
return ListView( return ListView(*[ListItem(Label(f"This is list item {n}")) for n in range(20)])
*[ ListItem( Label( f"This is list item {n}" ) ) for n in range( 20 ) ]
)
@property @property
def test_tree( self ) -> Tree: def test_tree(self) -> Tree:
tree = Tree[None]( label="This is a test tree" ) tree = Tree[None](label="This is a test tree")
for n in range( 10 ): for n in range(10):
tree.root.add_leaf( f"Leaf {n}" ) tree.root.add_leaf(f"Leaf {n}")
tree.root.expand() tree.root.expand()
return tree return tree
def compose( self ) -> ComposeResult: def compose(self) -> ComposeResult:
yield Header() yield Header()
yield Vertical( yield Vertical(
Button(), Horizontal(
Button(),
Button(variant="primary"),
Button(variant="success"),
Button(variant="warning"),
Button(variant="error"),
),
self.data_table, self.data_table,
self.list_view, self.list_view,
self.test_tree, self.test_tree,
@@ -54,5 +71,6 @@ class WidgetDisableTestApp( App[ None ] ):
self.query_one(TextLog).write("Hello, World!") self.query_one(TextLog).write("Hello, World!")
self.query_one("#test-container", Vertical).disabled = True self.query_one("#test-container", Vertical).disabled = True
if __name__ == "__main__": if __name__ == "__main__":
WidgetDisableTestApp().run() WidgetDisableTestApp().run()