/*
 * Copyright 2013 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.data.internal;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Iterables;
import com.google.common.collect.Maps;
import com.google.template.soy.data.SoyDict;
import com.google.template.soy.data.SoyValue;
import com.google.template.soy.data.SoyValueProvider;
import com.google.template.soy.data.restricted.BooleanData;
import com.google.template.soy.data.restricted.FloatData;
import com.google.template.soy.data.restricted.IntegerData;
import com.google.template.soy.data.restricted.StringData;

import junit.framework.TestCase;

import java.util.Map;

/**
 * Unit tests for DictImpl.
 *
 */
public class DictImplTest extends TestCase {


  public void testSoyValueMethods() {

    SoyValue val1 = DictImpl.forProviderMap(ImmutableMap.<String, SoyValue>of());
    assertTrue(val1.coerceToBoolean());  // DictImpl is always truthy.
    assertEquals("{}", val1.coerceToString());
    SoyValue val2 = DictImpl.forProviderMap(ImmutableMap.<String, SoyValue>of());
    assertFalse(val1.equals(val2));  // DictImpl uses object identity.

    SoyValue val3 = DictImpl.forProviderMap(ImmutableMap.<String, SoyValue>of(
        "foo", FloatData.forValue(3.14), "too", BooleanData.TRUE));
    assertTrue(val3.coerceToBoolean());
    assertEquals("{foo: 3.14, too: true}", val3.coerceToString());
  }


  public void testDictMethods() {

    SoyDict dict = DictImpl.forProviderMap(ImmutableMap.<String, SoyValue>of(
        "boo", StringData.forValue("aaah"), "foo", FloatData.forValue(3.14)));
    Map<String, ? extends SoyValueProvider> m1 = dict.asJavaStringMap();
    assertEquals(2, m1.size());
    assertEquals("aaah", m1.get("boo").resolve().stringValue());
    Map<String, ? extends SoyValue> m2 = dict.asResolvedJavaStringMap();
    assertEquals(2, m2.size());
    assertEquals(3.14, m2.get("foo").floatValue());
  }


  public void testRecordMethods() {

    Map<String, SoyValueProvider> providerMap = Maps.newHashMap();
    SoyDict dict = DictImpl.forProviderMap(providerMap);
    assertFalse(dict.hasField("boo"));
    assertNull(dict.getField("boo"));
    assertNull(dict.getFieldProvider("boo"));
    providerMap.put("boo", StringData.forValue("blah"));
    assertTrue(dict.hasField("boo"));
    assertEquals("blah", dict.getField("boo").stringValue());
    assertEquals("blah", dict.getFieldProvider("boo").resolve().stringValue());
    providerMap.remove("boo");
    assertFalse(dict.hasField("boo"));
    assertNull(dict.getField("boo"));
    assertNull(dict.getFieldProvider("boo"));

    providerMap.put("foo", FloatData.forValue(3.14));
    providerMap.put("too", BooleanData.TRUE);
    assertTrue(dict.hasField("foo"));
    assertEquals(3.14, dict.getField("foo").floatValue());
    assertEquals(true, dict.getField("too").booleanValue());
  }


  public void testMapMethods() {

    StringData BOO = StringData.forValue("boo");

    Map<String, SoyValueProvider> providerMap = Maps.newHashMap();
    SoyDict dict = DictImpl.forProviderMap(providerMap);
    assertEquals(0, dict.getItemCnt());
    assertEquals(0, Iterables.size(dict.getItemKeys()));
    assertFalse(dict.hasItem(BOO));
    assertNull(dict.getItem(BOO));
    assertNull(dict.getItemProvider(BOO));
    providerMap.put("boo", IntegerData.forValue(111));
    assertEquals(1, dict.getItemCnt());
    assertEquals(1, Iterables.size(dict.getItemKeys()));
    assertEquals("boo", Iterables.getOnlyElement(dict.getItemKeys()).stringValue());
    providerMap.put("foo", IntegerData.forValue(222));
    providerMap.put("goo", IntegerData.forValue(333));
    assertEquals(3, dict.getItemCnt());
    assertEquals(3, Iterables.size(dict.getItemKeys()));
    assertTrue(dict.hasItem(BOO));
    assertEquals(111, dict.getItem(BOO).integerValue());
    assertEquals(111, dict.getItemProvider(BOO).resolve().integerValue());
    providerMap.remove("foo");
    assertEquals(2, dict.getItemCnt());
    providerMap.remove("boo");
    providerMap.remove("goo");
    assertEquals(0, dict.getItemCnt());
    assertEquals(0, Iterables.size(dict.getItemKeys()));
    assertFalse(dict.hasItem(BOO));
    assertNull(dict.getItem(BOO));
    assertNull(dict.getItemProvider(BOO));
  }
}