| /** |
| * Copyright (c) 2008-2012, 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.representer; |
| |
| import junit.framework.TestCase; |
| |
| import org.yaml.snakeyaml.Yaml; |
| import org.yaml.snakeyaml.constructor.AbstractConstruct; |
| import org.yaml.snakeyaml.constructor.Constructor; |
| import org.yaml.snakeyaml.nodes.Node; |
| import org.yaml.snakeyaml.nodes.ScalarNode; |
| import org.yaml.snakeyaml.nodes.Tag; |
| |
| public class RepresentTest extends TestCase { |
| |
| public void testCustomRepresenter() { |
| Yaml yaml = new Yaml(new MyConstructor(), new MyRepresenter()); |
| CustomBean etalon = new CustomBean("A", 1); |
| String output = yaml.dump(etalon); |
| assertEquals("!!Dice 'Ad1'\n", output); |
| CustomBean bean = (CustomBean) yaml.load(output); |
| assertEquals("A", bean.getPrefix()); |
| assertEquals(1, bean.getSuffix()); |
| assertEquals(etalon, bean); |
| } |
| |
| class CustomBean { |
| private String prefix; |
| private int suffix; |
| |
| public CustomBean(String prefix, int suffix) { |
| this.prefix = prefix; |
| this.suffix = suffix; |
| } |
| |
| public String getPrefix() { |
| return prefix; |
| } |
| |
| public int getSuffix() { |
| return suffix; |
| } |
| |
| @Override |
| public boolean equals(Object obj) { |
| CustomBean bean = (CustomBean) obj; |
| return prefix.equals(bean.getPrefix()) && suffix == bean.getSuffix(); |
| } |
| } |
| |
| class MyRepresenter extends Representer { |
| public MyRepresenter() { |
| this.representers.put(CustomBean.class, new RepresentDice()); |
| } |
| |
| private class RepresentDice implements Represent { |
| public Node representData(Object data) { |
| CustomBean coin = (CustomBean) data; |
| String value = coin.getPrefix() + "d" + coin.getSuffix(); |
| return representScalar(new Tag("!!Dice"), value); |
| } |
| } |
| } |
| |
| class MyConstructor extends Constructor { |
| public MyConstructor() { |
| this.yamlConstructors.put(new Tag(Tag.PREFIX + "Dice"), new ConstructDice()); |
| } |
| |
| private class ConstructDice extends AbstractConstruct { |
| public Object construct(Node node) { |
| String val = (String) constructScalar((ScalarNode) node); |
| return new CustomBean(val.substring(0, 1), new Integer(val.substring(2))); |
| } |
| } |
| } |
| } |