/**
 * 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.constructor;

import java.util.LinkedHashMap;
import java.util.Map;
import java.util.TreeMap;
import junit.framework.TestCase;
import org.yaml.snakeyaml.LoaderOptions;
import org.yaml.snakeyaml.composer.Composer;
import org.yaml.snakeyaml.parser.Parser;
import org.yaml.snakeyaml.parser.ParserImpl;
import org.yaml.snakeyaml.reader.StreamReader;
import org.yaml.snakeyaml.resolver.Resolver;

public class ConstructorMappingTest extends TestCase {

  @SuppressWarnings("unchecked")
  public void testGetDefaultMap() {
    String data = "{ one: 1, two: 2, three: 3 }";
    Map<Object, Object> map = (Map<Object, Object>) construct(new CustomConstructor(), data);
    assertNotNull(map);
    assertTrue(map.getClass().toString(), map instanceof TreeMap);
  }

  @SuppressWarnings("unchecked")
  public void testGetArrayList() {
    String data = "{ one: 1, two: 2, three: 3 }";
    Map<Object, Object> map = (Map<Object, Object>) construct(data);
    assertNotNull(map);
    assertTrue(map.getClass().toString(), map instanceof LinkedHashMap);
  }

  private Object construct(String data) {
    return construct(new Constructor(new LoaderOptions()), data);
  }

  private Object construct(Constructor constructor, String data) {
    StreamReader reader = new StreamReader(data);
    LoaderOptions options = new LoaderOptions();
    Parser parser = new ParserImpl(reader, options);
    Resolver resolver = new Resolver();
    Composer composer = new Composer(parser, resolver, options);
    constructor.setComposer(composer);
    return constructor.getSingleData(Object.class);
  }

  class CustomConstructor extends Constructor {

    public CustomConstructor() {
      super(new LoaderOptions());
    }

    @Override
    protected Map<Object, Object> createDefaultMap(int initSize) {
      return new TreeMap<Object, Object>();
    }
  }
}
