Get mainViewController in automation server

Author: jonathanKingston

DuckDuckGo/AppLifecycle/AppStates/Launching.swift

@@ -183,9 +183,13 @@ struct Launching: AppState {
 
     private func startAutomationServerIfNeeded() {
         let launchOptionsHandler = LaunchOptionsHandler()
-        if launchOptionsHandler.isUITesting && launchOptionsHandler.automationPort != nil {
-            AutomationServer(main: mainViewController!, port: launchOptionsHandler.automationPort)
+        guard launchOptionsHandler.isUITesting && launchOptionsHandler.automationPort != nil else {
+            return
         }
+        guard let rootViewController = window.rootViewController as? MainViewController else {
+            return
+        }
+        AutomationServer(main: rootViewController, port: launchOptionsHandler.automationPort)
     }
 }
 

DuckDuckGo/TabViewController+Automation.swift

@@ -28,7 +28,7 @@ extension TabViewController {
                 arguments: args,
                 in: nil,
                 contentWorld: .page
-            )
+            ) ?? ""
             return .success(result)
         } catch {
             return .failure(error)