View Javadoc

    1   /**
    2    * Copyright (c) 2008-2011, http://www.snakeyaml.org
    3    *
    4    * Licensed under the Apache License, Version 2.0 (the "License");
    5    * you may not use this file except in compliance with the License.
    6    * You may obtain a copy of the License at
    7    *
    8    *     http://www.apache.org/licenses/LICENSE-2.0
    9    *
    10   * Unless required by applicable law or agreed to in writing, software
    11   * distributed under the License is distributed on an "AS IS" BASIS,
    12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13   * See the License for the specific language governing permissions and
    14   * limitations under the License.
    15   */
    16
    17  package examples;
    18
    19  import java.io.IOException;
    20  import java.util.Map;
    21  import java.util.TreeMap;
    22
    23  import junit.framework.TestCase;
    24
    25  import org.yaml.snakeyaml.Yaml;
    26  import org.yaml.snakeyaml.constructor.Constructor;
    27
    28  public class CustomMapExampleTest extends TestCase {
    29      public void testMap() throws IOException {
    30          Yaml yaml = new Yaml(new CustomConstructor());
    31          @SuppressWarnings("unchecked")
    32          Map<Integer, String> data = (Map<Integer, String>) yaml
    33                  .load("{2: '222', 1: '111', 3: '333'}");
    34          assertTrue(data instanceof TreeMap);
    35          Object[] keys = data.keySet().toArray();
    36          // must be sorted
    37          assertEquals(new Integer(1), keys[0]);
    38          assertEquals(new Integer(2), keys[1]);
    39          assertEquals(new Integer(3), keys[2]);
    40      }
    41
    42      class CustomConstructor extends Constructor {
    43          @Override
    44          protected Map<Object, Object> createDefaultMap() {
    45              return new TreeMap<Object, Object>();
    46          }
    47      }
    48  }