|
|
|
@ -56,20 +56,20 @@ abstract class AbstractMockitoBeanOverrideHandler extends BeanOverrideHandler { |
|
|
|
|
|
|
|
|
|
|
|
@Override |
|
|
|
@Override |
|
|
|
protected void trackOverrideInstance(Object mock, SingletonBeanRegistry trackingBeanRegistry) { |
|
|
|
protected void trackOverrideInstance(Object mock, SingletonBeanRegistry trackingBeanRegistry) { |
|
|
|
getMockitoBeans(trackingBeanRegistry).add(mock); |
|
|
|
getMockBeans(trackingBeanRegistry).add(mock); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
private static MockitoBeans getMockitoBeans(SingletonBeanRegistry trackingBeanRegistry) { |
|
|
|
private static MockBeans getMockBeans(SingletonBeanRegistry trackingBeanRegistry) { |
|
|
|
String beanName = MockitoBeans.class.getName(); |
|
|
|
String beanName = MockBeans.class.getName(); |
|
|
|
MockitoBeans mockitoBeans = null; |
|
|
|
MockBeans mockBeans = null; |
|
|
|
if (trackingBeanRegistry.containsSingleton(beanName)) { |
|
|
|
if (trackingBeanRegistry.containsSingleton(beanName)) { |
|
|
|
mockitoBeans = (MockitoBeans) trackingBeanRegistry.getSingleton(beanName); |
|
|
|
mockBeans = (MockBeans) trackingBeanRegistry.getSingleton(beanName); |
|
|
|
} |
|
|
|
} |
|
|
|
if (mockitoBeans == null) { |
|
|
|
if (mockBeans == null) { |
|
|
|
mockitoBeans = new MockitoBeans(); |
|
|
|
mockBeans = new MockBeans(); |
|
|
|
trackingBeanRegistry.registerSingleton(beanName, mockitoBeans); |
|
|
|
trackingBeanRegistry.registerSingleton(beanName, mockBeans); |
|
|
|
} |
|
|
|
} |
|
|
|
return mockitoBeans; |
|
|
|
return mockBeans; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
@Override |
|
|
|
@Override |
|
|
|
|