/*
 * Copyright (C) 2017 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 android.autofillservice.cts;

import static com.google.common.truth.Truth.assertThat;

import static org.testng.Assert.assertThrows;

import android.platform.test.annotations.AppModeFull;
import android.service.autofill.TextValueSanitizer;
import android.view.autofill.AutofillValue;

import androidx.test.runner.AndroidJUnit4;

import org.junit.Test;
import org.junit.runner.RunWith;

import java.util.regex.Pattern;

@RunWith(AndroidJUnit4.class)
@AppModeFull // Unit test
public class TextValueSanitizerTest {

    @Test
    public void testConstructor_nullValues() {
        assertThrows(NullPointerException.class,
                () -> new TextValueSanitizer(Pattern.compile("42"), null));
        assertThrows(NullPointerException.class,
                () -> new TextValueSanitizer(null, "42"));
    }

    @Test
    public void testSanitize_nullValue() {
        final TextValueSanitizer sanitizer = new TextValueSanitizer(Pattern.compile("42"), "42");
        assertThat(sanitizer.sanitize(null)).isNull();
    }

    @Test
    public void testSanitize_nonTextValue() {
        final TextValueSanitizer sanitizer = new TextValueSanitizer(Pattern.compile("42"), "42");
        final AutofillValue value = AutofillValue.forToggle(true);
        assertThat(sanitizer.sanitize(value)).isNull();
    }

    @Test
    public void testSanitize_badRegex() {
        final TextValueSanitizer sanitizer = new TextValueSanitizer(Pattern.compile(".*(\\d*).*"),
                "$2"); // invalid group
        final AutofillValue value = AutofillValue.forText("blah 42  blaH");
        assertThat(sanitizer.sanitize(value)).isNull();
    }

    @Test
    public void testSanitize_valueMismatch() {
        final TextValueSanitizer sanitizer = new TextValueSanitizer(Pattern.compile("42"), "xxx");
        final AutofillValue value = AutofillValue.forText("43");
        assertThat(sanitizer.sanitize(value)).isNull();
    }

    @Test
    public void testSanitize_simpleMatch() {
        final TextValueSanitizer sanitizer = new TextValueSanitizer(Pattern.compile("42"),
                "forty-two");
        assertThat(sanitizer.sanitize(AutofillValue.forText("42")).getTextValue())
            .isEqualTo("forty-two");
    }

    @Test
    public void testSanitize_multipleMatches() {
        final TextValueSanitizer sanitizer = new TextValueSanitizer(Pattern.compile(".*(\\d*).*"),
                "Number");
        assertThat(sanitizer.sanitize(AutofillValue.forText("blah 42  blaH")).getTextValue())
            .isEqualTo("NumberNumber");
    }

    @Test
    public void testSanitize_groupSubstitutionMatch() {
        final TextValueSanitizer sanitizer =
                new TextValueSanitizer(Pattern.compile("\\s*(\\d*)\\s*"), "$1");
        assertThat(sanitizer.sanitize(AutofillValue.forText("  42 ")).getTextValue())
                .isEqualTo("42");
    }
}