Expand the IME test framework to test the Compact layout.

chrome/test/data/chromeos/virtual_keyboard/inputview/keyset_transition_test.js

 /*
  * Copyright 2014 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.
  */
 
 function testShiftHighlight() {
   // Start in lower case.
   mockTouchType('l');
-  var shift = getShiftKey(Alignment.LEFT);
+  var shift = getKey("leftShift");
   generateTouchEvent(shift, 'touchstart', true, true);
   generateTouchEvent(shift, 'touchend', true, true);
   // Transitioned to upper case.
   mockTouchType('A');
   // Should revert to lower case.
   mockTouchType('p');
   // Should remain in lower case.
   mockTouchType('c');
 }
 
 function testCapslock() {
+  // Skip this test for compact layouts since they don't have capslock keys.
+  var id = getLayoutId();
+  if (id.indexOf("compact") > 0)
+    return;
+
   // Start in lower case.
   mockTouchType('l');
   // To upper case.
   // TODO(rsadam@): Only test this for the full layout.
-  var caps = document.querySelector('#' + CAPSLOCK_ID);
+  var caps = getKey("capslock")
   generateTouchEvent(caps, 'touchstart', true, true);
   generateTouchEvent(caps, 'touchend', true, true);
   mockTouchType('A');
   // Should persist upper case.
   mockTouchType('P');
   mockTouchType('C');
   // Back to lower case.
   generateTouchEvent(caps, 'touchstart', true, true);
   generateTouchEvent(caps, 'touchend', true, true);
   mockTouchType('p');
@@ skipping 10 matching lines @@
   // Should persist upper case.
   mockMouseType('P');
   mockMouseType('C');
   // Back to lower case.
   mockMouseTypeOnKey(caps);
   mockMouseType('p');
   // Persist lower case.
   mockMouseType('c')
   mockMouseType('d')
 }