Expand the IME test framework to test the Compact layout.

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

Index: chrome/test/data/chromeos/virtual_keyboard/inputview/keyset_transition_test.js
diff --git a/chrome/test/data/chromeos/virtual_keyboard/inputview/keyset_transition_test.js b/chrome/test/data/chromeos/virtual_keyboard/inputview/keyset_transition_test.js
index 79cd975c070d0dc8fe86d2eb28ef7faed400af5a..082bb5729e6909eedf9bdce931a2b7fdd946d52b 100644
--- a/chrome/test/data/chromeos/virtual_keyboard/inputview/keyset_transition_test.js
+++ b/chrome/test/data/chromeos/virtual_keyboard/inputview/keyset_transition_test.js
@@ -7,7 +7,7 @@
 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.
@@ -19,11 +19,16 @@ function testShiftHighlight() {
 }
 
 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');