Skip to content

Commit

Permalink
[DROOLS-1379] Add tests for OOPath in queries
Browse files Browse the repository at this point in the history
  • Loading branch information
@jiripetrlik
jiripetrlik committed Mar 14, 2017
1 parent 7d7087a commit 85baf6c
Show file tree
Hide file tree
Showing 5 changed files with 479 additions and 0 deletions.
290 changes: 290 additions & 0 deletions drools-compiler/src/test/java/org/drools/compiler/oopath/OOPathQueriesTest.java
View file
@@ -0,0 +1,290 @@
/*
* Copyright 2017 Red Hat, Inc. and/or its affiliates.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
*
* 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.drools.compiler.oopath;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;

import org.drools.compiler.oopath.model.Group;
import org.drools.compiler.oopath.model.Man;
import org.drools.compiler.oopath.model.Person;
import org.drools.compiler.oopath.model.Room;
import org.drools.compiler.oopath.model.SensorEvent;
import org.drools.compiler.oopath.model.Thing;
import org.drools.compiler.oopath.model.Woman;
import org.junit.Test;
import org.kie.api.io.ResourceType;
import org.kie.api.runtime.KieSession;
import org.kie.api.runtime.rule.QueryResults;
import org.kie.api.runtime.rule.QueryResultsRow;
import org.kie.api.runtime.rule.Variable;
import org.kie.internal.utils.KieHelper;
import static org.assertj.core.api.Assertions.assertThat;

public class OOPathQueriesTest {

@Test
public void testQueryFromCode() {
final String drl =
"import org.drools.compiler.oopath.model.Thing;\n" +
"query isContainedIn( Thing $x, Thing $y )\n" +
" $y := /$x/children\n" +
"or\n" +
" ( $z := /$x/children and isContainedIn( $z, $y; ) )\n" +
"end\n";

final List<String> itemList = Arrays.asList(new String[] { "display", "keyboard", "processor" });

final KieSession ksession = new KieHelper().addContent( drl, ResourceType.DRL )
.build()
.newKieSession();

Thing smartphone = new Thing("smartphone");
for ( String item : itemList ) {
Thing component = new Thing(item);
smartphone.addChild(component);
}

ksession.insert(smartphone);

QueryResults queryResults = ksession.getQueryResults("isContainedIn", new Object[] { smartphone, Variable.v });
List<String> resultList = new ArrayList<>();
for(QueryResultsRow row : queryResults) {
Thing component = (Thing) row.get("$y");
resultList.add(component.getName());
}

assertThat(resultList).as("Querry does not contain all items").containsAll(itemList);

ksession.dispose();
}

@Test
public void testReactiveQuery() {
final String drl =
"import org.drools.compiler.oopath.model.Room;\n" +
"import org.drools.compiler.oopath.model.Sensor;\n" +
"import org.drools.compiler.oopath.model.SensorEvent;\n" +
"query temperature ( Room $r, double $t )\n" +
" $t := /$r/temperatureSensor/value\n" +
"end\n" +
"" +
"rule \"Change sensor value\" when\n" +
" $e : SensorEvent( $s : sensor, $v : value)\n" +
"then\n" +
" modify($s) { setValue($v) }\n" +
" retract($e)\n" +
"end\n" +
"" +
"rule \"Turn heating on\" when\n" +
" $r : Room()\n" +
" temperature( $r, $t; )\n" +
" eval( $t < 20 )" +
"then\n" +
" $r.getHeating().setOn(true);\n" +
"end\n" +
"rule \"Turn heating off\" when\n" +
" $r : Room()\n" +
" temperature( $r, $t; )\n" +
" eval( $t > 20 )" +
"then\n" +
" $r.getHeating().setOn(false);\n" +
"end\n";

Room room = new Room("Room");

final KieSession ksession = new KieHelper().addContent( drl, ResourceType.DRL )
.build()
.newKieSession();

room.getTemperatureSensor().setValue(15);
room.getHeating().setOn(false);
ksession.insert(room);
ksession.insert(room.getTemperatureSensor());
ksession.insert(room.getHeating());
ksession.fireAllRules();
assertThat(room.getHeating().isOn()).as("Temperature is bellow 20 degrees of Celsius. Heating should be turned on.").isTrue();

ksession.insert(new SensorEvent(room.getTemperatureSensor(), 25));
ksession.fireAllRules();
assertThat(room.getHeating().isOn()).as("Temperature is higher than 20 degrees of Celsius. Heating should be turned off.").isFalse();

ksession.dispose();
}

@Test
public void testNonReactiveOOPathInQuery() {
final String drl =
"import org.drools.compiler.oopath.model.Room;\n" +
"import org.drools.compiler.oopath.model.Sensor;\n" +
"import org.drools.compiler.oopath.model.SensorEvent;\n" +
"query temperature ( Room $r, double $t )\n" +
" $t := /$r?/temperatureSensor/value\n" +
"end\n" +
"" +
"rule \"Change sensor value\" when\n" +
" $e : SensorEvent( $s : sensor, $v : value)\n" +
"then\n" +
" modify($s) { setValue($v) }\n" +
" retract($e)\n" +
"end\n" +
"" +
"rule \"Turn heating on\" when\n" +
" $r : Room()\n" +
" temperature( $r, $t; )\n" +
" eval( $t < 20 )" +
"then\n" +
" $r.getHeating().setOn(true);\n" +
"end\n" +
"rule \"Turn heating off\" when\n" +
" $r : Room()\n" +
" temperature( $r, $t; )\n" +
" eval( $t > 20 )" +
"then\n" +
" $r.getHeating().setOn(false);\n" +
"end\n";

Room room = new Room("Room");

final KieSession ksession = new KieHelper().addContent( drl, ResourceType.DRL )
.build()
.newKieSession();

room.getTemperatureSensor().setValue(15);
room.getHeating().setOn(false);
ksession.insert(room);
ksession.insert(room.getTemperatureSensor());
ksession.insert(room.getHeating());
ksession.fireAllRules();
assertThat(room.getHeating().isOn()).as("Temperature is bellow 20 degrees of Celsius. Heating should be turned on.").isTrue();

ksession.insert(new SensorEvent(room.getTemperatureSensor(), 25));
ksession.fireAllRules();
assertThat(room.getHeating().isOn()).as("Query is not reactive. Heating should still be turned on.").isTrue();

ksession.dispose();
}

@Test
public void testPullOnlyQueryWithOOPath() {
final String drl =
"import org.drools.compiler.oopath.model.Room;\n" +
"import org.drools.compiler.oopath.model.Sensor;\n" +
"import org.drools.compiler.oopath.model.SensorEvent;\n" +
"query temperature ( Room $r, double $t )\n" +
" $t := /$r/temperatureSensor/value\n" +
"end\n" +
"" +
"rule \"Change sensor value\" when\n" +
" $e : SensorEvent( $s : sensor, $v : value)\n" +
"then\n" +
" modify($s) { setValue($v) }\n" +
" retract($e)\n" +
"end\n" +
"" +
"rule \"Turn heating on\" when\n" +
" $r : Room()\n" +
" ?temperature( $r, $t; )\n" +
" eval( $t < 20 )" +
"then\n" +
" $r.getHeating().setOn(true);\n" +
"end\n" +
"rule \"Turn heating off\" when\n" +
" $r : Room()\n" +
" ?temperature( $r, $t; )\n" +
" eval( $t > 20 )" +
"then\n" +
" $r.getHeating().setOn(false);\n" +
"end\n";

Room room = new Room("Room");

final KieSession ksession = new KieHelper().addContent( drl, ResourceType.DRL )
.build()
.newKieSession();

room.getTemperatureSensor().setValue(15);
room.getHeating().setOn(false);
ksession.insert(room);
ksession.insert(room.getTemperatureSensor());
ksession.insert(room.getHeating());
ksession.fireAllRules();
assertThat(room.getHeating().isOn()).as("Temperature is bellow 20 degrees of Celsius. Heating should be turned on.").isTrue();

ksession.insert(new SensorEvent(room.getTemperatureSensor(), 25));
ksession.fireAllRules();
assertThat(room.getHeating().isOn()).as("Query is not reactive. Heating should still be turned on.").isTrue();

ksession.dispose();
}

@Test
public void testPullOnlyQueryWithoutOOPath() {
final String drl =
"import org.drools.compiler.oopath.model.Room;\n" +
"import org.drools.compiler.oopath.model.Sensor;\n" +
"import org.drools.compiler.oopath.model.SensorEvent;\n" +
"query temperature ( Room $r, double $t )\n" +
" $r := Room()\n" +
" $s : Sensor() from $r.getTemperatureSensor()\n" +
" $t := Double() from $s.getValue()\n" +
"end\n" +
"" +
"rule \"Change sensor value\" when\n" +
" $e : SensorEvent( $s : sensor, $v : value)\n" +
"then\n" +
" modify($s) { setValue($v) }\n" +
" retract($e)\n" +
"end\n" +
"" +
"rule \"Turn heating on\" when\n" +
" $r : Room()\n" +
" ?temperature( $r, $t; )\n" +
" eval( $t < 20 )" +
"then\n" +
" $r.getHeating().setOn(true);\n" +
"end\n" +
"rule \"Turn heating off\" when\n" +
" $r : Room()\n" +
" ?temperature( $r, $t; )\n" +
" eval( $t > 20 )" +
"then\n" +
" $r.getHeating().setOn(false);\n" +
"end\n";

Room room = new Room("Room");

final KieSession ksession = new KieHelper().addContent( drl, ResourceType.DRL )
.build()
.newKieSession();

room.getTemperatureSensor().setValue(15);
room.getHeating().setOn(false);
ksession.insert(room);
ksession.insert(room.getTemperatureSensor());
ksession.insert(room.getHeating());
ksession.fireAllRules();
assertThat(room.getHeating().isOn()).as("Temperature is bellow 20 degrees of Celsius. Heating should be turned on.").isTrue();

ksession.insert(new SensorEvent(room.getTemperatureSensor(), 25));
ksession.fireAllRules();
assertThat(room.getHeating().isOn()).as("Query is not reactive. Heating should still be turned on.").isTrue();

ksession.dispose();
}
}
39 changes: 39 additions & 0 deletions drools-compiler/src/test/java/org/drools/compiler/oopath/model/Appliance.java
View file
@@ -0,0 +1,39 @@
/*
* Copyright 2017 Red Hat, Inc. and/or its affiliates.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
*
* 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.drools.compiler.oopath.model;

import org.drools.core.phreak.AbstractReactiveObject;

public class Appliance extends AbstractReactiveObject {
private boolean on;

public Appliance() {
on = false;
}

public Appliance(boolean on) {
this.on = on;
notifyModification();
}

public boolean isOn() {
return on;
}

public void setOn(boolean on) {
this.on = on;
}
}

0 comments on commit 85baf6c

Please sign in to comment.