fix test

Textualize · May 12, 2024 · 10887ee · 10887ee
1 parent 9e094e6
commit 10887ee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/textual/dom.py b/src/textual/dom.py
@@ -1234,7 +1234,7 @@ def walk_children(
         return cast("list[DOMNode]", nodes)
 
     @overload
-    def query(self, selector: str | None) -> DOMQuery[Widget]: ...
+    def query(self, selector: str | None = None) -> DOMQuery[Widget]: ...
 
     @overload
     def query(self, selector: type[QueryType]) -> DOMQuery[QueryType]: ...
@@ -1259,7 +1259,7 @@ def query(
             return DOMQuery[QueryType](self, filter=selector.__name__)
 
     @overload
-    def query_children(self, selector: str | None) -> DOMQuery[Widget]: ...
+    def query_children(self, selector: str | None = None) -> DOMQuery[Widget]: ...
 
     @overload
     def query_children(self, selector: type[QueryType]) -> DOMQuery[QueryType]: ...

diff --git a/tests/test_query.py b/tests/test_query.py
@@ -56,6 +56,7 @@ class App(Widget):
 
     # repeat tests to account for caching
     for repeat in range(3):
+        assert list(app.query_children()) == [main_view, help_view]
         assert list(app.query_children("*")) == [main_view, help_view]
         assert list(app.query_children("#help")) == [help_view]
         assert list(main_view.query_children(".float")) == [sidebar]