如何使用Spring从同一个类中调用method1调用的method2?

时间:2018-05-21 23:01:25

标签: java spring-boot mocking automated-tests mockito

我有一个采用这种架构的spring-boot应用程序:

@Controller > @Service > @Component

这是我的组件:

@Component
public class myComponent {

    @Autowired
    private ObjectMapper mapper;
    //com.fasterxml.jackson.databind.ObjectMapper

    @Autowired
    private Component2 component2;
    // some 3rd part REST service

    @Autowired
    private Component3 component3;
    //database access

    @Transactional(propagation = Propagation.REQUIRED)
    public List<SomePOJO> method1(String otherString) {
        String newOne = myString + otherString; //more logic here, but without components
        return this.method2(newOne);
    }

    public List<SomePojo> method2(String newOne){
        String one = component3.methodX(newOne); //access database with component3
        return component2.methodY(one); //more logic here, including component2 and mapper!
    }

}

使用Mockito我在我的测试类中实现了这个:

@MockBean
public myComponent component;

@Before
public void before() throws Exception {
    MockitoAnnotations.initMocks(this);
    List<POJO> someList = new ArrayList<>(); //list populated with some specific POJO's
    Mockito.when(component.method1("specificString")).
        thenReturn(someList);
}

@Test
public void recuperaDados() throws Exception  {
    String response = given().authentication().preemptive().oauth2("loginInfo")
            .get("myComponent_method1_path/" + "specificString1").asString();
    //this specificString1 leads to myComponent.method1("specificString"), satisfying Mockito 
}

有了这个,我成功完成了我的测试。但是当我使用

Mockito.when(component.method2("specificString2")).thenReturn(someList);

然后

String response1 = given().authentication().preemptive().oauth2("loginInfo")
            .get("myComponent_method1_path/" + "specificString1").asString();
// this specificString1 leads to myComponent.method2("specificString2"), also satisfying Mockito
String response2 = component.method2("specificString2");
String response3 = component.method2("randomString");

response1为"",response2为someList正确字符串,response3为null。预计会response2response3。但我希望response1与response2或至少null相同。我怎样才能正确地模拟同一个类中的其他人调用的方法并测试另一个?

修改

我的测试类扩展了这个类:

@ActiveProfiles("test")
@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = WebEnvironment.DEFINED_PORT)
public abstract class AbstractRestTest {

    @Autowired
    private WebApplicationContext wac;

    protected Dsl dsl = Dsl.getInstance();

    @Before
    public void globalSetup() {
        RestAssuredMockMvc.webAppContextSetup(wac);
    }

}

1 个答案:

答案 0 :(得分:1)

Mockito通过代理您希望存根的对象来工作,因此在调用模拟对象上的方法时永远不会执行您的实现代码,除非明确定义这样做。因此,通过在模拟对象上调用method1,将永远不会调用method2作为结果。

然而,有一种方法可以通过部分嘲弄来达到你想要的效果。在您希望测试method2的测试中,您可以指示应该执行真正的method1:

when(component.method1("specificString")).thenCallRealMethod();

应该注意的是,mockito documentation表示使用部分模拟可能表示代码气味。在这个原因中,问题是,method2真的必须公开吗?如果它只是公开的,因为你想嘲笑它的实现,这是一个错误。使用模拟,您只希望存根对象的公共API。任何私有或帮助方法都应被视为内部方法并被忽略。如果你测试那个对象,只测试公共方法,间接测试辅助方法,也是如此。

你是对的,第一个结果是空字符串而不是null是奇数。我认为它可能与代码中没有提供的东西有关,例如控制器或服务。