public class DefaultExhaustiveSearchPhaseTest {
@Test
public void restoreWorkingSolution() {
ExhaustiveSearchPhaseScope phaseScope = mock(ExhaustiveSearchPhaseScope.class);
ExhaustiveSearchStepScope lastCompletedStepScope = mock(ExhaustiveSearchStepScope.class);
when(phaseScope.getLastCompletedStepScope()).thenReturn(lastCompletedStepScope);
ExhaustiveSearchStepScope stepScope = mock(ExhaustiveSearchStepScope.class);
when(stepScope.getPhaseScope()).thenReturn(phaseScope);
Solution workingSolution = mock(Solution.class);
when(phaseScope.getWorkingSolution()).thenReturn(workingSolution);
ExhaustiveSearchLayer layer0 = new ExhaustiveSearchLayer(0, mock(Object.class), 100);
ExhaustiveSearchLayer layer1 = new ExhaustiveSearchLayer(1, mock(Object.class), 99);
ExhaustiveSearchLayer layer2 = new ExhaustiveSearchLayer(2, mock(Object.class), 98);
ExhaustiveSearchLayer layer3 = new ExhaustiveSearchLayer(3, mock(Object.class), 97);
ExhaustiveSearchLayer layer4 = new ExhaustiveSearchLayer(4, mock(Object.class), 96);
ExhaustiveSearchNode node0 = new ExhaustiveSearchNode(layer0, null);
node0.setMove(mock(Move.class));
node0.setUndoMove(mock(Move.class));
ExhaustiveSearchNode node1 = new ExhaustiveSearchNode(layer1, node0);
node1.setMove(mock(Move.class));
node1.setUndoMove(mock(Move.class));
ExhaustiveSearchNode node2A = new ExhaustiveSearchNode(layer2, node1);
node2A.setMove(mock(Move.class));
node2A.setUndoMove(mock(Move.class));
ExhaustiveSearchNode node3A = new ExhaustiveSearchNode(layer3, node2A); // oldNode
node3A.setMove(mock(Move.class));
node3A.setUndoMove(mock(Move.class));
ExhaustiveSearchNode node2B = new ExhaustiveSearchNode(layer2, node1);
node2B.setMove(mock(Move.class));
node2B.setUndoMove(mock(Move.class));
ExhaustiveSearchNode node3B = new ExhaustiveSearchNode(layer3, node2B);
node3B.setMove(mock(Move.class));
node3B.setUndoMove(mock(Move.class));
ExhaustiveSearchNode node4B = new ExhaustiveSearchNode(layer4, node3B); // newNode
node4B.setMove(mock(Move.class));
node4B.setUndoMove(mock(Move.class));
Score newScore = SimpleScore.valueOf(7);
node4B.setScore(newScore);
when(lastCompletedStepScope.getExpandingNode()).thenReturn(node3A);
when(stepScope.getExpandingNode()).thenReturn(node4B);
DefaultExhaustiveSearchPhase phase = new DefaultExhaustiveSearchPhase();
phase.setEntitySelector(mock(EntitySelector.class));
phase.setDecider(mock(ExhaustiveSearchDecider.class));
phase.restoreWorkingSolution(stepScope);