mps-coderules/reactor/Test/test/TestController.kt

959 lines
39 KiB
Kotlin

import jetbrains.mps.logic.reactor.core.*
import jetbrains.mps.logic.reactor.evaluation.*
import jetbrains.mps.logic.reactor.logical.Logical
import jetbrains.mps.logic.reactor.program.*
import jetbrains.mps.unification.Term
import jetbrains.mps.unification.test.MockTerm.*
import org.junit.After
import org.junit.Assert.*
import org.junit.Before
import org.junit.Test
import solver.EqualsSolver
import solver.MockSessionSolver
import solver.eq
import solver.is_eq
/**
* @author Fedor Isakov
*/
class TestController {
@Before fun beforeTest() {
}
@After fun afterTest() {
MockSession.deinit()
}
private class MockSession(val program: Program, val solver: SessionSolver) : EvaluationSession(), SessionObjects {
lateinit var controller: Controller
override fun controller(): Controller = controller
override fun sessionSolver(): SessionSolver = solver
override fun program(): Program = program
override fun storeView(): StoreView = TODO()
class MockBackend(val session: MockSession) : Backend {
override fun current(): EvaluationSession = session
override fun createConfig(program: Program): Config = TODO()
}
companion object {
lateinit var ourBackend : MockBackend
fun init(program: Program, solver: SessionSolver) {
ourBackend = MockBackend(MockSession(program, solver))
setBackend(ourBackend)
}
fun deinit() {
clearBackend(ourBackend)
}
}
}
private fun sessionSolver(expressionSolver: ExpressionSolver, equalsSolver: EqualsSolver): SessionSolver =
MockSessionSolver(expressionSolver, equalsSolver).apply {
init(PredicateSymbol("equals", 2), JavaPredicateSymbol.EXPRESSION0, JavaPredicateSymbol.EXPRESSION1, JavaPredicateSymbol.EXPRESSION2, JavaPredicateSymbol.EXPRESSION3) }
private fun Builder.controller(vararg occurrences: ConstraintOccurrence): Controller {
val solver = sessionSolver(env.expressionSolver, env.equalsSolver)
val program = MockProgram("test", handlers, registry = MockConstraintRegistry(solver))
MockSession.init(program, solver)
val controller = Controller(program, storeView = MockStoreView(listOf(* occurrences)))
MockSession.ourBackend.session.controller = controller
return controller
}
private fun Builder.controllerWithFeedback(feedbackHandler: EvaluationFeedbackHandler,
vararg occurrences: ConstraintOccurrence): Controller
{
val solver = sessionSolver(env.expressionSolver, env.equalsSolver)
val program = MockProgram("test", handlers, registry = MockConstraintRegistry(solver))
MockSession.init(program, solver)
val controller = Controller(program, storeView = MockStoreView(listOf(* occurrences)), feedbackHandler = feedbackHandler)
MockSession.ourBackend.session.controller = controller
return controller
}
private fun detailFeedback(msg: String): DetailedFeedback = object : DetailedFeedback(msg) {
override fun toString(): String = "{" + msg + "}"
}
private class MockStoreView(val occurrences: List<ConstraintOccurrence>) : StoreView {
val symbols = occurrences.map { it.constraint().symbol() }.toSet()
override fun constraintSymbols(): Iterable<ConstraintSymbol> = symbols
override fun allOccurrences(): Iterable<ConstraintOccurrence> = occurrences
override fun occurrences(symbol: ConstraintSymbol): Iterable<ConstraintOccurrence> =
occurrences.filter { it.constraint().symbol() == symbol }
}
private fun <T : Any> eq(left: T, right: T) = left eq right
private fun <T : Any> is_eq(left: T, right: T): Boolean = left is_eq right
@Test
fun processSingle() {
programWithRules(
rule("main1",
headKept(
constraint("main")
),
body(
constraint("foo")
))
).run {
controller().evaluate(occurrence("main")).let { view ->
assertEquals(
setOf(ConstraintSymbol("main", 0), ConstraintSymbol("foo", 0)),
view.allOccurrences().map { it.constraint().symbol() }.toSet())
}
}
}
@Test
fun processReplaced() {
programWithRules(
rule("main1",
headKept(
constraint("main")
),
body(
constraint("foo")
)),
rule("main2",
headKept(
constraint("foo")
),
headReplaced(
constraint("main")
),
body(
constraint("bar")
))
).run {
controller().evaluate(occurrence("main")).let { view ->
assertEquals(
setOf(ConstraintSymbol("bar", 0), ConstraintSymbol("foo", 0)),
view.allOccurrences().map { it.constraint().symbol() }.toSet())
}
}
}
@Test
fun basicExpression() {
var test : String = "not initialized"
programWithRules(
rule("main",
headKept(
constraint("main")
),
body(
statement { test = "value" }
))
).run {
controller().evaluate(occurrence("main"))
assertEquals("value", test)
}
}
@Test
fun paramExpression() {
var test = logical<String>("X")
//"not initialized"
programWithRules(
rule("main",
headKept(
constraint("main")
),
body(
statement ({ test.set("value") })
))
).run {
controller().evaluate(occurrence("main"))
assertEquals("value", test.get())
}
}
@Test
fun basicLogical() {
var test : String? = "not initialized"
val x = logical<String>("x")
x.setValue("expected")
programWithRules(
rule("main",
headKept(
constraint("main")
),
body(
statement ({ test = x.get() } )
))
).run {
controller().evaluate(occurrence("main"))
assertEquals("expected", test)
}
}
@Test
fun logicalCopy() {
var test : String? = "not initialized"
val (x,y) = logical<String>("x", "y")
x.setValue("expected")
programWithRules(
rule("main",
headKept(
constraint("main")
),
body(
equals(x, y),
constraint("next")
)),
rule("aux",
headKept(
constraint("next")
),
body(
statement ({ test = y.get() })
))
).run {
controller().evaluate(occurrence("main")).run {
assertEquals(
setOf(ConstraintSymbol("main", 0), ConstraintSymbol("next", 0)),
allOccurrences().map { it.constraint().symbol() }.toSet())
assertEquals("expected", test)
}
}
}
@Test
fun basicGuard() {
var test1 : String = "not initialized 1"
var test2 : String = "not initialized 2"
programWithRules(
rule("main1",
headKept(
constraint("main")
),
guard(
expression { false }
),
body(
statement { test1 = "not expected" }
)),
rule("main2",
headKept(
constraint("main")
),
guard(
expression { true }
),
body(
statement { test2 = "expected" }
))
).run {
controller().evaluate(occurrence("main"))
assertEquals("not initialized 1", test1)
assertEquals("expected", test2)
}
}
@Test
fun basicMetaLogical() {
val (X, Y) = metaLogical<String>("X", "Y")
val a = logical<String>("a")
programWithRules(
rule("rule1",
headReplaced(
constraint("foo", X)
),
body(
equals(X, Y),
constraint("bar", Y)
)
)
).controller().apply {
a.set("value") }.evaluate(occurrence("foo", a)).run {
assertSame(1, allOccurrences().toList().size)
val co = allOccurrences().first()
assertEquals(ConstraintSymbol("bar",1), co.constraint().symbol())
assertEquals(1, co.arguments().size)
val arg = co.arguments().first()
assertNotEquals(a, arg)
assertEquals(a.get(), (arg as Logical<String>).get())
}
}
@Test
fun occurrenceTerminated() {
programWithRules(
rule("first",
headKept( constraint("foo") ), body( constraint("expected1") )
),
rule("second",
headKept( constraint("foo") ), body( constraint("bar") )
),
rule("third",
headKept( constraint("foo") ), body( constraint("unexpected") )
),
rule("fourth",
headReplaced( constraint("bar"),
constraint("foo") ),
body( constraint("expected2") )
)
).controller().evaluate(occurrence("foo")).run {
assertEquals(
setOf(ConstraintSymbol("expected1", 0), ConstraintSymbol("expected2", 0)),
allOccurrences().map { co -> co.constraint().symbol() }.toSet())
}
}
@Test
fun occurrenceKeptActive() {
programWithRules(
rule("first",
headKept( constraint("foo") ), body( constraint("bar") )
),
rule("second",
headReplaced( constraint("foo") ), body( constraint("expected1") )
),
rule("third",
headReplaced( constraint("foo") ), body( constraint("unexpected") )
),
rule("fourth",
headKept( constraint("foo") ),
headReplaced( constraint("bar") ),
body( constraint("expected2") )
)
).controller().evaluate(occurrence("foo")).run {
assertEquals(
setOf(ConstraintSymbol("expected1", 0), ConstraintSymbol("expected2", 0)),
allOccurrences().map { co -> co.constraint().symbol() }.toSet())
}
}
@Test
fun occurrenceReactivated() {
val X = metaLogical<Int>("X")
programWithRules(
rule("zeroth",
headKept( constraint("foo") ), body( statement({ x -> x.set(999) }, X),
constraint("bar", X))
),
rule("first",
headKept( constraint("foo") ),
body( constraint("bar", X),
constraint("qux", X))
),
rule("second",
headReplaced( constraint("qux", X) ),
body( constraint("expected1"),
statement({ x -> x.set(123) }, X))
),
rule("third",
headReplaced( constraint("foo") ),
body( constraint("unexpected"))
),
rule("fourth",
headReplaced( constraint("foo") ),
headReplaced( constraint("bar", X) ), guard(expression({ x -> x.getNullable() == 123 }, X)),
body( constraint("expected2") )
),
rule("fifth",
headReplaced( constraint("bar", X) ), guard(expression({ x -> x.getNullable() == 999 }, X)),
body( constraint("expected3", X))
)
).controller().evaluate(occurrence("foo")).run {
assertEquals(3, allOccurrences().count())
assertEquals(
setOf(ConstraintSymbol("expected1", 0), ConstraintSymbol("expected2", 0), ConstraintSymbol("expected3", 1)),
allOccurrences().map { co -> co.constraint().symbol() }.toSet())
val ex3 = allOccurrences().filter { co -> co.constraint().symbol() == ConstraintSymbol("expected3", 1) }.first()
assertEquals(999, (ex3.arguments().first() as Logical<Int>).value())
}
}
@Test
fun occurrenceReactivatedAfterUnion() {
val (X, Y) = metaLogical<Int>("X", "Y")
programWithRules(
rule("first",
headKept( constraint("foo") ),
body( constraint("bar", X),
constraint("qux", Y),
statement({ x, y -> eq(x, y) }, X, Y))
),
rule("second",
headReplaced( constraint("qux", Y) ),
body( constraint("expected1"),
statement({ y -> y.set(123) }, Y))
),
rule("third",
headReplaced( constraint("foo") ),
body( constraint("unexpected"))
),
rule("fourth",
headReplaced( constraint("foo") ),
headReplaced( constraint("bar", X) ), guard(expression({ x -> x.getNullable() == 123 }, X)),
body( constraint("expected2") )
),
rule("fifth",
headKept( constraint("bar", X) ),
body( constraint("expected3", X))
)
).controller().evaluate(occurrence("foo")).run {
assertEquals(3, allOccurrences().count())
assertEquals(
setOf(ConstraintSymbol("expected1", 0), ConstraintSymbol("expected2", 0), ConstraintSymbol("expected3", 1)),
allOccurrences().map { co -> co.constraint().symbol() }.toSet())
val ex3 = allOccurrences().filter { co -> co.constraint().symbol() == ConstraintSymbol("expected3", 1) }.first()
assertEquals(123, (ex3.arguments().first() as Logical<Int>).value())
}
}
@Test
fun occurrenceReactivatedAfterUnionUnbound() {
val (X, Y) = metaLogical<Int>("X", "Y")
programWithRules(
rule("first",
headKept( constraint("foo") ),
body( constraint("bar", X),
statement({ x, y -> eq(x, y) }, X, Y),
constraint("qux", Y))
),
rule("second",
headReplaced( constraint("qux", Y) ),
body( constraint("expected1"),
statement({ y -> y.set(123) }, Y))
),
rule("third",
headReplaced( constraint("foo") ),
body( constraint("unexpected"))
),
rule("fourth",
headReplaced( constraint("foo") ),
headReplaced( constraint("bar", X) ), guard(expression({ x -> x.getNullable() == 123 }, X)),
body( constraint("expected2") )
),
rule("fifth",
headKept( constraint("bar", X) ),
body( constraint("expected3", X))
)
).controller().evaluate(occurrence("foo")).run {
assertEquals(3, allOccurrences().count())
assertEquals(setOf( ConstraintSymbol("expected1", 0),
ConstraintSymbol("expected2", 0),
ConstraintSymbol("expected3", 1)),
allOccurrences().map { co -> co.constraint().symbol() }.toSet())
val ex3 = allOccurrences().filter { co -> co.constraint().symbol() == ConstraintSymbol("expected3", 1) }.first()
assertEquals(123, (ex3.arguments().first() as Logical<Int>).value())
}
}
@Test
fun correctRulesOrder() {
val X= metaLogical<Int>("X")
programWithRules(
rule("main",
headReplaced( constraint("main") ), body( statement({ x -> x.set(1) }, X),
constraint("bar"),
constraint("foo", X) )
),
rule("foo_if_zero",
headReplaced( constraint("foo", X) ), guard( expression({ x -> x.get() == 0 }, X) ),
body( constraint("foo_zero") )
),
rule("foo_and_bar",
headReplaced( constraint("foo", X) ),
headKept( constraint("bar") ),
body( constraint("foo_and_bar") )
),
rule("foo_if_non_zero",
headReplaced( constraint("foo", X) ),
guard( expression({ x -> x.get() != 0 }, X) ),
body( constraint("foo_non_zero") )
)
).controller().evaluate(occurrence("main")).run {
assertEquals(setOf(ConstraintSymbol("bar", 0), ConstraintSymbol("foo_and_bar", 0)), constraintSymbols())
assertEquals(1, occurrences(ConstraintSymbol("foo_and_bar", 0)).count())
assertEquals(1, occurrences(ConstraintSymbol("bar", 0)).count())
}
}
@Test
fun inverseOccurrenceOrder() {
val (X, Y, Z) = metaLogical<Int>("X", "Y", "Z")
val W = metaLogical<Int>("W")
programWithRules(
rule("main",
headReplaced(constraint("main")), body( statement({ w -> w.set(42) }, W),
constraint("foo", W, "a{c}"),
constraint("foo", W, "a{b}"),
constraint("foo", W, "a{d}"))
),
rule("expected",
headReplaced(
constraint("foo", X, "a{d}"),
constraint("foo", Y, "a{b}"),
constraint("foo", Z, "a{c}")), body(constraint("done")))
).controller().evaluate(occurrence("main")).run {
assertEquals(setOf(ConstraintSymbol("done", 0)), constraintSymbols())
assertEquals(1, occurrences(ConstraintSymbol.symbol("done", 0)).count())
}
}
@Test
fun reactivateOnUnion() {
val (X1,Y1,Z1) = metaLogical<Int>("X1", "Y1", "Z1")
val (X2,Y2,Z2) = metaLogical<Int>("X2", "Y2", "Z2")
val (X3,Y3,Z3) = metaLogical<Int>("X3", "Y3", "Z3")
var count = 0
programWithRules(
rule("main",
headReplaced( constraint("main") ),
body(
statement({ z -> z.set(0) }, Z1),
constraint("foo", X1, Z1),
constraint("foo", Y1, Z1),
statement({ x, y -> eq(x, y) }, X1, Y1) )
),
rule("capture_foo",
headKept( constraint("foo", X2, Y2) ),
body(
statement({ z -> z.set(count++) }, Z2),
constraint("capture", Z2) )
),
rule("capture_foo_foo",
headKept( constraint("foo", X3, Z3) ),
headReplaced( constraint("foo", Y3, Z3) ),
guard(
expression({ x, y -> is_eq(x, y) }, X3, Y3)),
body(
constraint("replaced") )
)
).controller().evaluate(occurrence("main")).run {
assertEquals(setOf( ConstraintSymbol("foo", 2),
ConstraintSymbol("capture", 1),
ConstraintSymbol("replaced", 0)),
constraintSymbols())
assertEquals(1, occurrences(ConstraintSymbol("foo", 2)).count())
assertEquals(2, occurrences(ConstraintSymbol("capture", 1)).count())
assertEquals(1, occurrences(ConstraintSymbol("replaced", 0)).count())
}
}
@Test
fun propagationHistory() {
val (X,Y,Z) = metaLogical<Int>("X", "Y", "Z")
programWithRules(
rule("main",
headReplaced( constraint("main") ), body( statement({ x, y -> eq(x, y) }, X, Y), // rank(X) = 1
constraint("foo", Y),
constraint("bar", Z),
// update Z's parent
statement({ x, z -> eq(x, z) }, X, Z) )
),
rule("foobar",
headKept( constraint("foo", X) ),
headKept( constraint("bar", Y) ),
body( constraint("foobar") )
)
).controller().evaluate(occurrence("main")).run {
assertEquals(setOf( ConstraintSymbol("foo", 1),
ConstraintSymbol("bar", 1),
ConstraintSymbol("foobar", 0)),
constraintSymbols())
assertEquals(1, occurrences(ConstraintSymbol("foo", 1)).count())
assertEquals(1, occurrences(ConstraintSymbol("bar", 1)).count())
assertEquals(1, occurrences(ConstraintSymbol("foobar", 0)).count())
}
}
@Test
fun propagationHistoryLogical() {
val X = metaLogical<Int>("X")
programWithRules(
rule("main",
headReplaced( constraint("main") ), body( constraint("foo", X),
statement({ x -> eq(x, "doh") }, X)
)
),
rule("foobar",
headKept( constraint("foo", X) ),
guard( expression ({ x -> x.isBound }, X) ),
body( constraint("foobar") )
)
).controller().evaluate(occurrence("main")).run {
assertEquals(setOf( ConstraintSymbol("foo", 1),
ConstraintSymbol("foobar", 0)),
constraintSymbols())
assertEquals(1, occurrences(ConstraintSymbol("foo", 1)).count())
assertEquals(1, occurrences(ConstraintSymbol("foobar", 0)).count())
}
}
@Test
fun propagationHistoryLogical2() {
val (X, Y) = metaLogical<Int>("X", "Y")
programWithRules(
rule("main",
headReplaced( constraint("main") ), body( constraint("foo", X),
constraint("bar", Y),
statement({ x -> eq(x, "doh") }, X)
)
),
rule("foobar",
headKept( constraint("foo", X),
constraint("bar", Y)
),
guard( expression ({ x -> x.isBound }, X) ),
body( constraint("foobar") )
)
).controller().evaluate(occurrence("main")).run {
assertEquals(setOf( ConstraintSymbol("foo", 1),
ConstraintSymbol("bar", 1),
ConstraintSymbol("foobar", 0)),
constraintSymbols())
assertEquals(1, occurrences(ConstraintSymbol("foo", 1)).count())
assertEquals(1, occurrences(ConstraintSymbol("bar", 1)).count())
assertEquals(1, occurrences(ConstraintSymbol("foobar", 0)).count())
}
}
@Test
fun removeObserver() {
val (X1,Y1,Z1) = metaLogical<Int>("X1", "Y1", "Z1")
val X2 = metaLogical<Int>("X2")
val (X3,Y3) = metaLogical<Int>("X3", "Y3")
val X4 = metaLogical<Int>("X4")
programWithRules(
rule("main",
headReplaced( constraint("main") ), body( statement({ x, y -> eq(x, y) }, X1, Y1), // rank(X) = 1
statement({ x -> x.set(42) }, X1),
constraint("match", Z1, X1),
constraint("trigger", Z1) )
),
rule("trigger",
headReplaced( constraint("trigger", X2) ),
body( constraint("foobar", X2) )
),
rule("nofoobar",
headReplaced( constraint("foobar", X3),
constraint("match", X3, Y3) ),
body( constraint("expected"),
constraint("blah"),
statement({ x, z -> eq(x, z) }, X3, Y3) )
),
rule("blah",
headReplaced( constraint("blah") ),
headReplaced( constraint("foobar", X4) ),
body( constraint("unexpected") )
)
).controller().evaluate(occurrence("main")).run {
assertEquals(setOf( ConstraintSymbol("blah", 0),
ConstraintSymbol("expected", 0)),
constraintSymbols())
assertEquals(1, occurrences(ConstraintSymbol("blah", 0)).count())
assertEquals(1, occurrences(ConstraintSymbol("expected", 0)).count())
}
}
@Test
fun reactivateOnUnionKeepValue() {
val (X,Y,Z) = metaLogical<Int>("X", "Y", "Z")
programWithRules(
rule("main",
headReplaced( constraint("main") ), body( statement({ x, y -> eq(x, y) }, X, Y), // rank(X) = 1
statement({ z -> z.set(42) }, Z),
constraint("foo", Z),
statement({ x, z -> eq(x, z) }, X, Z) )
),
rule("capture_foo_free",
headKept( constraint("foo", X) ), guard( expression({ x -> x.getNullable() == null }, X) ),
body( constraint("free") )
),
rule("capture_foo_assigned",
headKept( constraint("foo", X) ), guard( expression({ x -> x.getNullable() != null }, X) ),
body( constraint("assigned") )
)
).controller().evaluate(occurrence("main")).run {
assertEquals(setOf( ConstraintSymbol("foo", 1),
ConstraintSymbol("assigned", 0)),
constraintSymbols())
assertEquals(1, occurrences(ConstraintSymbol("foo", 1)).count())
assertEquals(1, occurrences(ConstraintSymbol("assigned", 0)).count())
}
}
@Test
fun firstAlternative() {
val (X, Y) = metaLogical<Int>("X", "Y")
programWithRules(
rule("main",
headReplaced( constraint("main") ), body( statement({ x -> x.set(7) }, X),
statement({ y -> y.set(7) }, Y),
constraint("aux", X, Y) )
),
rule("aux",
headReplaced( constraint("aux", X, Y) ), body( equals(X, Y),
constraint("expected") ),
altBody( constraint("unexpected") )
)
).controller().evaluate(occurrence("main")).run {
assertEquals(setOf(ConstraintSymbol("expected", 0)), constraintSymbols())
assertEquals(1, occurrences(ConstraintSymbol.symbol("expected", 0)).count())
}
}
@Test
fun secondAlternative() {
val (X, Y) = metaLogical<Int>("X", "Y")
programWithRules(
rule("main",
headReplaced( constraint("main") ), body( statement({ x -> x.set(7) }, X),
statement({ y -> y.set(13) }, Y),
constraint("aux", X, Y) )
),
rule("aux",
headReplaced( constraint("aux", X, Y) ), body( equals(X, Y),
constraint("unexpected") ),
altBody( constraint("expected") )
)
).controller().evaluate(occurrence("main")).run {
assertEquals(setOf(ConstraintSymbol("expected", 0)), constraintSymbols())
assertEquals(1, occurrences(ConstraintSymbol.symbol("expected", 0)).count())
}
}
@Test(expected = EvaluationFailureException::class)
fun lastAlternativeFail() {
val (X, Y, Z) = metaLogical<Int>("X", "Y", "Z")
programWithRules(
rule("main",
headReplaced( constraint("main") ), body( statement({ x -> x.set(7) }, X),
statement({ y -> y.set(13) }, Y),
statement({ z -> z.set(17) }, Z),
constraint("aux", X, Y, Z) )
),
rule("aux",
headReplaced( constraint("aux", X, Y, Z) ), body( equals(X, Y),
constraint("unexpected1") ),
altBody( equals(X, Z),
constraint("unexpected2") )
)
).controller().run {
try {
evaluate(occurrence("main"))
}
finally {
assertEquals(emptySet<ConstraintSymbol>(), storeView().constraintSymbols())
}
}
}
@Test
fun matchHeadWithTerm() {
val (X, Y, Z) = metaLogical<Term>("X", "Y", "Z")
val xLogical = X.logical()
val yLogical = Y.logical()
val zLogical = Z.logical()
programWithRules(
rule("main",
headReplaced(constraint("main")),
body(
constraint("foo", term("bar", logicalVar(zLogical))),
constraint("foo", term("bar", logicalVar(xLogical))),
constraint("trigger", xLogical)
)
),
rule("rule1",
headReplaced(
constraint("trigger", Y),
constraint("foo", term("bar", metaVar(X)))
),
guard(
expression ({ x, y -> x.findRoot() == y.findRoot() }, X, Y)
),
body(
constraint("triggered1", X, Y)
)
),
rule("rule2",
headReplaced(
constraint("trigger", Y),
constraint("foo", X)
),
guard(
expression ({ x, y -> !y.isBound && x.findRoot() == y.findRoot() }, X, Y)
),
body(
constraint("triggered2", X, Y)
)
)
).controller().run {
evaluate(occurrence("main"))
storeView().run {
val foo = ConstraintSymbol("foo", 1)
val t1 = ConstraintSymbol("triggered1", 2)
constraintSymbols() shouldBe setOf(t1, foo)
occurrences(t1).count() shouldBe 1
occurrences(t1).first().arguments().first() shouldBe xLogical
occurrences(foo).count() shouldBe 1
occurrences(foo).first().arguments().first() shouldBe term("bar", logicalVar(zLogical))
}
}
}
@Test(expected = EvaluationFailureException::class)
fun failureHandler() {
val failureHandler = object : FailureHandler {
val failures = ArrayList<Pair<EvaluationFailure, String>>()
override fun handleFailure(failure: EvaluationFailure, rule: Rule): EvaluationFailure? {
failures.add(failure to rule.tag())
return failure
}
}
programWithRules(
rule("main",
headReplaced(constraint("main")),
body (
constraint("foo")
)
),
rule("rule1",
headReplaced(constraint("foo")),
body(
constraint("yes"),
constraint("bar"),
constraint("no")
)
),
rule("rule2",
headReplaced(constraint("bar")),
body(
statement { throw EvaluationFailureException("unhandled") }
)
)
).controllerWithFeedback(failureHandler).run {
try {
evaluate(occurrence("main"))
} finally {
failureHandler.failures.map { (f, t) -> "${f.cause.message}@${t}" }.toList() shouldBe
listOf("unhandled@rule2", "unhandled@rule1", "unhandled@main")
}
}
}
@Test
fun failureHandlerRecover() {
val failureHandler = object : FailureHandler {
val failures = ArrayList<Pair<EvaluationFailure, String>>()
override fun handleFailure(failure: EvaluationFailure, rule: Rule): EvaluationFailure? {
failures.add(failure to rule.tag())
if (rule.tag()?.startsWith("recoverable") == true) return null
return failure
}
}
programWithRules(
rule("main",
headReplaced(constraint("main")),
body (
constraint("foo")
)
),
rule("rule1",
headReplaced(constraint("foo")),
body(
statement {
// this failure does not propagate because of the alt body branch
throw EvaluationFailureException("unhandled")
}
),
altBody(
constraint("recovered", 1),
constraint("bar"),
constraint("recovered", 2)
)
),
rule ("recoverable",
headReplaced(constraint("bar")),
body(
constraint("bazz")
)
),
rule("rule3",
headReplaced(constraint("bazz")),
body(
statement { throw EvaluationFailureException("handled") }
)
)
).controllerWithFeedback(failureHandler).run {
evaluate(occurrence("main"))
storeView().run {
val recovered = ConstraintSymbol("recovered", 1)
constraintSymbols() shouldBe setOf(recovered)
occurrences(recovered).map { it.arguments()[0] }.toSet() shouldBe setOf(1, 2)
}
}
failureHandler.failures.map { (f, t) -> "${f.cause.message}@$t"}.toList() shouldBe
listOf("handled@rule3", "handled@recoverable")
}
@Test
fun detailsFeedbackHandler() {
val feedbackHandler = object : EvaluationFeedbackHandler {
val feedbacks = arrayListOf<Pair<EvaluationFeedback, String>>()
override fun handleFeedback(rule: Rule, feedback: EvaluationFeedback): Boolean {
feedbacks.add(feedback to rule.tag())
return feedback.message.startsWith("catchme")
}
}
programWithRules(
rule("main",
headReplaced(constraint("main")),
body (
constraint("foo")
)
),
rule("rule1",
headReplaced(constraint("foo")),
body(
statement { invocationContext().report(detailFeedback("catchme")) },
constraint("bar")
)
),
rule("rule2",
headReplaced(constraint("bar")),
body(
statement { invocationContext().report(detailFeedback("propagateme")) }
)
)
).controllerWithFeedback(feedbackHandler).run {
evaluate(occurrence("main"))
}
feedbackHandler.feedbacks.map { (f, t) -> "${f.message}@$t"}.toList() shouldBe
listOf("catchme@rule1", "propagateme@rule2", "propagateme@rule1", "propagateme@main")
}
}