/*
* Copyright 2014 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.impl.domain.variable.anchor;
import java.util.Arrays;
import org.junit.Test;
import org.optaplanner.core.impl.domain.variable.descriptor.GenuineVariableDescriptor;
import org.optaplanner.core.impl.domain.variable.inverserelation.ExternalizedSingletonInverseVariableSupply;
import org.optaplanner.core.impl.score.director.ScoreDirector;
import org.optaplanner.core.impl.testdata.domain.chained.TestdataChainedAnchor;
import org.optaplanner.core.impl.testdata.domain.chained.TestdataChainedEntity;
import org.optaplanner.core.impl.testdata.domain.chained.TestdataChainedSolution;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;
public class ExternalizedAnchorVariableSupplyTest {
@Test
public void chainedEntity() {
GenuineVariableDescriptor variableDescriptor = TestdataChainedEntity.buildVariableDescriptorForChainedObject();
ScoreDirector scoreDirector = mock(ScoreDirector.class);
ExternalizedSingletonInverseVariableSupply nextVariableSupply = new ExternalizedSingletonInverseVariableSupply(variableDescriptor);
ExternalizedAnchorVariableSupply supply = new ExternalizedAnchorVariableSupply(variableDescriptor, nextVariableSupply);
TestdataChainedAnchor a0 = new TestdataChainedAnchor("a0");
TestdataChainedEntity a1 = new TestdataChainedEntity("a1", a0);
TestdataChainedEntity a2 = new TestdataChainedEntity("a2", a1);
TestdataChainedEntity a3 = new TestdataChainedEntity("a3", a2);
TestdataChainedAnchor b0 = new TestdataChainedAnchor("b0");
TestdataChainedEntity b1 = new TestdataChainedEntity("b1", b0);
TestdataChainedSolution solution = new TestdataChainedSolution("solution");
solution.setChainedAnchorList(Arrays.asList(a0, b0));
solution.setChainedEntityList(Arrays.asList(a1, a2, a3, b1));
when(scoreDirector.getWorkingSolution()).thenReturn(solution);
nextVariableSupply.resetWorkingSolution(scoreDirector);
supply.resetWorkingSolution(scoreDirector);
assertSame(a0, supply.getAnchor(a1));
assertSame(a0, supply.getAnchor(a2));
assertSame(a0, supply.getAnchor(a3));
assertSame(b0, supply.getAnchor(b1));
nextVariableSupply.beforeVariableChanged(scoreDirector, a3);
supply.beforeVariableChanged(scoreDirector, a3);
a3.setChainedObject(b1);
nextVariableSupply.afterVariableChanged(scoreDirector, a3);
supply.afterVariableChanged(scoreDirector, a3);
assertSame(a0, supply.getAnchor(a1));
assertSame(a0, supply.getAnchor(a2));
assertSame(b0, supply.getAnchor(a3));
assertSame(b0, supply.getAnchor(b1));
nextVariableSupply.clearWorkingSolution(scoreDirector);
supply.clearWorkingSolution(scoreDirector);
}
}