01: package org.drools.eclipse.wizard.rule;
02:
03: import java.io.BufferedReader;
04: import java.io.IOException;
05: import java.io.InputStream;
06: import java.io.InputStreamReader;
07:
08: import junit.framework.TestCase;
09:
10: public class DRLGenTemplateTest extends TestCase {
11:
12: public void testNewRule() throws Exception {
13: DRLGenerator gen = new DRLGenerator();
14: InputStream result = gen.generateRule("myPackage", getClass()
15: .getResourceAsStream("new_rule.drl.template"));
16:
17: assertNotNull(result);
18: StringBuffer buf = getResult(result);
19: assertTrue(buf.toString().indexOf("package myPackage") > -1);
20: assertFalse(buf.toString().indexOf("$date$") > -1);
21:
22: }
23:
24: public void testNewPackage() throws Exception {
25: DRLGenerator gen = new DRLGenerator();
26: InputStream result = gen.generatePackage("myPackage", true,
27: true, getClass().getResourceAsStream(
28: "new_rule.drl.template"));
29:
30: assertNotNull(result);
31: StringBuffer buf = getResult(result);
32: assertTrue(buf.toString().indexOf("package myPackage") > -1);
33: assertFalse(buf.toString().indexOf("$date$") > -1);
34: assertFalse(buf.toString().indexOf("$expander$") > -1);
35: assertFalse(buf.toString().indexOf("$functions$") > -1);
36:
37: }
38:
39: private StringBuffer getResult(InputStream result)
40: throws IOException {
41: BufferedReader reader = new BufferedReader(
42: new InputStreamReader(result));
43: String line = null;
44: StringBuffer buf = new StringBuffer();
45: while ((line = reader.readLine()) != null) {
46: buf.append(line + "\n");
47: }
48: return buf;
49: }
50:
51: }
|