Fix browser_tests failure.

chrome/browser/chromeos/input_method/xkeyboard.cc

 // Copyright (c) 2012 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.
 
 #include "chrome/browser/chromeos/input_method/xkeyboard.h"
 
 #include <cstdlib>
 #include <cstring>
 #include <queue>
 #include <set>
@@ skipping 265 matching lines @@
         break;
       }
     }
   }
   XkbFreeKeyboard(xkb_desc, 0, True /* free all components */);
   return real_mask;
 }
 
 void XKeyboardImpl::GetLockedModifiers(bool* out_caps_lock_enabled,
                                        bool* out_num_lock_enabled) {
-  // For now, don't call CHECK() here to make
-  // TabRestoreServiceTest.DontRestorePrintPreviewTab test happy.
-  // TODO(yusukes): Fix the test, then fix the if(!BrowserThread...) line below.
-  // CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
-
-  if (!BrowserThread::CurrentlyOn(BrowserThread::UI) ||
-      (out_num_lock_enabled && !num_lock_mask_)) {
-    VLOG(1) << "Cannot get locked modifiers. Num Lock mask unknown.";
+  // The method might be called by browser_tests running on Linux.
+  if (!is_running_on_chrome_os_ || (out_num_lock_enabled && !num_lock_mask_)) {
+    VLOG(1) << "Cannot get locked modifiers.";
     if (out_caps_lock_enabled) {
       *out_caps_lock_enabled = false;
     }
     if (out_num_lock_enabled) {
       *out_num_lock_enabled = false;
     }
     return;
   }
 
+  CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   XkbStateRec status;
   XkbGetState(ui::GetXDisplay(), XkbUseCoreKbd, &status);
   if (out_caps_lock_enabled) {
     *out_caps_lock_enabled = status.locked_mods & LockMask;
   }
   if (out_num_lock_enabled) {
     *out_num_lock_enabled = status.locked_mods & num_lock_mask_;
   }
 }
 
@@ skipping 242 matching lines @@
   return false;
 }
 
 // static
 XKeyboard* XKeyboard::Create(const InputMethodUtil& util) {
   return new XKeyboardImpl(util);
 }
 
 }  // namespace input_method
 }  // namespace chromeos