### Question:
StunnerWiresShapeStateHighlight implements WiresShapeHighlight<PickerPart.ShapePart> { @Override public void restore() { if (null != current && (Objects.isNull(currentPartState) || Objects.equals(PickerPart.ShapePart.BODY, currentPartState))) { current.applyState(ShapeState.NONE); setCurrent(null); } else { delegate.restore(); } currentPartState = null; } StunnerWiresShapeStateHighlight(final WiresManager wiresManager,
                                           final Event<CanvasUnhighlightEvent> unhighlightEvent);   StunnerWiresShapeStateHighlight(final StunnerWiresShapeHighlight delegate); @Override void highlight(final WiresShape shape,
                          final PickerPart.ShapePart part); @Override void error(final WiresShape shape,
                      final PickerPart.ShapePart shapePart); @Override void restore();  }### Answer:
@Test public void testRestore() { tested.setCurrent(viewWithState); tested.restore(); verify(stateExecutor, times(1)).accept(eq(ShapeState.NONE)); verify(delegate, never()).highlight(eq(viewWithState), any(PickerPart.ShapePart.class)); }