Customize login test utils to wait for cookies to be set.

chrome/browser/ui/webui/signin/login_ui_test_utils.h

Index: chrome/browser/ui/webui/signin/login_ui_test_utils.h
diff --git a/chrome/browser/ui/webui/signin/login_ui_test_utils.h b/chrome/browser/ui/webui/signin/login_ui_test_utils.h
index bf9a11051041af161ad1597ee3b8be20de348c58..acddd3934aeb888109aae6ad4c13d116a4c5993c 100644
--- a/chrome/browser/ui/webui/signin/login_ui_test_utils.h
+++ b/chrome/browser/ui/webui/signin/login_ui_test_utils.h
@@ -7,6 +7,8 @@
 
 #include <string>
 
+#include "components/signin/core/browser/signin_metrics.h"
+
 class Browser;
 
 namespace login_ui_test_utils {
@@ -41,6 +43,17 @@ void SigninInOldGaiaFlow(Browser* browser,
 
 // A function to sign in a user using Chrome sign-in UI interface.
 // This will block until a signin succeeded or failed notification is observed.
+// In case |wait_for_account_cookies|, the call will block until the account
+// cookies have been written to the cookie jar.
+// |signin_source| identifies the source used to load the signin page.
+bool SignInWithUI(Browser* browser,
+                  const std::string& email,
+                  const std::string& password,
+                  bool wait_for_account_cookies,
+                  signin_metrics::Source signin_source);
+
+// Most common way to sign in a user, it does not wait for cookies to be set
+// and uses the SOURCE_START_PAGE as signin source.
 bool SignInWithUI(Browser* browser,
                   const std::string& email,
                   const std::string& password);