/** | |
* Copyright (c) 2008-2010, 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.issues.issue60; | |
import java.util.Arrays; | |
import junit.framework.TestCase; | |
import org.yaml.snakeyaml.Util; | |
import org.yaml.snakeyaml.Yaml; | |
import org.yaml.snakeyaml.introspector.Property; | |
import org.yaml.snakeyaml.nodes.CollectionNode; | |
import org.yaml.snakeyaml.nodes.MappingNode; | |
import org.yaml.snakeyaml.nodes.Node; | |
import org.yaml.snakeyaml.nodes.NodeTuple; | |
import org.yaml.snakeyaml.nodes.SequenceNode; | |
import org.yaml.snakeyaml.nodes.Tag; | |
import org.yaml.snakeyaml.representer.Representer; | |
public class SkipBeanTest extends TestCase { | |
public void testSkipNull() { | |
Yaml yaml = new Yaml(new SkipNullRepresenter()); | |
String output = yaml.dump(getBean()); | |
// System.out.println(output); | |
assertEquals(Util.getLocalResource("issues/issue60-1.yaml"), output); | |
} | |
private class SkipNullRepresenter extends Representer { | |
@Override | |
protected NodeTuple representJavaBeanProperty(Object javaBean, Property property, | |
Object propertyValue, Tag customTag) { | |
if (propertyValue == null) { | |
return null; | |
} else { | |
return super | |
.representJavaBeanProperty(javaBean, property, propertyValue, customTag); | |
} | |
} | |
} | |
public void testSkipEmptyCollections() { | |
Yaml yaml = new Yaml(new SkipEmptyRepresenter()); | |
String output = yaml.dump(getBean()); | |
// System.out.println(output); | |
assertEquals(Util.getLocalResource("issues/issue60-2.yaml"), output); | |
} | |
private class SkipEmptyRepresenter extends Representer { | |
@Override | |
protected NodeTuple representJavaBeanProperty(Object javaBean, Property property, | |
Object propertyValue, Tag customTag) { | |
NodeTuple tuple = super.representJavaBeanProperty(javaBean, property, propertyValue, | |
customTag); | |
Node valueNode = tuple.getValueNode(); | |
if (Tag.NULL.equals(valueNode.getTag())) { | |
return null;// skip 'null' values | |
} | |
if (valueNode instanceof CollectionNode) { | |
if (Tag.SEQ.equals(valueNode.getTag())) { | |
SequenceNode seq = (SequenceNode) valueNode; | |
if (seq.getValue().isEmpty()) { | |
return null;// skip empty lists | |
} | |
} | |
if (Tag.MAP.equals(valueNode.getTag())) { | |
MappingNode seq = (MappingNode) valueNode; | |
if (seq.getValue().isEmpty()) { | |
return null;// skip empty maps | |
} | |
} | |
} | |
return tuple; | |
} | |
} | |
private SkipBean getBean() { | |
SkipBean bean = new SkipBean(); | |
bean.setText("foo"); | |
bean.setListDate(null); | |
bean.setListInt(Arrays.asList(new Integer[] { null, 1, 2, 3 })); | |
bean.setListStr(Arrays.asList(new String[] { "bar", null, "foo", null })); | |
return bean; | |
} | |
} |