fix codemirror test

zauberzeug · May 10, 2024 · 154cb73 · 154cb73
1 parent 26e7fd2
commit 154cb73
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/tests/test_codemirror.py b/tests/test_codemirror.py
@@ -13,16 +13,21 @@ def test_codemirror(screen: Screen):
 
 def test_supported_values(screen: Screen):
     values: dict[str, List[str]] = {}
-    editor = ui.codemirror()
 
-    async def fetch():
-        values['languages'] = await editor.run_method('getLanguages')
-        values['themes'] = await editor.run_method('getThemes')
-        ui.label('Done')
-    ui.button('Fetch', on_click=fetch)
+    @ui.page('/')
+    def page():
+        editor = ui.codemirror()
+
+        async def fetch():
+            values['languages'] = await editor.run_method('getLanguages')
+            values['themes'] = await editor.run_method('getThemes')
+            values['supported_themes'] = editor.supported_themes
+            values['supported_languages'] = editor.supported_languages
+            ui.label('Done')
+        ui.button('Fetch', on_click=fetch)
 
     screen.open('/')
     screen.click('Fetch')
     screen.wait_for('Done')
-    assert values['languages'] == editor.supported_languages
-    assert values['themes'] == editor.supported_themes
+    assert values['languages'] == values['supported_languages']
+    assert values['themes'] == values['supported_themes']