fix select

Author: willmcgugan

src/textual/widgets/_option_list.py

@@ -363,6 +363,7 @@ def _add_lines(
 
         self.virtual_size = Size(width, len(self._lines))
         self.refresh(layout=self.styles.auto_dimensions)
+        self._scroll_update(self.virtual_size)
 
     def _populate(self) -> None:
         """Populate the lines data-structure."""
@@ -879,9 +880,12 @@ def scroll_to_highlight(self, top: bool = False) -> None:
             top: Scroll highlight to top of the list.
         """
         highlighted = self.highlighted
-        if highlighted is None or self._spans is None:
+        if highlighted is None or not self.is_mounted:
             return
 
+        if not self._spans:
+            self._populate()
+
         try:
             y, height = self._spans[highlighted]
         except IndexError:
@@ -894,6 +898,7 @@ def scroll_to_highlight(self, top: bool = False) -> None:
             force=True,
             animate=False,
             top=top,
+            immediate=True,
         )
 
     def validate_highlighted(self, highlighted: int | None) -> int | None:

src/textual/widgets/_select.py

@@ -66,7 +66,7 @@ def select(self, index: int | None) -> None:
             index: Index of new selection.
         """
         self.highlighted = index
-        self.scroll_to_highlight(top=True)
+        self.scroll_to_highlight()
 
     def action_dismiss(self) -> None:
         """Dismiss the overlay."""
@@ -520,7 +520,7 @@ def _watch_expanded(self, expanded: bool) -> None:
                 value = self.value
                 for index, (_prompt, prompt_value) in enumerate(self._options):
                     if value == prompt_value:
-                        overlay.select(index)
+                        self.call_after_refresh(overlay.select, index)
                         break
                 self.query_one(SelectCurrent).has_value = True