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 org.yaml.snakeyaml.constructor;
18  
19  import java.io.IOException;
20  import java.util.ArrayList;
21  import java.util.List;
22  
23  import junit.framework.TestCase;
24  
25  import org.yaml.snakeyaml.TypeDescription;
26  import org.yaml.snakeyaml.Util;
27  import org.yaml.snakeyaml.Yaml;
28  import org.yaml.snakeyaml.error.YAMLException;
29  import org.yaml.snakeyaml.nodes.Tag;
30  import org.yaml.snakeyaml.representer.Representer;
31  
32  public class ClassTagsTest extends TestCase {
33  
34      public void testDefaultRepresenter() throws IOException {
35          Car car = new Car();
36          car.setPlate("12-XP-F4");
37          List<Wheel> wheels = new ArrayList<Wheel>();
38          for (int i = 1; i < 6; i++) {
39              Wheel wheel = new Wheel();
40              wheel.setId(i);
41              wheels.add(wheel);
42          }
43          car.setWheels(wheels);
44          assertEquals(Util.getLocalResource("constructor/car-with-tags.yaml"), new Yaml().dump(car));
45      }
46  
47      public void testDumpClassTag() throws IOException {
48          Car car = new Car();
49          car.setPlate("12-XP-F4");
50          List<Wheel> wheels = new ArrayList<Wheel>();
51          for (int i = 1; i < 6; i++) {
52              Wheel wheel = new Wheel();
53              wheel.setId(i);
54              wheels.add(wheel);
55          }
56          car.setWheels(wheels);
57          Representer representer = new Representer();
58          representer.addClassTag(Car.class, new Tag("!car"));
59          representer.addClassTag(Wheel.class, Tag.MAP);
60          Yaml yaml = new Yaml(representer);
61          String output = yaml.dump(car);
62          assertEquals(Util.getLocalResource("constructor/car-without-tags.yaml"), output);
63      }
64  
65      public void testLoadUnknounClassTag() throws IOException {
66          try {
67              Yaml yaml = new Yaml();
68              yaml.load(Util.getLocalResource("constructor/car-without-tags.yaml"));
69              fail("Must fail because of unknown tag: !car");
70          } catch (YAMLException e) {
71              assertTrue(e.getMessage().contains("Invalid tag: !car"));
72          }
73  
74      }
75  
76      public void testLoadClassTag() throws IOException {
77          Constructor constructor = new Constructor();
78          constructor.addTypeDescription(new TypeDescription(Car.class, "!car"));
79          Yaml yaml = new Yaml(constructor);
80          String source = Util.getLocalResource("constructor/car-without-tags.yaml");
81          Car car = (Car) yaml.load(source);
82          assertEquals("12-XP-F4", car.getPlate());
83          List<Wheel> wheels = car.getWheels();
84          assertNotNull(wheels);
85          assertEquals(5, wheels.size());
86      }
87  
88      public void testNullDescription() throws IOException {
89          Constructor constructor = new Constructor();
90          try {
91              constructor.addTypeDescription(null);
92              fail("Description is required.");
93          } catch (Exception e) {
94              assertEquals("TypeDescription is required.", e.getMessage());
95          }
96      }
97  
98      public void testLoadClassNoRoot() throws IOException {
99          Constructor constructor = new Constructor(new TypeDescription(Car.class));
100         Yaml yaml = new Yaml(constructor);
101         Car car = (Car) yaml.load(Util.getLocalResource("constructor/car-no-root-class.yaml"));
102         assertEquals("12-XP-F4", car.getPlate());
103         List<Wheel> wheels = car.getWheels();
104         assertNotNull(wheels);
105         assertEquals(5, wheels.size());
106     }
107 }