Add a timeout on the EG DrainUntilIdle

ios/chrome/test/earl_grey/chrome_test_case.mm

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #import "ios/chrome/test/earl_grey/chrome_test_case.h"
 
 #import <objc/runtime.h>
 
 #import <EarlGrey/EarlGrey.h>
 
@@ skipping 52 matching lines @@
   @"testSignInOneUser",               // SigninInteractionControllerTestCase
   @"testSwitchTabs",                  // StackViewTestCase
   @"testTabStripSwitchTabs",          // TabStripTestCase
   @"testTabHistoryMenu",              // TabHistoryPopupControllerTestCase
   @"testEnteringTabSwitcher",         // TabSwitcherControllerTestCase
   @"testEnterURL",                    // ToolbarTestCase
   @"testOpenAndCloseToolsMenu",       // ToolsPopupMenuTestCase
   @"testUserFeedbackPageOpenPrivacyPolicy",  // UserFeedbackTestCase
   @"testVersion",                            // WebUITestCase
 ];
+
+const CFTimeInterval kDrainTimeout = 5;
 }  // namespace
 
 @interface ChromeTestCase () {
   // Block to be executed during object tearDown.
   base::mac::ScopedBlock<ProceduralBlock> _tearDownHandler;
 
   BOOL _isHTTPServerStopped;
   BOOL _isMockAuthenticationDisabled;
 }
 
@@ skipping 114 matching lines @@
   DCHECK(!_tearDownHandler);
   _tearDownHandler.reset([tearDownHandler copy]);
 }
 
 + (void)removeAnyOpenMenusAndInfoBars {
   chrome_test_util::RemoveAllInfoBars();
   chrome_test_util::ClearPresentedState();
   // After programatically removing UI elements, allow Earl Grey's
   // UI synchronization to become idle, so subsequent steps won't start before
   // the UI is in a good state.
-  [[GREYUIThreadExecutor sharedInstance] drainUntilIdle];
+  [[GREYUIThreadExecutor sharedInstance]
+      drainUntilIdleWithTimeout:kDrainTimeout];
 }
 
 + (void)closeAllTabs {
   chrome_test_util::CloseAllTabs();
-  [[GREYUIThreadExecutor sharedInstance] drainUntilIdle];
+  [[GREYUIThreadExecutor sharedInstance]
+      drainUntilIdleWithTimeout:kDrainTimeout];
 }
 
 - (void)disableMockAuthentication {
   // Enforce that disableMockAuthentication can only be called once.
   DCHECK(!_isMockAuthenticationDisabled);
   [[self class] disableMockAuthentication];
   _isMockAuthenticationDisabled = YES;
 }
 
 - (void)stopHTTPServer {
@@ skipping 67 matching lines @@
           [NSInvocation invocationWithMethodSignature:methodSignature];
       invocation.selector = selector;
       [multitaskingTestNames addObject:invocation];
     }
   }
   free(methods);
   return multitaskingTestNames;
 }
 
 @end