/**
 * Copyright (c) 2008, SnakeYAML
 *
 * 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 org.yaml.snakeyaml.issues.issue61;

import java.util.LinkedHashMap;
import java.util.Map;
import junit.framework.TestCase;
import org.yaml.snakeyaml.Util;
import org.yaml.snakeyaml.Yaml;
import org.yaml.snakeyaml.YamlCreator;

public class GenericMapBeanTest extends TestCase {

  @SuppressWarnings("unchecked")
  public void testGenericMap() {
    Yaml yaml = new Yaml();
    MapProvider<String, Integer> listProvider = new MapProvider<String, Integer>();
    listProvider.getMap().put("foo", 17);
    listProvider.getMap().put("bar", 19);
    String s = yaml.dumpAsMap(listProvider);
    // System.out.println(s);
    assertEquals("map:\n  foo: 17\n  bar: 19\n", s);
    // parse
    Yaml loader = new Yaml();
    MapProvider<String, Integer> listProvider2 = loader.loadAs(s, MapProvider.class);
    assertEquals(Integer.valueOf(17), listProvider2.getMap().get("foo"));
    assertEquals(Integer.valueOf(19), listProvider2.getMap().get("bar"));
    assertEquals(listProvider, listProvider2);
  }

  @SuppressWarnings("rawtypes")
  public void testGenericBean() {
    Yaml yaml = new Yaml();
    MapProvider<String, Bean> listProvider = new MapProvider<String, Bean>();
    Bean foo = new Bean();
    foo.setName("foo");
    listProvider.getMap().put("foo", foo);
    Bean bar = new Bean();
    bar.setName("bar");
    bar.setNumber(3);
    listProvider.getMap().put("bar", bar);
    String s = yaml.dumpAsMap(listProvider);
    // System.out.println(s);
    String etalon = Util.getLocalResource("issues/issue61-2.yaml");
    assertEquals(etalon, s);
    // parse
    Yaml loader = YamlCreator.allowClassPrefix("org.yaml.snakeyaml");
    MapProvider listProvider2 = loader.loadAs(s, MapProvider.class);
    Bean foo2 = (Bean) listProvider2.getMap().get("foo");
    assertEquals("foo", foo2.getName());
    assertEquals(0, foo2.getNumber());
    Bean bar2 = (Bean) listProvider2.getMap().get("bar");
    assertEquals("bar", bar2.getName());
    assertEquals(3, bar2.getNumber());
  }

  public static class MapProvider<K, V> {

    private Map<K, V> map = new LinkedHashMap<K, V>();

    public Map<K, V> getMap() {
      return map;
    }

    public void setMap(Map<K, V> map) {
      this.map = map;
    }

    @SuppressWarnings("rawtypes")
    @Override
    public boolean equals(Object obj) {
      if (obj instanceof MapProvider) {
        return map.equals(((MapProvider) obj).getMap());
      } else {
        return false;
      }
    }

    @Override
    public int hashCode() {
      return map.hashCode();
    }
  }

  public static class Bean {

    private String name;
    private int number;

    public String getName() {
      return name;
    }

    public void setName(String name) {
      this.name = name;
    }

    public int getNumber() {
      return number;
    }

    public void setNumber(int number) {
      this.number = number;
    }
  }
}
