/*
 * Copyright (C) 2018 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.textclassifier.common.intent;

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

import android.content.Intent;
import android.net.Uri;
import android.view.textclassifier.TextClassifier;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import androidx.test.filters.SmallTest;
import com.google.android.textclassifier.NamedVariant;
import com.google.android.textclassifier.RemoteActionTemplate;
import java.util.List;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;

@SmallTest
@RunWith(AndroidJUnit4.class)
public class TemplateIntentFactoryTest {

  @Rule public final MockitoRule mocks = MockitoJUnit.rule();

  private static final String TITLE_WITHOUT_ENTITY = "Map";
  private static final String TITLE_WITH_ENTITY = "Map NW14D1";
  private static final String DESCRIPTION = "Check the map";
  private static final String DESCRIPTION_WITH_APP_NAME = "Use %1$s to open map";
  private static final String ACTION = Intent.ACTION_VIEW;
  private static final String DATA = Uri.parse("http://www.android.com").toString();
  private static final String TYPE = "text/html";
  private static final Integer FLAG = Intent.FLAG_ACTIVITY_NEW_TASK;
  private static final String[] CATEGORY =
      new String[] {Intent.CATEGORY_DEFAULT, Intent.CATEGORY_APP_BROWSER};
  private static final String PACKAGE_NAME = "pkg.name";
  private static final String KEY_ONE = "key1";
  private static final String VALUE_ONE = "value1";
  private static final String KEY_TWO = "key2";
  private static final int VALUE_TWO = 42;
  private static final String KEY_STRING_ARRAY = "string_array_key";
  private static final String[] VALUE_STRING_ARRAY = new String[] {"a", "b"};
  private static final String KEY_FLOAT_ARRAY = "float_array_key";
  private static final float[] VALUE_FLOAT_ARRAY = new float[] {3.14f, 2.718f};
  private static final String KEY_INT_ARRAY = "int_array_key";
  private static final int[] VALUE_INT_ARRAY = new int[] {7, 2, 1};

  private static final NamedVariant[] NAMED_VARIANTS =
      new NamedVariant[] {
        new NamedVariant(KEY_ONE, VALUE_ONE),
        new NamedVariant(KEY_TWO, VALUE_TWO),
        new NamedVariant(KEY_STRING_ARRAY, VALUE_STRING_ARRAY),
        new NamedVariant(KEY_FLOAT_ARRAY, VALUE_FLOAT_ARRAY),
        new NamedVariant(KEY_INT_ARRAY, VALUE_INT_ARRAY)
      };
  private static final Integer REQUEST_CODE = 10;

  private TemplateIntentFactory templateIntentFactory;

  @Before
  public void setup() {
    templateIntentFactory = new TemplateIntentFactory();
  }

  @Test
  public void create_full() {
    RemoteActionTemplate remoteActionTemplate =
        new RemoteActionTemplate(
            TITLE_WITHOUT_ENTITY,
            TITLE_WITH_ENTITY,
            DESCRIPTION,
            DESCRIPTION_WITH_APP_NAME,
            ACTION,
            DATA,
            TYPE,
            FLAG,
            CATEGORY,
            /* packageName */ null,
            NAMED_VARIANTS,
            REQUEST_CODE);

    List<LabeledIntent> intents =
        templateIntentFactory.create(new RemoteActionTemplate[] {remoteActionTemplate});

    assertThat(intents).hasSize(1);
    LabeledIntent labeledIntent = intents.get(0);
    assertThat(labeledIntent.titleWithoutEntity).isEqualTo(TITLE_WITHOUT_ENTITY);
    assertThat(labeledIntent.titleWithEntity).isEqualTo(TITLE_WITH_ENTITY);
    assertThat(labeledIntent.description).isEqualTo(DESCRIPTION);
    assertThat(labeledIntent.descriptionWithAppName).isEqualTo(DESCRIPTION_WITH_APP_NAME);
    assertThat(labeledIntent.requestCode).isEqualTo(REQUEST_CODE);
    Intent intent = labeledIntent.intent;
    assertThat(intent.getAction()).isEqualTo(ACTION);
    assertThat(intent.getData().toString()).isEqualTo(DATA);
    assertThat(intent.getType()).isEqualTo(TYPE);
    assertThat(intent.getFlags()).isEqualTo(FLAG);
    assertThat(intent.getCategories()).containsExactlyElementsIn(CATEGORY);
    assertThat(intent.getPackage()).isNull();
    assertThat(intent.getStringExtra(KEY_ONE)).isEqualTo(VALUE_ONE);
    assertThat(intent.getIntExtra(KEY_TWO, 0)).isEqualTo(VALUE_TWO);
    assertThat(intent.getStringArrayExtra(KEY_STRING_ARRAY)).isEqualTo(VALUE_STRING_ARRAY);
    assertThat(intent.getFloatArrayExtra(KEY_FLOAT_ARRAY)).isEqualTo(VALUE_FLOAT_ARRAY);
    assertThat(intent.getIntArrayExtra(KEY_INT_ARRAY)).isEqualTo(VALUE_INT_ARRAY);
    assertThat(intent.hasExtra(TextClassifier.EXTRA_FROM_TEXT_CLASSIFIER)).isTrue();
  }

  @Test
  public void normalizesScheme() {
    RemoteActionTemplate remoteActionTemplate =
        new RemoteActionTemplate(
            TITLE_WITHOUT_ENTITY,
            TITLE_WITH_ENTITY,
            DESCRIPTION,
            DESCRIPTION_WITH_APP_NAME,
            ACTION,
            "HTTp://www.android.com",
            TYPE,
            FLAG,
            CATEGORY,
            /* packageName */ null,
            NAMED_VARIANTS,
            REQUEST_CODE);

    List<LabeledIntent> intents =
        templateIntentFactory.create(new RemoteActionTemplate[] {remoteActionTemplate});

    String data = intents.get(0).intent.getData().toString();
    assertThat(data).isEqualTo("http://www.android.com");
  }

  @Test
  public void create_minimal() {
    RemoteActionTemplate remoteActionTemplate =
        new RemoteActionTemplate(
            TITLE_WITHOUT_ENTITY,
            null,
            DESCRIPTION,
            null,
            ACTION,
            null,
            null,
            null,
            null,
            null,
            null,
            null);

    List<LabeledIntent> intents =
        templateIntentFactory.create(new RemoteActionTemplate[] {remoteActionTemplate});

    assertThat(intents).hasSize(1);
    LabeledIntent labeledIntent = intents.get(0);
    assertThat(labeledIntent.titleWithoutEntity).isEqualTo(TITLE_WITHOUT_ENTITY);
    assertThat(labeledIntent.titleWithEntity).isNull();
    assertThat(labeledIntent.description).isEqualTo(DESCRIPTION);
    assertThat(labeledIntent.requestCode).isEqualTo(LabeledIntent.DEFAULT_REQUEST_CODE);
    Intent intent = labeledIntent.intent;
    assertThat(intent.getAction()).isEqualTo(ACTION);
    assertThat(intent.getData()).isNull();
    assertThat(intent.getType()).isNull();
    assertThat(intent.getFlags()).isEqualTo(0);
    assertThat(intent.getCategories()).isNull();
    assertThat(intent.getPackage()).isNull();
  }

  @Test
  public void invalidTemplate_nullTemplate() {
    RemoteActionTemplate remoteActionTemplate = null;

    List<LabeledIntent> intents =
        templateIntentFactory.create(new RemoteActionTemplate[] {remoteActionTemplate});

    assertThat(intents).isEmpty();
  }

  @Test
  public void invalidTemplate_nonEmptyPackageName() {
    RemoteActionTemplate remoteActionTemplate =
        new RemoteActionTemplate(
            TITLE_WITHOUT_ENTITY,
            TITLE_WITH_ENTITY,
            DESCRIPTION,
            DESCRIPTION_WITH_APP_NAME,
            ACTION,
            DATA,
            TYPE,
            FLAG,
            CATEGORY,
            PACKAGE_NAME,
            NAMED_VARIANTS,
            REQUEST_CODE);

    List<LabeledIntent> intents =
        templateIntentFactory.create(new RemoteActionTemplate[] {remoteActionTemplate});

    assertThat(intents).isEmpty();
  }

  @Test
  public void invalidTemplate_emptyTitle() {
    RemoteActionTemplate remoteActionTemplate =
        new RemoteActionTemplate(
            null,
            null,
            DESCRIPTION,
            DESCRIPTION_WITH_APP_NAME,
            ACTION,
            null,
            null,
            null,
            null,
            null,
            null,
            null);

    List<LabeledIntent> intents =
        templateIntentFactory.create(new RemoteActionTemplate[] {remoteActionTemplate});

    assertThat(intents).isEmpty();
  }

  @Test
  public void invalidTemplate_emptyDescription() {
    RemoteActionTemplate remoteActionTemplate =
        new RemoteActionTemplate(
            TITLE_WITHOUT_ENTITY,
            TITLE_WITH_ENTITY,
            null,
            null,
            ACTION,
            null,
            null,
            null,
            null,
            null,
            null,
            null);

    List<LabeledIntent> intents =
        templateIntentFactory.create(new RemoteActionTemplate[] {remoteActionTemplate});

    assertThat(intents).isEmpty();
  }

  @Test
  public void invalidTemplate_emptyIntentAction() {
    RemoteActionTemplate remoteActionTemplate =
        new RemoteActionTemplate(
            TITLE_WITHOUT_ENTITY,
            TITLE_WITH_ENTITY,
            DESCRIPTION,
            DESCRIPTION_WITH_APP_NAME,
            null,
            null,
            null,
            null,
            null,
            null,
            null,
            null);

    List<LabeledIntent> intents =
        templateIntentFactory.create(new RemoteActionTemplate[] {remoteActionTemplate});

    assertThat(intents).isEmpty();
  }
}