如何将Spring配置文件已配置的类Mock

1问题：

       手上在做一个历史遗留项目，整个项目有N个模块，我只负责其中一个子模块的功能的维护工作，项目的各个模块是通过spring来将各个模块耦合起来的。

启动项目已经有一个配置文件，并且已经打入了jar包。

      为了精简我所描述的场景，避免引入不必要的复杂性。我将不介绍每个模块中的具体实现，重点只描述各个模块之间的依赖关系。

假设我所维护的模块是A，模块A运行需要依赖模块B，模块B是其他人开发的。这个项目在前期是是基于接口开发的，这是一个非常好的前提，这样，我轻而易举就能做些小动作了。以下是这个系统的类图（虽然简单，但是能说明问题）

Spring的配置文件“application-context.xml“是这样写的：

<?xml version="1.0" encoding="utf-8" ?><beans xmlns:xsi="http://www.w3.org/2001/xmlschema-instance"><!-- 模块配置 --><bean id="moduleb" class="modulemock.defaultmodulebimpl" /><bean id="modulea" class="modulemock.defaultmoduleaimpl"><property name="moduleb" ref="moduleb" /></bean></beans>

      模块B在配置文件中的实现类已经写死了，在IModuleB的默认实现中是会去链接数据库，这样在测试过程中就需要去配置数据库连接，增加了测试的复杂度，所以很自然地就想到了用Mock技术，需要将ModuleB做成模拟对象。

2 问题解决：

首先很自然地就想到重写写一个配置文件,可以将moduleB的实现类改成模拟类，改成如下这种形式：

<pre name="code"><?xml version="1.0" encoding="utf-8" ?><beans xmlns:xsi="http://www.w3.org/2001/xmlschema-instance"><!-- 模块配置成mock类 --><bean id="moduleb" class="modulemock.mockmoduleb" /><bean id="modulea" class="modulemock.defaultmoduleaimpl"><property name="moduleb" ref="moduleb" /></bean></beans>

以上这个类只能在测试过程中使用，这样正式项目中和测试环境中使用的是两个spring配置文件，这两个配置文件中绝大数配置项是一样的只有ModuleB的配置是不一样的。这样的做法如果项目小的话没有问题，但是项目比较大的就会有问题。

而且，真实的项目中spring的配置文件远远比以上这个demo配置文件复杂许多，而且项目维护过程中经常会更改spring配置文件的配置，这样就需要保证测试环境的spring配置文件和正式项目中的spring配置文件保持同步，实际证明这是一件非常麻烦的事，难免会有遗漏。

现在的问题是，如何能不重写也不改写spring配置文件的前提下，将IModuleB接口Mock化。也就是测试环境和正式环境中只适用同一个 application-context.xml配置文件。办法当然有，那就是利用Spring的org.springframework.beans.factory.config.BeanPostProcessor 接口，利用它在初始化spring容器的过程中会对spring容器中配置的所有Bean会进行拦截：

1. 
   

2. package org.springframework.beans.factory.config;import org.springframework.beans.beansexception;public interface beanpostprocessor {object postprocessbeforeinitialization(object bean, string beanname) throws beansexception;object postprocessafterinitialization(object bean, string beanname) throws beansexception;}

   
   
   
3. postProcessBeforeInitialization 是在spring中配置的bean初始化之前调用的，如果配置的class实现了InitializingBean的afterPropertiesSet的方法，那就会在这bean调用afterPropertiesSet方法前调用该方法
4. postProcessAfterInitialization 这个方法会是在InitializingBean调用了afterPropertiesSet这个方法之后来执行

如果配置的bean没有实现InitializingBean这个类的话，无论实现哪个方法都无所了。

要将spring配置项中的对象mock的话，只要在postProcessBeforeInitialization方法上作文章，方法的第一个参数bean是spring容器中配置的bean，spring在实例化bean并且为其设置好依赖之后将这个对象传到post*方法中，原则上需要将这个对象作为返回值返回给容器，但是我们为了mock这个类，我们可以在post*方法中创建一个Mock对象，将这个mock对象作为返回值返回，如果移花接木一般在这里将spring容器中的实现类给改了。

以下是改动之前的项目启动代码：

import org.springframework.context.applicationcontext;import org.springframework.context.support.classpathxmlapplicationcontext;public class main {public static void main(string[] args) {applicationcontext context = new classpathxmlapplicationcontext("application-context.xml");imodulea modulea = context.getbean("modulea", imodulea.class);modulea.execute();}}

现在对他进行改动：

import org.springframework.beans.beansexception;import org.springframework.beans.factory.config.beanpostprocessor;import org.springframework.beans.factory.config.configurablelistablebeanfactory;import org.springframework.context.applicationcontext;import org.springframework.context.support.classpathxmlapplicationcontext;public class main {public static void main(string[] args) {applicationcontext context = new classpathxmlapplicationcontext("application-context.xml") {protected void preparebeanfactory(configurablelistablebeanfactory beanfactory) {beanfactory.addbeanpostprocessor(new beanpostprocessor() {@overridepublic object postprocessafterinitialization(object bean,string beanname) throws beansexception {return bean;}public object postprocessbeforeinitialization(object bean,string beanname) throws beansexception {if ("moduleb".equals(beanname)) {return new mockmoduleb();}return bean;}});}};imodulea modulea = context.getbean("modulea", imodulea.class);modulea.execute();}}以上最核心的代码是以下这段：public object postprocessbeforeinitialization(object bean,string beanname) throws beansexception {if ("moduleb".equals(beanname)) {return new mockmoduleb();}return bean;}

判断被拦截的bean的name属性是否等于“moduleB”，如果相等的话就是要拦截的目标，就实例化一个mock对象，将他返回。

3 总结：

以上利用BeanPostProcessor接口，在spring容器初始化过程中对容器中的bean进行拦截的方法，将容器中的bean进行mock，这个方法的好处是：

1. 测试环境和正式环境共用一个spring配置文件，减少配置维护工作。
2. spring配置中有任何改动马上可以通过测试验证。

在代码中能够利用spring的机制开创建mock对象，完全依赖于前期代码是面向接口开发的，如果不是这样的话，根本没有办法做Mock对象。在项目中使用面向接口开发，虽然会增加代码维护量，也许在项目前期好处不是很明显，但是在项目维护期会给我们带来方便。类似的例子还有很多，我遇见过一些项目开发前期为了赶进度，方便而放弃了一些软件开发的一些基本原则（ocp原则,里氏替换原则，面向接口开发原则等等。。。），导致项目开发到后期很难维护，造成堆代码的局面。

所以，在项目开发过程中，我们前期我们您可多花一些“无用功”，项目后期我们一定会受益的。