mirror of
https://github.com/spring-projects/spring-boot.git
synced 2024-07-15 01:07:30 +08:00
Reset mocks produced by FactoryBeans
An unwanted side-effect of the changes made inc6bdd136
to fix gh-7271 is that a mock produced by a factory bean is not reset. To allow such a mock to be reset without regressing the fix we now call getBean(…) as we did beforec6bdd136
, however the call is now performed in a defensive manner falling back to getSingleton(…) when it fails. Closes gh-33830
This commit is contained in:
parent
4aa82a3dc0
commit
9940fcfe77
@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright 2012-2021 the original author or authors.
|
||||
* Copyright 2012-2023 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
@ -78,7 +78,7 @@ public class ResetMocksTestExecutionListener extends AbstractTestExecutionListen
|
||||
for (String name : names) {
|
||||
BeanDefinition definition = beanFactory.getBeanDefinition(name);
|
||||
if (definition.isSingleton() && instantiatedSingletons.contains(name)) {
|
||||
Object bean = beanFactory.getSingleton(name);
|
||||
Object bean = getBean(beanFactory, name);
|
||||
if (reset.equals(MockReset.get(bean))) {
|
||||
Mockito.reset(bean);
|
||||
}
|
||||
@ -100,4 +100,13 @@ public class ResetMocksTestExecutionListener extends AbstractTestExecutionListen
|
||||
}
|
||||
}
|
||||
|
||||
private Object getBean(ConfigurableListableBeanFactory beanFactory, String name) {
|
||||
try {
|
||||
return beanFactory.getBean(name);
|
||||
}
|
||||
catch (Exception ex) {
|
||||
return beanFactory.getSingleton(name);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright 2012-2020 the original author or authors.
|
||||
* Copyright 2012-2023 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
@ -52,6 +52,7 @@ class ResetMocksTestExecutionListenerTests {
|
||||
given(getMock("none").greeting()).willReturn("none");
|
||||
given(getMock("before").greeting()).willReturn("before");
|
||||
given(getMock("after").greeting()).willReturn("after");
|
||||
given(getMock("fromFactoryBean").greeting()).willReturn("fromFactoryBean");
|
||||
}
|
||||
|
||||
@Test
|
||||
@ -59,6 +60,7 @@ class ResetMocksTestExecutionListenerTests {
|
||||
assertThat(getMock("none").greeting()).isEqualTo("none");
|
||||
assertThat(getMock("before").greeting()).isNull();
|
||||
assertThat(getMock("after").greeting()).isNull();
|
||||
assertThat(getMock("fromFactoryBean").greeting()).isNull();
|
||||
}
|
||||
|
||||
ExampleService getMock(String name) {
|
||||
@ -102,6 +104,11 @@ class ResetMocksTestExecutionListenerTests {
|
||||
return new BrokenFactoryBean();
|
||||
}
|
||||
|
||||
@Bean
|
||||
WorkingFactoryBean fromFactoryBean() {
|
||||
return new WorkingFactoryBean();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
static class BrokenFactoryBean implements FactoryBean<String> {
|
||||
@ -123,4 +130,23 @@ class ResetMocksTestExecutionListenerTests {
|
||||
|
||||
}
|
||||
|
||||
static class WorkingFactoryBean implements FactoryBean<ExampleService> {
|
||||
|
||||
@Override
|
||||
public ExampleService getObject() {
|
||||
return mock(ExampleService.class, MockReset.before());
|
||||
}
|
||||
|
||||
@Override
|
||||
public Class<?> getObjectType() {
|
||||
return ExampleService.class;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean isSingleton() {
|
||||
return true;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user