blob: 41c6bf23252686b59e7199cd60afc797aea611ca (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
/**
* Copyright (c) 2008, 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 java.beans.IntrospectionException;
import java.util.Set;
import java.util.TreeSet;
import junit.framework.TestCase;
import org.yaml.snakeyaml.DumperOptions;
import org.yaml.snakeyaml.Yaml;
import org.yaml.snakeyaml.introspector.Property;
public class FilterPropertyToDumpTest extends TestCase {
public void testFilterPropertyInJavaBeanDumper() {
BeanToRemoveProperty bean = new BeanToRemoveProperty();
bean.setNumber(24);
bean.setId("ID124");
Yaml d = new Yaml();
String dump = d.dumpAsMap(bean);
// System.out.println(dump);
assertEquals("id: ID124\nnumber: 24\n", dump);
}
public void testFilterPropertyInYaml() {
BeanToRemoveProperty bean = new BeanToRemoveProperty();
bean.setNumber(25);
bean.setId("ID125");
Yaml yaml = new Yaml(new MyRepresenter());
String dump = yaml.dumpAsMap(bean);
// System.out.println(dump);
assertEquals("number: 25\n", dump);
}
public void testDoNotFilterPropertyIncludeReadOnly() {
BeanToRemoveProperty bean = new BeanToRemoveProperty();
bean.setNumber(26);
bean.setId("ID126");
DumperOptions options = new DumperOptions();
options.setAllowReadOnlyProperties(true);
Yaml yaml = new Yaml(options);
String dump = yaml.dump(bean);
// System.out.println(dump);
assertEquals(
"!!org.yaml.snakeyaml.representer.FilterPropertyToDumpTest$BeanToRemoveProperty {id: ID126,\n number: 26, something: true}\n",
dump);
}
public class BeanToRemoveProperty {
private int number;
private String id;
public boolean isSomething() {
return true;
}
public int getNumber() {
return number;
}
public void setNumber(int number) {
this.number = number;
}
public void setId(String id) {
this.id = id;
}
public String getId() {
return id;
}
}
private class MyRepresenter extends Representer {
@Override
protected Set<Property> getProperties(Class<? extends Object> type)
throws IntrospectionException {
Set<Property> set = super.getProperties(type);
Set<Property> filtered = new TreeSet<Property>();
if (type.equals(BeanToRemoveProperty.class)) {
// filter properties
for (Property prop : set) {
String name = prop.getName();
if (!name.equals("id")) {
filtered.add(prop);
}
}
}
return filtered;
}
}
}
|