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
|
/**
* Copyright (c) 2008, 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 org.yaml.snakeyaml.emitter.template;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List;
import junit.framework.TestCase;
import org.apache.velocity.Template;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.VelocityEngine;
import org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader;
import org.yaml.snakeyaml.DumperOptions;
import org.yaml.snakeyaml.Util;
import org.yaml.snakeyaml.Yaml;
import org.yaml.snakeyaml.immutable.Point;
public class VelocityTest extends TestCase {
public void testNoTemplate() {
DumperOptions options = new DumperOptions();
options.setAllowReadOnlyProperties(true);
Yaml yaml = new Yaml(options);
String output = yaml.dumpAsMap(createBean());
// System.out.println(output);
assertEquals(Util.getLocalResource("template/etalon1.yaml"), output);
}
public void testTemplate1() throws Exception {
VelocityContext context = new VelocityContext();
MyBean bean = createBean();
context.put("bean", bean);
Yaml yaml = new Yaml();
context.put("list", yaml.dump(bean.getList()));
VelocityEngine ve = new VelocityEngine();
ve.setProperty("file.resource.loader.class", ClasspathResourceLoader.class.getName());
ve.init();
Template t = ve.getTemplate("template/mybean1.vm");
StringWriter writer = new StringWriter();
t.merge(context, writer);
String output = writer.toString().trim().replaceAll("\\r\\n", "\n");
// System.out.println(output);
String etalon = Util.getLocalResource("template/etalon2-template.yaml").trim();
assertEquals(etalon.length(), output.length());
assertEquals(etalon, output);
// parse the YAML document
Yaml loader = new Yaml();
MyBean parsedBean = loader.loadAs(output, MyBean.class);
assertEquals(bean, parsedBean);
}
private MyBean createBean() {
MyBean bean = new MyBean();
bean.setId("id123");
List<String> list = new ArrayList<String>();
list.add("aaa");
list.add("bbb");
list.add("ccc");
bean.setList(list);
Point p = new Point(1.0, 2.0);
bean.setPoint(p);
return bean;
}
}
|
