/** | |
* Copyright (c) 2008-2010, http://code.google.com/p/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 examples; | |
import java.io.IOException; | |
import java.util.List; | |
import junit.framework.TestCase; | |
import org.yaml.snakeyaml.SnakeYaml; | |
import org.yaml.snakeyaml.constructor.SafeConstructor; | |
public class SafeConstructorExampleTest extends TestCase { | |
@SuppressWarnings("unchecked") | |
public void testConstruct() throws IOException { | |
String doc = "- 5\n- Person\n- true"; | |
SnakeYaml yaml = new SnakeYaml(new SafeConstructor()); | |
List<Object> list = (List<Object>) yaml.load(doc); | |
assertEquals(3, list.size()); | |
assertEquals(new Integer(5), list.get(0)); | |
assertEquals("Person", list.get(1)); | |
assertEquals(Boolean.TRUE, list.get(2)); | |
} | |
public void testSafeConstruct() throws IOException { | |
String doc = "- 5\n- !org.yaml.snakeyaml.constructor.Person\n firstName: Andrey\n age: 99\n- true"; | |
SnakeYaml yaml = new SnakeYaml(new SafeConstructor()); | |
try { | |
yaml.load(doc); | |
fail("Custom Java classes should not be created."); | |
} catch (Exception e) { | |
assertEquals( | |
"null; could not determine a constructor for the tag !org.yaml.snakeyaml.constructor.Person", | |
e.getMessage()); | |
} | |
} | |
} |