Examples with ClassTypeResolver org.kie.soup.project.datamodel.commons.types.ClassTypeResolver used on opensource projects

Example 1 with ClassTypeResolver

use of org.kie.soup.project.datamodel.commons.types.ClassTypeResolver in project drools by kiegroup.

the class ScenarioRunnerTest method testRuleFlowGroupActivation.

@Test
public void testRuleFlowGroupActivation() throws Exception {
    Scenario scenario = new Scenario();
    scenario.getImports().addImport(new Import("foo.bar.Coolness"));
    Fixture[] given = new Fixture[] { new FactData("Coolness", "c", Arrays.<Field>asList(new FieldData("num", "42"), new FieldData("name", "mic")), false) };
    scenario.getFixtures().addAll(Arrays.asList(given));
    ExecutionTrace executionTrace = new ExecutionTrace();
    scenario.getRules().add("rule1");
    scenario.setInclusive(true);
    scenario.getFixtures().add(executionTrace);
    Expectation[] assertions = new Expectation[2];
    assertions[0] = new VerifyFact("c", ls(new VerifyField("num", "42", "==")));
    assertions[1] = new VerifyRuleFired("rule1", 1, null);
    scenario.getFixtures().addAll(Arrays.asList(assertions));
    KieSession ksession = getKieSession("rule_flow_actication.drl");
    ClassLoader classLoader = ((KnowledgeBaseImpl) ksession.getKieBase()).getRootClassLoader();
    HashSet<String> imports = new HashSet<String>();
    imports.add("foo.bar.*");
    TypeResolver resolver = new ClassTypeResolver(imports, classLoader);
    Class<?> coolnessClass = classLoader.loadClass("foo.bar.Coolness");
    assertNotNull(coolnessClass);
    ClassLoader cl_ = Thread.currentThread().getContextClassLoader();
    Thread.currentThread().setContextClassLoader(classLoader);
    // resolver will need to have generated beans in it - possibly using a composite classloader from the package,
    // including whatever CL has the generated beans...
    ScenarioRunner scenarioRunner = new ScenarioRunner(ksession);
    scenarioRunner.run(scenario);
    assertEquals(0, executionTrace.getNumberOfRulesFired().intValue());
    assertFalse(scenario.wasSuccessful());
    // Activate rule flow
    scenario.getFixtures().clear();
    given = new Fixture[] { new FactData("Coolness", "c", Arrays.<Field>asList(new FieldData("num", "42"), new FieldData("name", "mic")), false), new ActivateRuleFlowGroup("asdf") };
    scenario.getFixtures().addAll(Arrays.asList(given));
    scenario.getFixtures().add(executionTrace);
    ((InternalAgendaGroup) ksession.getAgenda().getRuleFlowGroup("asdf")).setAutoDeactivate(false);
    scenarioRunner = new ScenarioRunner(ksession);
    scenarioRunner.run(scenario);
    assertTrue(scenario.wasSuccessful());
    Thread.currentThread().setContextClassLoader(cl_);
}

Example 2 with ClassTypeResolver

use of org.kie.soup.project.datamodel.commons.types.ClassTypeResolver in project drools by kiegroup.

the class FactAssignmentFieldPopulatorTest method testSimpleFields.

@Test
public void testSimpleFields() throws Exception {
    ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
    TypeResolver typeResolver = new ClassTypeResolver(new HashSet<String>(), classLoader);
    typeResolver.addImport("org.drools.workbench.models.testscenarios.backend.Cheese");
    Mouse mouse = new Mouse();
    FactAssignmentField factAssignmentField = new FactAssignmentField("cheese", "Cheese");
    factAssignmentField.getFact().getFieldData().add(new FieldData("type", "Best cheddar EVER! (tm)"));
    FactAssignmentFieldPopulator factAssignmentFieldPopulator = new FactAssignmentFieldPopulator(mouse, factAssignmentField, typeResolver);
    factAssignmentFieldPopulator.populate(new HashMap<String, Object>());
    assertEquals("Best cheddar EVER! (tm)", mouse.getCheese().getType());
}

Example 3 with ClassTypeResolver

use of org.kie.soup.project.datamodel.commons.types.ClassTypeResolver in project drools by kiegroup.

the class FactAssignmentFieldPopulatorTest method testFactAssignmentField.

@Test
public void testFactAssignmentField() throws Exception {
    ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
    TypeResolver typeResolver = new ClassTypeResolver(new HashSet<String>(), classLoader);
    typeResolver.addImport("org.drools.workbench.models.testscenarios.backend.Cheese");
    Mouse mouse = new Mouse();
    FactAssignmentField factAssignmentField = new FactAssignmentField("cheese", "Cheese");
    FactAssignmentFieldPopulator factAssignmentFieldPopulator = new FactAssignmentFieldPopulator(mouse, factAssignmentField, typeResolver);
    factAssignmentFieldPopulator.populate(new HashMap<String, Object>());
    assertNotNull(mouse.getCheese());
}

Example 4 with ClassTypeResolver

use of org.kie.soup.project.datamodel.commons.types.ClassTypeResolver in project drools by kiegroup.

the class FactPopulatorTest method getTypeResolver.

private TypeResolver getTypeResolver() {
    TypeResolver resolver = new ClassTypeResolver(new HashSet<String>(), getClassLoader());
    resolver.addImport("org.drools.workbench.models.testscenarios.backend.MyCollectionWrapper");
    resolver.addImport("org.drools.workbench.models.testscenarios.backend.Cheesery");
    resolver.addImport("org.drools.workbench.models.testscenarios.backend.Cheese");
    resolver.addImport("org.drools.workbench.models.testscenarios.backend.SqlDateWrapper");
    resolver.addImport("org.drools.workbench.models.testscenarios.backend.CheeseType");
    resolver.addImport("org.drools.workbench.models.testscenarios.backend.Person");
    resolver.addImport("org.drools.workbench.models.testscenarios.backend.OuterFact");
    return resolver;
}

Example 5 with ClassTypeResolver

use of org.kie.soup.project.datamodel.commons.types.ClassTypeResolver in project drools by kiegroup.

the class GlobalFactPopulatorTest method testWithGlobals.

@Test
public void testWithGlobals() throws Exception {
    FactData global = new FactData("Cheese", "c", Arrays.<Field>asList(new FieldData("type", "cheddar")), false);
    TypeResolver resolver = new ClassTypeResolver(new HashSet<String>(), Thread.currentThread().getContextClassLoader());
    resolver.addImport("org.drools.workbench.models.testscenarios.backend.Cheese");
    KieSession ksession = mock(KieSession.class);
    Map<String, Object> populatedData = new HashMap<String, Object>();
    Map<String, Object> globalData = new HashMap<String, Object>();
    GlobalFactPopulator globalFactPopulator = new GlobalFactPopulator(populatedData, resolver, global, globalData);
    globalFactPopulator.populate(ksession, new HashMap<String, FactHandle>());
    verify(ksession).setGlobal(eq(global.getName()), any(Object.class));
    assertEquals(1, globalData.size());
    assertEquals(0, populatedData.size());
}