用参数模拟构造函数

时间:2012-11-13 16:17:10

标签: java junit mocking mockito powermock

我有一个课程如下:

public class A {
    public A(String test) {
        bla bla bla
    }

    public String check() {
        bla bla bla
    }
}

构造函数A(String test)check()中的逻辑是我试图模仿的东西。我想要任何调用:new A($$$any string$$$).check()返回一个虚拟字符串"test"

我试过了:

 A a = mock(A.class); 
 when(a.check()).thenReturn("test");

 String test = a.check(); // to this point, everything works. test shows as "tests"

 whenNew(A.class).withArguments(Matchers.anyString()).thenReturn(rk);
 // also tried:
 //whenNew(A.class).withParameterTypes(String.class).withArguments(Matchers.anyString()).thenReturn(rk);

 new A("random string").check();  // this doesn't work

但它似乎没有起作用。 new A($$$any string$$$).check()仍然通过构造函数逻辑而不是获取A的模拟对象。

6 个答案:

答案 0 :(得分:73)

您发布的代码适用于最新版本的Mockito和Powermockito。也许你还没准备好A? 试试这个:

A.java

public class A {
     private final String test;

    public A(String test) {
        this.test = test;
    }

    public String check() {
        return "checked " + this.test;
    }
}

MockA.java

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.equalTo;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

@RunWith(PowerMockRunner.class)
@PrepareForTest(A.class)
public class MockA {
    @Test
    public void test_not_mocked() throws Throwable {
        assertThat(new A("random string").check(), equalTo("checked random string"));
    }
    @Test
    public void test_mocked() throws Throwable {
         A a = mock(A.class); 
         when(a.check()).thenReturn("test");
         PowerMockito.whenNew(A.class).withArguments(Mockito.anyString()).thenReturn(a);
         assertThat(new A("random string").check(), equalTo("test"));
    }
}

两个测试都应该通过mockito 1.9.0,powermockito 1.4.12和junit 4.8.2

答案 1 :(得分:44)

据我所知,你不能用mockito模拟构造函数,只能使用方法。但根据Mockito谷歌代码页上的维基,有一种方法可以通过在类中创建一个返回该类的新实例的方法来模拟构造函数行为。然后你可以模拟出那种方法。以下是excerpt directly from the Mockito wiki

  

模式1 - 使用单行方法创建对象

     

要使用模式1(测试名为MyClass的类),您将替换类似

的调用
   Foo foo = new Foo( a, b, c );
     

   Foo foo = makeFoo( a, b, c );
     

并编写单行方法

   Foo makeFoo( A a, B b, C c ) { 
        return new Foo( a, b, c );
   }
     

在方法中不包含任何逻辑非常重要;只是一条线创造了   宾语。原因是该方法本身永远不会发生   进行单元测试。

     

当您来测试该类时,您测试的对象将是   实际上是一个Mockito间谍,用这种方法覆盖,返回一个   嘲笑。因此,你所测试的不是班级本身,而是a   非常轻微的修改版本。

     

您的测试类可能包含

等成员
  @Mock private Foo mockFoo;
  private MyClass toTest = spy(new MyClass());
     

最后,在你的测试方法中,你模拟了对它的调用   makeFoo有一行像

  doReturn( mockFoo )
      .when( toTest )
      .makeFoo( any( A.class ), any( B.class ), any( C.class ));
     

如果您愿意,可以使用比任何()更具体的匹配器   检查传递给构造函数的参数。

如果您只是想要返回课程中的模拟对象,我认为这应该适合您。在任何情况下,您都可以在此处阅读有关模拟对象创建的更多信息:

http://code.google.com/p/mockito/wiki/MockingObjectCreation

答案 2 :(得分:10)

不使用Powermock ....根据Ben Glasser的回答,请参阅下面的示例,因为我花了一些时间才弄清楚..希望能节省一些时间......

原始类:

public class AClazz {

    public void updateObject(CClazz cClazzObj) {
        log.debug("Bundler set.");
        cClazzObj.setBundler(new BClazz(cClazzObj, 10));
    } 
}

修改后的课程:

@Slf4j
public class AClazz {

    public void updateObject(CClazz cClazzObj) {
        log.debug("Bundler set.");
        cClazzObj.setBundler(getBObject(cClazzObj, 10));
    }

    protected BClazz getBObject(CClazz cClazzObj, int i) {
        return new BClazz(cClazzObj, 10);
    }
 }

测试类

public class AClazzTest {

    @InjectMocks
    @Spy
    private AClazz aClazzObj;

    @Mock
    private CClazz cClazzObj;

    @Mock
    private BClazz bClassObj;

    @Before
    public void setUp() throws Exception {
        Mockito.doReturn(bClassObj)
               .when(aClazzObj)
               .getBObject(Mockito.eq(cClazzObj), Mockito.anyInt());
    }

    @Test
    public void testConfigStrategy() {
        aClazzObj.updateObject(cClazzObj);

        Mockito.verify(cClazzObj, Mockito.times(1)).setBundler(bClassObj);
    }
}

答案 3 :(得分:6)

使用Mockito,您可以使用withSettings(),例如,如果CounterService需要2个依赖项,则可以将它们作为模拟传递:

UserService userService = Mockito.mock(UserService.class); SearchService searchService = Mockito.mock(SearchService.class); CounterService counterService = Mockito.mock(CounterService.class, withSettings().useConstructor(userService, searchService));

答案 4 :(得分:5)

version 3.5.0 of Mockito 开始并使用 InlineMockMaker,您现在可以模拟对象构造:

 try (MockedConstruction mocked = mockConstruction(A.class)) {
   A a = new A();
   when(a.check()).thenReturn("bar");
 }

try-with-resources 构造中所有对象构造都返回一个模拟。

答案 5 :(得分:4)

Mockito在测试最终,静态和私有方法方面存在局限性。

使用jMockit测试库,您可以轻松简单地执行以下操作:

java.io.File类的模拟构造函数:

new MockUp<File>(){
    @Mock
    public void $init(String pathname){
        System.out.println(pathname);
        // or do whatever you want
    }
};
  • 公共构造函数名称应替换为$ init
  • 抛出的参数和异常保持不变
  • 返回类型应定义为void

模拟静态方法:

  • 从方法模拟签名中删除静态
  • 方法签名保持不变<否则