Force tap 'Done' button when using SafariViewController to fix flaky automation test

Author: juan-arias

MSAL/test/automation/tests/MSALBaseUITest.m

@@ -214,9 +214,21 @@ - (void)closeAuthUIUsingWebViewType:(MSIDWebviewType)webViewType
         // We take the second one and tap it
         XCUIElement *secondButton = [elementQuery elementBoundByIndex:1];
         [secondButton msidTap];
-    } else
+    } 
+    else
     {
-        [self.testApp.buttons[buttonTitle] msidTap];
+        if (webViewType == MSIDWebviewTypeSafariViewController)
+        {
+            // We take the first one and force tap it, for some reason tap doesn't work
+            XCUIElement *firstButton = [elementQuery elementBoundByIndex:0];
+            
+            __auto_type coordinate = [firstButton coordinateWithNormalizedOffset:CGVectorMake(0, 0)];
+            [coordinate tap];
+        }
+        else
+        {
+            [self.testApp.buttons[buttonTitle] msidTap];
+        }
     }
 }