安卓单元测试 (十)：DaggerMock, 让 Dagger2 与单元测试的结合易如反掌-白红宇

The Old Way

我们在系列的第六篇文章前面介绍了。大致过程是这样的，首先，你要mock出一个Module，让它的某个Provider方法在被调用的时候，返回你想到的mock的Dependency。然后使用这个mock的module来build出一个Component，再把这个Component放到你的ComponentHolder。举个例子说明一下，假设你有一个LoginActivity，里面有一个LoginPresenter，是通过Dagger2 inject进去的，如下：

public class LoginActivity extends AppCompatActivity {    @Inject    LoginPresenter mLoginPresenter;    @Override    protected void onCreate(Bundle savedInstanceState) {        //...other code        ComponentHolder.getAppComponent().inject(this);    }}//对应的Test类如下：@RunWith(RobolectricGradleTestRunner.class)@Config(constants = BuildConfig.class, sdk = 21)public class LoginActivityTest {    @Test    public void testLogin() {        AppModule mockAppModule = Mockito.mock(AppModule.class);        LoginPresenter mockLoginPresenter = mock(LoginPresenter.class);        Mockito.when(mockAppModule.provideLoginPresenter(any(UserManager.class), any(PasswordValidator.class))).thenReturn(mockLoginPresenter);  //当mockAppModule的provideLoginPresenter()方法被调用时，让它返回mockLoginPresenter        AppComponent appComponent = DaggerAppComponent.builder().appModule(mockAppModule).build();  //用mockAppModule来创建DaggerAppComponent        ComponentHolder.setAppComponent(appComponent);  //假设你的Component是放在ComponentHolder里面的        LoginActivity loginActivity = Robolectric.setupActivity(LoginActivity.class);        ((EditText) loginActivity.findViewById(R.id.username)).setText("xiaochuang");        ((EditText) loginActivity.findViewById(R.id.password)).setText("xiaochuang is handsome");        loginActivity.findViewById(R.id.login).performClick();            verify(mockLoginPresenter).login("xiaochuang", "xiaochuang is handsome");    }}复制代码

可以看到，为了让Dagger2返回一个Mock对象，我们需要写5行代码。再多写几个测试，我保证你一定会觉得繁琐的。当然，我们可以使用里面说的方式，和其它的一些手段，来简化代码，以下是我作出的一些努力，应该说，代码已经比较简洁了：

@RunWith(RobolectricGradleTestRunner.class)@Config(constants = BuildConfig.class, sdk = 21)public class LoginActivityTest {    @Rule    public MockitoRule mockitoRule = MockitoJUnit.rule();    @Mock    LoginPresenter loginPresenter;    @Test    public void testLogin() {        Mockito.when(TestUtils.appModule.provideLoginPresenter(any(UserManager.class), any(PasswordValidator.class))).thenReturn(loginPresenter);  //当mockAppModule的provideLoginPresenter()方法被调用时，让它返回mockLoginPresenter        TestUtils.setupDagger();        LoginActivity loginActivity = Robolectric.setupActivity(LoginActivity.class);        ((EditText) loginActivity.findViewById(R.id.username)).setText("xiaochuang");        ((EditText) loginActivity.findViewById(R.id.password)).setText("xiaochuang is handsome");        loginActivity.findViewById(R.id.login).performClick();        verify(loginPresenter).login("xiaochuang", "xiaochuang is handsome");    }}public class TestUtils {    public static final AppModule appModule = spy(new AppModule(RuntimeEnvironment.application));    public static void setupDagger() {        AppComponent appComponent = DaggerAppComponent.builder().appModule(appModule).build();        ComponentHolder.setAppComponent(appComponent);    }}复制代码

上面把dagger设置相关的代码减少到了两行，应该说，已经不再是一个负担了。然而哪怕是这样，如果写多了的话，依然会让人感觉略烦，因为这也完全是Boilerplate code（这里为什么要用“也”？）。再多写一点，你就会自然而然的想，如果能有一个工具，能达到这样的效果就好了：我们在Test类里面定义一个@Mock field（比如上面的loginPresenter），这个工具就能自动把这个field作为dagger的module对应的provider方法（provideLoginPresenter(...)）的返回值。也就是说，自动的mock module，让它返回这个@Mock field，然后用这个mock的module来build一个component，并放到ComponentHolder里面去。

New Hope

Well，我写这篇文章，就是想告诉大家，还真有人写了这样的一个工具，这就是这篇文章要介绍的。它就能达到我们上面描述的那种效果，让我们像使用Mockito Annotation一样来定义Mock，却能自动把它们作为Dagger2生产的Dependency。达到的效果如下：

@RunWith(RobolectricGradleTestRunner.class)@Config(constants = BuildConfig.class, sdk = 21)public class LoginActivityTest {    @Rule public DaggerRule daggerRule = new DaggerRule();    @Mock    LoginPresenter loginPresenter;    @Test    public void testLogin_shinny_way() {        LoginActivity loginActivity = Robolectric.setupActivity(LoginActivity.class);        ((EditText) loginActivity.findViewById(R.id.username)).setText("xiaochuang");        ((EditText) loginActivity.findViewById(R.id.password)).setText("xiaochuang is handsome");        loginActivity.findViewById(R.id.login).performClick();        verify(loginPresenter).login("xiaochuang", "xiaochuang is handsome");    }}复制代码

在上面的代码中，已经没有多余的Boilerplate code，要写的代码，基本是必需写的了。上面起作用的是@Rule public DaggerRule daggerRule = new DaggerRule(); 这行代码。可见，它是通过JUnit Rule来实现的。如果你熟悉，那么你很容易猜到这个DaggerRule的工作原理：

初始化一个测试类里面的所有用@Mock field为mock对象(loginPresenter)

mock AppModule，通过反射的方式得到AppModule的所有provider方法，如果有某个方法的返回值是一个LoginPresenter，那么就使用Mockito，让这个方法（provideLoginPresenter(...))被调用时，返回我们在测试类里面定义的mock loginPresenter。

使用这个mock AppModule来构建一个Component，并且放到ComponentHolder里面去。

我相信看到这里，你一定有很多疑问：

它怎么知道要使用AppModule

它怎么知道要build什么样的Componant

它怎么知道要把build出来的Component放到哪？

好吧，其实上面的DaggerRule，不是DaggerMock这个library自带的，是我们自己实现的。然而别着急，DaggerMock给了我们提供了一个父类Rule：DaggerMockRule，这个Rule已经帮我们做了绝大多数事情了。我们自定义的DaggerRule，其实也是继承自DaggerMockRule的，而我们在自定义Rule里面做的事情，也只不过是告诉DaggerMock，上面说到的三个问题的答案：要使用哪个Module、要build哪个Component、要把build好的Component放到哪，仅此而已。不信请看代码：

public class DaggerRule extends DaggerMockRule
    
      {    public DaggerRule() {        //告诉DaggerMock要build什么样的Component，使用哪个module        super(AppComponent.class, new AppModule(RuntimeEnvironment.application));        //告诉DaggerMock把build好的Component放到哪        set(new ComponentSetter
     
      () {            @Override            public void setComponent(AppComponent appComponent) {                ComponentHolder.setAppComponent(appComponent);            }        });    }}复制代码

怎么样，很简单吧？这个DaggerRule是可以重复使用的，一般来说，一个Component类对应于一个这样的DaggerRule就好了。自此，你可以只负责使用@Mock来定义mock了，dagger的事情就交给这个DaggerRule就好了。

是不是很爽！

哦对了，将这个library加到项目里面的姿势说一下，在build.gradle文件里面加入：

repositories {    jcenter()    maven { url "https://jitpack.io" }}复制代码

和

dependencies {    //...others dependencies    testCompile 'com.github.fabioCollini:DaggerMock:0.6.1'    androidTestCompile 'com.github.fabioCollini:DaggerMock:0.6.1' //如果你需要在Instrumentation、Espresso、UiAutomator里面使用的话}复制代码

我刚开始使用这个lib的时候，还是花了点时间来理解的，个人认为作者的和对应的写得都不算是很容易看懂，希望这篇文章能让帮助到各位一点点。

照例文中的代码在。

获取最新文章或想加入安卓单元测试交流群，请关注下方公众号