1 |
|
|
2 |
|
|
3 |
|
|
4 |
|
|
5 |
|
|
6 |
|
|
7 |
|
|
8 |
|
|
9 |
|
|
10 |
|
|
11 |
|
|
12 |
|
|
13 |
|
|
14 |
|
|
15 |
|
|
16 |
|
package org.doxla.spring.automock; |
17 |
|
|
18 |
|
import java.util.Collection; |
19 |
|
|
20 |
|
import org.doxla.spring.automock.resolver.DefaultProxyMockNameResolver; |
21 |
|
import org.doxla.spring.automock.resolver.MockClassResolver; |
22 |
|
import org.doxla.spring.automock.resolver.ProxyMockNameResolver; |
23 |
|
import org.jmock.cglib.Mock; |
24 |
|
import org.springframework.beans.BeansException; |
25 |
|
import org.springframework.beans.factory.annotation.Required; |
26 |
|
import org.springframework.beans.factory.config.BeanFactoryPostProcessor; |
27 |
|
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory; |
28 |
|
|
29 |
|
|
30 |
|
|
31 |
|
|
32 |
|
|
33 |
|
|
34 |
|
|
35 |
|
|
36 |
|
|
37 |
|
|
38 |
|
|
39 |
|
|
40 |
|
|
41 |
|
|
42 |
2 |
public class AutoMockExposingPostProcessor implements BeanFactoryPostProcessor { |
43 |
|
|
44 |
|
private MockClassResolver mockClassResolver; |
45 |
2 |
private ProxyMockNameResolver proxyMockNameResolver = new DefaultProxyMockNameResolver(); |
46 |
|
|
47 |
|
public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException { |
48 |
2 |
Collection<Class> mockClasses = mockClassResolver.resolveClassesToMock(beanFactory); |
49 |
2 |
for (Class clazz : mockClasses) { |
50 |
|
|
51 |
3 |
Mock mock = new Mock(clazz); |
52 |
3 |
Object proxy = mock.proxy(); |
53 |
|
|
54 |
3 |
beanFactory.registerSingleton(proxyMockNameResolver.mockNameFromClass(clazz), mock); |
55 |
3 |
beanFactory.registerSingleton(proxyMockNameResolver.proxyNameFromClass(clazz), proxy); |
56 |
|
|
57 |
3 |
} |
58 |
2 |
} |
59 |
|
|
60 |
|
@Required |
61 |
|
public void setMockClassResolver(MockClassResolver mockClassResolver) { |
62 |
2 |
this.mockClassResolver = mockClassResolver; |
63 |
2 |
} |
64 |
|
|
65 |
|
public void setProxyMockNameResolver(ProxyMockNameResolver proxyMockNameResolver) { |
66 |
0 |
this.proxyMockNameResolver = proxyMockNameResolver; |
67 |
0 |
} |
68 |
|
|
69 |
|
} |