Merge branch 'qa-fixes' into 'enterprise'

Fix slow load in customer instances & snowflake connection form

See merge request dkinternal/testgen/dataops-testgen!212

Author: ci bot

testgen/ui/navigation/router.py

@@ -59,7 +59,11 @@ def run(self) -> None:
             # This hack is needed because the auth cookie is not retrieved on the first run
             # We have to store the page and wait until cookies are ready
             session.page_pending_cookies = current_page
-            st.rerun()
+
+            # Don't use st.rerun() here!
+            # It will work fine locally, but cause a long initial load on deployed instances
+            # The time.sleep somehow causes the cookie to be detected quicker
+            time.sleep(0.3)
 
     def queue_navigation(self, /, to: str, with_args: dict | None = None) -> None:
         self._pending_navigation = {"to": to, "with_args": with_args or {}}

testgen/ui/views/connections/page.py

@@ -98,6 +98,7 @@ def show_connection_form(self, selected_connection: dict, _mode: str, project_co
             FlavorForm = BaseConnectionForm.for_flavor(sql_flavor)
             if connection:
                 connection["password"] = connection["password"] or ""
+                connection["private_key"] = connection["private_key"] or ""
 
             form_kwargs = connection or {"sql_flavor": sql_flavor, "connection_id": connection_id, "connection_name": connection_name}
             form = FlavorForm(**form_kwargs)