/*
 * Copyright 2009 Google Inc.
 *
 * 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.google.template.soy.bididirectives;

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

import com.google.common.collect.ImmutableList;
import com.google.inject.util.Providers;
import com.google.template.soy.data.Dir;
import com.google.template.soy.data.SanitizedContents;
import com.google.template.soy.internal.i18n.BidiGlobalDir;
import com.google.template.soy.jssrc.restricted.JsExpr;
import com.google.template.soy.pysrc.restricted.PyExpr;
import com.google.template.soy.pysrc.restricted.PyStringExpr;
import com.google.template.soy.shared.AbstractSoyPrintDirectiveTestCase;
import com.google.template.soy.shared.SharedRestrictedTestUtils;

/**
 * Unit tests for BidiSpanWrapDirective.
 *
 */
public class BidiSpanWrapDirectiveTest extends AbstractSoyPrintDirectiveTestCase {


  private static final BidiSpanWrapDirective BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_LTR =
      new BidiSpanWrapDirective(Providers.of(BidiGlobalDir.LTR));

  private static final BidiSpanWrapDirective BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_RTL =
      new BidiSpanWrapDirective(Providers.of(BidiGlobalDir.RTL));


  public void testApplyForTofu() {
    assertTofuOutput("", "", BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_LTR);
    assertTofuOutput("blah", "blah", BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_LTR);
    assertTofuOutput("<span dir=\"rtl\">\u05E0</span>\u200E", "\u05E0",
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_LTR);

    assertTofuOutput("<span dir=\"rtl\">\u05E0</span>\u200E",
        SanitizedContents.unsanitizedText("\u05E0"),
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_LTR);
    assertTofuOutput("<span dir=\"rtl\">\u05E0</span>\u200E",
        SanitizedContents.unsanitizedText("\u05E0", Dir.RTL),
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_LTR);
    assertTofuOutput("\u05E0\u200E",
        SanitizedContents.unsanitizedText("\u05E0", Dir.LTR),
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_LTR);
    assertTofuOutput("\u05E0\u200E",
        SanitizedContents.unsanitizedText("\u05E0", Dir.NEUTRAL),
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_LTR);
    assertTofuOutput("blah",
        SanitizedContents.unsanitizedText("blah", Dir.NEUTRAL),
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_LTR);

    assertTofuOutput("", "", BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_RTL);
    assertTofuOutput("\u05E0", "\u05E0", BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_RTL);
    assertTofuOutput("<span dir=\"ltr\">blah</span>\u200F", "blah",
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_RTL);

    assertTofuOutput("<span dir=\"ltr\">blah</span>\u200F",
        SanitizedContents.unsanitizedText("blah"),
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_RTL);
    assertTofuOutput("<span dir=\"ltr\">blah</span>\u200F",
        SanitizedContents.unsanitizedText("blah", Dir.LTR),
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_RTL);
    assertTofuOutput("blah\u200F",
        SanitizedContents.unsanitizedText("blah", Dir.RTL),
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_RTL);
    assertTofuOutput("blah\u200F",
        SanitizedContents.unsanitizedText("blah", Dir.NEUTRAL),
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_RTL);
    assertTofuOutput("\u05E0",
        SanitizedContents.unsanitizedText("\u05E0", Dir.NEUTRAL),
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_RTL);
  }

  public void testApplyForJsSrc() {
    JsExpr dataRef = new JsExpr("opt_data.myKey", Integer.MAX_VALUE);
    assertThat(
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_LTR.applyForJsSrc(dataRef, ImmutableList.<JsExpr>of())
            .getText()).isEqualTo("soy.$$bidiSpanWrap(1, opt_data.myKey)");
    assertThat(
        BIDI_SPAN_WRAP_DIRECTIVE_FOR_STATIC_RTL.applyForJsSrc(dataRef, ImmutableList.<JsExpr>of())
            .getText()).isEqualTo("soy.$$bidiSpanWrap(-1, opt_data.myKey)");

    BidiSpanWrapDirective codeSnippet = new BidiSpanWrapDirective(
        SharedRestrictedTestUtils.BIDI_GLOBAL_DIR_FOR_JS_ISRTL_CODE_SNIPPET_PROVIDER);
    assertThat(
        codeSnippet.applyForJsSrc(dataRef, ImmutableList.<JsExpr>of())
            .getText()).isEqualTo("soy.$$bidiSpanWrap(IS_RTL?-1:1, opt_data.myKey)");
  }

  public void testApplyForPySrc() {
    BidiSpanWrapDirective codeSnippet = new BidiSpanWrapDirective(
        SharedRestrictedTestUtils.BIDI_GLOBAL_DIR_FOR_PY_ISRTL_CODE_SNIPPET_PROVIDER);

    PyExpr data = new PyStringExpr("'data'");
    assertThat(codeSnippet.applyForPySrc(data, ImmutableList.<PyExpr>of()).getText())
        .isEqualTo("bidi.span_wrap(-1 if IS_RTL else 1, 'data')");
  }
}