fix tests: await all mounts.

This commit is contained in:
Rodrigo Girão Serrão
2023-01-13 10:58:57 +00:00
parent feb8591e72
commit 7e428e2f4c

View File

@@ -42,8 +42,8 @@ async def test_input_value_visible_if_mounted_later():
class MyApp(App): class MyApp(App):
BINDINGS = [("a", "add_input", "add_input")] BINDINGS = [("a", "add_input", "add_input")]
def action_add_input(self): async def action_add_input(self):
self.mount(Input(value="value")) await self.mount(Input(value="value"))
app = MyApp() app = MyApp()
async with app.run_test() as pilot: async with app.run_test() as pilot:
@@ -83,8 +83,8 @@ async def test_input_value_visible_if_mounted_later_and_assigned_after():
("v", "set_value", "set_value"), ("v", "set_value", "set_value"),
] ]
def action_add_input(self): async def action_add_input(self):
self.mount(Input()) await self.mount(Input())
def action_set_value(self): def action_set_value(self):
self.query_one(Input).value = "value" self.query_one(Input).value = "value"