LatinIME/tests/src/com/android/inputmethod/latin/ShiftModeTests.java

/*
 * Copyright (C) 2014 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.inputmethod.latin;

import android.os.Build;
import android.test.suitebuilder.annotation.LargeTest;
import android.text.TextUtils;
import android.view.inputmethod.EditorInfo;

import com.android.inputmethod.latin.Constants;
import com.android.inputmethod.latin.WordComposer;

@LargeTest
public class ShiftModeTests extends InputTestsBase {

    @Override
    protected EditorInfo enrichEditorInfo(final EditorInfo ei) {
        ei.inputType |= TextUtils.CAP_MODE_SENTENCES;
        ei.initialCapsMode = TextUtils.CAP_MODE_SENTENCES;
        return ei;
    }

    private boolean isCapsModeAutoShifted() {
        return mLatinIME.mKeyboardSwitcher.getKeyboardShiftMode()
                == WordComposer.CAPS_MODE_AUTO_SHIFTED;
    }

    public void testTypicalSentence() {
        assertTrue("Initial auto caps state", isCapsModeAutoShifted());
        type("Test");
        assertFalse("Caps after letter", isCapsModeAutoShifted());
        type(" ");
        assertFalse("Caps after space", isCapsModeAutoShifted());
        type("some,");
        assertFalse("Caps after comma", isCapsModeAutoShifted());
        type(" ");
        assertFalse("Caps after comma space", isCapsModeAutoShifted());
        type("words.");
        assertFalse("Caps directly after period", isCapsModeAutoShifted());
        type(" ");
        assertTrue("Caps after period space", isCapsModeAutoShifted());
    }

    public void testBackspace() {
        assertTrue("Initial auto caps state", isCapsModeAutoShifted());
        type("A");
        assertFalse("Caps state after one letter", isCapsModeAutoShifted());
        type(Constants.CODE_DELETE);
        assertTrue("Auto caps state at start after delete", isCapsModeAutoShifted());
    }

    public void testRepeatingBackspace() {
        final String SENTENCE_TO_TYPE = "Test sentence. Another.";
        final int BACKSPACE_COUNT =
                SENTENCE_TO_TYPE.length() - SENTENCE_TO_TYPE.lastIndexOf(' ') - 1;

        type(SENTENCE_TO_TYPE);
        assertFalse("Caps after typing \"" + SENTENCE_TO_TYPE + "\"", isCapsModeAutoShifted());
        type(Constants.CODE_DELETE);
        for (int i = 1; i < BACKSPACE_COUNT; ++i) {
            repeatKey(Constants.CODE_DELETE);
        }
        assertFalse("Caps immediately after repeating Backspace a lot", isCapsModeAutoShifted());
        sleep(DELAY_TO_WAIT_FOR_PREDICTIONS);
        runMessages();
        assertTrue("Caps after a while after repeating Backspace a lot", isCapsModeAutoShifted());
    }

    public void testAutoCapsAfterDigitsPeriod() {
        changeLanguage("en");
        type("On 22.11.");
        assertFalse("(English) Auto caps after digits-period", isCapsModeAutoShifted());
        type(" ");
        assertTrue("(English) Auto caps after digits-period-whitespace", isCapsModeAutoShifted());
        mEditText.setText("");
        changeLanguage("fr");
        type("Le 22.");
        assertFalse("(French) Auto caps after digits-period", isCapsModeAutoShifted());
        type(" ");
        assertTrue("(French) Auto caps after digits-period-whitespace", isCapsModeAutoShifted());
        mEditText.setText("");
        changeLanguage("de");
        type("Am 22.");
        assertFalse("(German) Auto caps after digits-period", isCapsModeAutoShifted());
        type(" ");
        // For German, no auto-caps in this case
        assertFalse("(German) Auto caps after digits-period-whitespace", isCapsModeAutoShifted());
    }

    public void testAutoCapsAfterInvertedMarks() {
        changeLanguage("es");
        assertTrue("(Spanish) Auto caps at start", isCapsModeAutoShifted());
        type("Hey. ¿");
        assertTrue("(Spanish) Auto caps after inverted what", isCapsModeAutoShifted());
        mEditText.setText("");
        type("¡");
        assertTrue("(Spanish) Auto caps after inverted bang", isCapsModeAutoShifted());
    }

    public void DISABLED_testOtherSentenceSeparators() {
        // We only run this test on Kitkat+ because previous versions of Android don't
        // have an Armenian locale. For some reason I don't know, when the requested
        // locale is not present as a device locale, then the application under test can't
        // access the resources in that locale -- though it works when the app is actually
        // running on the device and not under test. If we ever figure out what's going
        // on, remove this test.
        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.KITKAT) {
            changeLanguage("hy-AM");
            assertTrue("(Armenian) Auto caps at start", isCapsModeAutoShifted());
            type("Hey. ");
            assertFalse("(Armenian) No auto-caps after latin period", isCapsModeAutoShifted());
            type("Hey\u0589");
            assertFalse("(Armenian) No auto-caps directly after armenian period",
                    isCapsModeAutoShifted());
            type(" ");
            assertTrue("(Armenian) Auto-caps after armenian period-whitespace",
                    isCapsModeAutoShifted());
        }
    }
}