/**
* Copyright (c) 2008-2011, http://www.snakeyaml.org
*
* 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;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.EnumMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import junit.framework.TestCase;
import org.yaml.snakeyaml.constructor.Constructor;
public class EnumTest extends TestCase {
// Dumping
public void testDumpEnum() {
Yaml yaml = new Yaml();
String output = yaml.dump(Suit.CLUBS);
assertEquals("!!org.yaml.snakeyaml.Suit 'CLUBS'\n", output);
}
public void testDumpEnumArray() {
DumperOptions options = new DumperOptions();
options.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
Yaml yaml = new Yaml(options);
String output = yaml.dump(Suit.values());
assertEquals(
"- !!org.yaml.snakeyaml.Suit 'CLUBS'\n- !!org.yaml.snakeyaml.Suit 'DIAMONDS'\n- !!org.yaml.snakeyaml.Suit 'HEARTS'\n- !!org.yaml.snakeyaml.Suit 'SPADES'\n",
output);
}
public void testDumpEnumList() {
DumperOptions options = new DumperOptions();
options.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
Yaml yaml = new Yaml(options);
List<Suit> list = Arrays.asList(Suit.values());
String output = yaml.dump(list);
assertEquals(
"- !!org.yaml.snakeyaml.Suit 'CLUBS'\n- !!org.yaml.snakeyaml.Suit 'DIAMONDS'\n- !!org.yaml.snakeyaml.Suit 'HEARTS'\n- !!org.yaml.snakeyaml.Suit 'SPADES'\n",
output);
}
public void testDumpEnumListNoAnchor() {
DumperOptions options = new DumperOptions();
options.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
Yaml yaml = new Yaml(options);
List<Suit> list = new ArrayList<Suit>(3);
list.add(Suit.CLUBS);
list.add(Suit.DIAMONDS);
list.add(Suit.CLUBS);
String output = yaml.dump(list);
assertEquals(
"- !!org.yaml.snakeyaml.Suit 'CLUBS'\n- !!org.yaml.snakeyaml.Suit 'DIAMONDS'\n- !!org.yaml.snakeyaml.Suit 'CLUBS'\n",
output);
}
public void testDumpEnumMap() {
DumperOptions options = new DumperOptions();
options.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
Yaml yaml = new Yaml(options);
Map<String, Suit> map = new LinkedHashMap<String, Suit>();
map.put("c", Suit.CLUBS);
map.put("d", Suit.DIAMONDS);
String output = yaml.dump(map);
assertEquals(
"c: !!org.yaml.snakeyaml.Suit 'CLUBS'\nd: !!org.yaml.snakeyaml.Suit 'DIAMONDS'\n",
output);
}
public void testDumpEnumMap2() {
DumperOptions options = new DumperOptions();
options.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
Yaml yaml = new Yaml(options);
Map<Suit, Integer> map = new EnumMap<Suit, Integer>(Suit.class);
map.put(Suit.CLUBS, 0);
map.put(Suit.DIAMONDS, 123);
String output = yaml.dump(map);
assertEquals(
"!!org.yaml.snakeyaml.Suit 'CLUBS': 0\n!!org.yaml.snakeyaml.Suit 'DIAMONDS': 123\n",
output);
}
public void testDumpEnumBean() {
DumperOptions options = new DumperOptions();
options.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
Yaml yaml = new Yaml(options);
EnumBean bean = new EnumBean();
bean.setId(17);
bean.setSuit(Suit.SPADES);
LinkedHashMap<Suit, Integer> map = new LinkedHashMap<Suit, Integer>();
map.put(Suit.CLUBS, 1);
map.put(Suit.DIAMONDS, 2);
bean.setMap(map);
String output = yaml.dump(bean);
assertEquals(
"!!org.yaml.snakeyaml.EnumBean\nid: 17\nmap:\n CLUBS: 1\n DIAMONDS: 2\nsuit: SPADES\n",
output);
}
// Loading
public void testLoadEnum() {
Yaml yaml = new Yaml();
Suit suit = (Suit) yaml.load("!!org.yaml.snakeyaml.Suit 'CLUBS'\n");
assertEquals(Suit.CLUBS, suit);
}
@SuppressWarnings("unchecked")
public void testLoadEnumList() {
Yaml yaml = new Yaml();
List<Suit> list = (List<Suit>) yaml
.load("- !!org.yaml.snakeyaml.Suit 'CLUBS'\n- !!org.yaml.snakeyaml.Suit 'DIAMONDS'\n- !!org.yaml.snakeyaml.Suit 'HEARTS'\n- !!org.yaml.snakeyaml.Suit 'SPADES'");
assertEquals(4, list.size());
assertEquals(Suit.CLUBS, list.get(0));
assertEquals(Suit.DIAMONDS, list.get(1));
assertEquals(Suit.HEARTS, list.get(2));
assertEquals(Suit.SPADES, list.get(3));
}
@SuppressWarnings("unchecked")
public void testLoadEnumMap() {
Yaml yaml = new Yaml();
Map<Integer, Suit> map = (Map<Integer, Suit>) yaml
.load("1: !!org.yaml.snakeyaml.Suit 'HEARTS'\n2: !!org.yaml.snakeyaml.Suit 'DIAMONDS'");
assertEquals(2, map.size());
assertEquals(Suit.HEARTS, map.get(1));
assertEquals(Suit.DIAMONDS, map.get(2));
}
public void testLoadEnumBean() {
Yaml yaml = new Yaml();
EnumBean bean = (EnumBean) yaml
.load("!!org.yaml.snakeyaml.EnumBean\nid: 174\nmap:\n !!org.yaml.snakeyaml.Suit 'CLUBS': 1\n !!org.yaml.snakeyaml.Suit 'DIAMONDS': 2\nsuit: CLUBS");
LinkedHashMap<Suit, Integer> map = new LinkedHashMap<Suit, Integer>();
map.put(Suit.CLUBS, 1);
map.put(Suit.DIAMONDS, 2);
assertEquals(Suit.CLUBS, bean.getSuit());
assertEquals(174, bean.getId());
assertEquals(map, bean.getMap());
}
public void testLoadEnumBean2() {
Constructor c = new Constructor();
TypeDescription td = new TypeDescription(EnumBean.class);
td.putMapPropertyType("map", Suit.class, Object.class);
c.addTypeDescription(td);
Yaml yaml = new Yaml(c);
EnumBean bean = (EnumBean) yaml
.load("!!org.yaml.snakeyaml.EnumBean\nid: 174\nmap:\n CLUBS: 1\n DIAMONDS: 2\nsuit: CLUBS");
LinkedHashMap<Suit, Integer> map = new LinkedHashMap<Suit, Integer>();
map.put(Suit.CLUBS, 1);
map.put(Suit.DIAMONDS, 2);
assertEquals(Suit.CLUBS, bean.getSuit());
assertEquals(174, bean.getId());
assertEquals(map, bean.getMap());
}
public void testLoadWrongEnum() {
Yaml yaml = new Yaml();
try {
yaml
.load("1: !!org.yaml.snakeyaml.Suit 'HEARTS'\n2: !!org.yaml.snakeyaml.Suit 'KOSYR'");
fail("KOSYR is not Suit");
} catch (Exception e) {
assertTrue("KOSYR must be reported", e.getMessage().contains(
"Unable to find enum value 'KOSYR' for enum"));
}
}
}