package org.drools.assistant;
import java.util.List;
import junit.framework.Assert;
import junit.framework.TestCase;
import org.drools.assistant.option.AssistantOption;
import org.drools.assistant.option.ReplaceAssistantOption;
import org.drools.assistant.processor.AbstractRuleAssistantProcessor;
import org.drools.assistant.processor.DRLRefactorProcessor;
public class FixImportTest extends TestCase {
private AbstractRuleAssistantProcessor ruleAssistant;
private String rule;
protected void setUp() throws Exception {
ruleAssistant = new DRLRefactorProcessor();
rule = "package com.sample\n\n" +
"import com.sample.DroolsTest.Message;\n" +
"rule \"Hello World\"\n" +
"\twhen\n" +
"\t\tm : Message( status == Message.HELLO, myMessage : message )\n" +
"\t\tPrueba()\n" +
"\tthen\n" +
"\t\tSystem.out.println( myMessage );\n" +
"\t\tm.setMessage( \"Goodbye cruel world\" );\n" +
"\t\tm.setStatus( Message.GOODBYE );\n" +
"\t\tupdate( m );\n" +
"end";
}
public void testFirstTest() {
List<AssistantOption> options = ruleAssistant.getRuleAssistant(rule, 150);
assertEquals(1, options.size());
ReplaceAssistantOption assistantOption = (ReplaceAssistantOption) options.get(0);
Assert.assertEquals("\t\t$prueba : Prueba()", assistantOption.getContent());
}
}