/*
* Copyright 2015 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.
* You may obtain a copy of the License at
*
* 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.optaplanner.core.api.score.buildin.bendable;
import org.junit.Test;
import org.kie.api.runtime.rule.RuleContext;
import org.optaplanner.core.api.score.holder.AbstractScoreHolderTest;
import static org.junit.Assert.*;
public class BendableScoreHolderTest extends AbstractScoreHolderTest {
@Test
public void addConstraintMatchWithConstraintMatch() {
addConstraintMatch(true);
}
@Test
public void addConstraintMatchWithoutConstraintMatch() {
addConstraintMatch(false);
}
public void addConstraintMatch(boolean constraintMatchEnabled) {
BendableScoreHolder scoreHolder = new BendableScoreHolder(constraintMatchEnabled, 1, 2);
RuleContext hard1 = mockRuleContext("hard1");
scoreHolder.addHardConstraintMatch(hard1, 0, -1);
assertEquals(BendableScore.valueOf(new int[]{-1}, new int[]{0, 0}), scoreHolder.extractScore(0));
RuleContext hard2Undo = mockRuleContext("hard2Undo");
scoreHolder.addHardConstraintMatch(hard2Undo, 0, -8);
assertEquals(BendableScore.valueOf(new int[]{-9}, new int[]{0, 0}), scoreHolder.extractScore(0));
callOnDelete(hard2Undo);
assertEquals(BendableScore.valueOf(new int[]{-1}, new int[]{0, 0}), scoreHolder.extractScore(0));
RuleContext medium1 = mockRuleContext("medium1");
scoreHolder.addSoftConstraintMatch(medium1, 0, -10);
callOnUpdate(medium1);
scoreHolder.addSoftConstraintMatch(medium1, 0, -20); // Overwrite existing
RuleContext soft1 = mockRuleContext("soft1", DEFAULT_JUSTIFICATION, OTHER_JUSTIFICATION);
scoreHolder.addSoftConstraintMatch(soft1, 1, -100);
callOnUpdate(soft1);
scoreHolder.addSoftConstraintMatch(soft1, 1, -300); // Overwrite existing
RuleContext multi1 = mockRuleContext("multi1");
scoreHolder.addMultiConstraintMatch(multi1, new int[]{-1000}, new int[]{-10000, -100000});
callOnUpdate(multi1);
scoreHolder.addMultiConstraintMatch(multi1, new int[]{-4000}, new int[]{-50000, -600000}); // Overwrite existing
RuleContext hard3 = mockRuleContext("hard3");
scoreHolder.addHardConstraintMatch(hard3, 0, -1000000);
callOnUpdate(hard3);
scoreHolder.addHardConstraintMatch(hard3, 0, -7000000); // Overwrite existing
RuleContext soft2Undo = mockRuleContext("soft2Undo", UNDO_JUSTIFICATION);
scoreHolder.addSoftConstraintMatch(soft2Undo, 1, -99);
callOnDelete(soft2Undo);
RuleContext multi2Undo = mockRuleContext("multi2Undo");
scoreHolder.addMultiConstraintMatch(multi2Undo, new int[]{-999}, new int[]{-999, -999});
callOnDelete(multi2Undo);
RuleContext medium2Undo = mockRuleContext("medium2Undo");
scoreHolder.addSoftConstraintMatch(medium2Undo, 0, -9999);
callOnDelete(medium2Undo);
assertEquals(BendableScore.valueOf(new int[]{-7004001}, new int[]{-50020, -600300}), scoreHolder.extractScore(0));
assertEquals(BendableScore.valueOfUninitialized(-7, new int[]{-7004001}, new int[]{-50020, -600300}), scoreHolder.extractScore(-7));
if (constraintMatchEnabled) {
assertEquals(BendableScore.valueOf(new int[]{-1}, new int[]{0, 0}), findConstraintMatchTotal(scoreHolder, "hard1").getScoreTotal());
assertEquals(BendableScore.valueOf(new int[]{0}, new int[]{0, -300}), scoreHolder.getIndictmentMap().get(OTHER_JUSTIFICATION).getScoreTotal());
assertNull(scoreHolder.getIndictmentMap().get(UNDO_JUSTIFICATION));
}
}
}