Мощный частный метод с использованием PowerMockito

Question

Я использую PowerMockito для издевательства частного вызова метода (privateApi), но он по-прежнему делает вызов privateApi, который, в свою очередь, делает другой thirdPartCall. Я попадаю в проблему, когда thirdPartyCall выдает исключение. Насколько я понимаю, если я издеваюсь над privateApi, он не должен разбираться в деталях реализации метода и возвращать макет ответа.

public class MyClient { 

    public void publicApi() { 
     System.out.println("In publicApi"); 
     int result = 0; 
     try { 
      result = privateApi("hello", 1); 
     } catch (Exception e) { 
      Assert.fail(); 
     } 
     System.out.println("result : "+result); 
     if (result == 20) { 
      throw new RuntimeException("boom"); 
     } 
    } 

    private int privateApi(String whatever, int num) throws Exception { 
     System.out.println("In privateAPI"); 
     thirdPartyCall(); 
     int resp = 10; 
     return resp; 
    } 

    private void thirdPartyCall() throws Exception{ 
     System.out.println("In thirdPartyCall"); 
     //Actual WS call which may be down while running the test cases 
    } 
} 

Вот тестовый пример:

@RunWith(PowerMockRunner.class) 
@PrepareForTest(MyClient.class) 
public class MyclientTest { 

    @Test(expected = RuntimeException.class) 
    public void testPublicAPI() throws Exception { 
     MyClient classUnderTest = PowerMockito.spy(new MyClient()); 
     PowerMockito.when(classUnderTest, "privateApi", anyString(), anyInt()).thenReturn(20); 
     classUnderTest.publicApi(); 
    } 
} 

консоли трассировки:

In privateAPI 
In thirdPartyCall 
In publicApi 
result : 20 

Answer 1

Вам просто нужно изменить макет вызова метода использовать doReturn.

Example Partial Mocking of Private Method

Код испытания

@RunWith(PowerMockRunner.class) 
@PrepareForTest(MyClient.class) 
public class MyClientTest { 

    @Test(expected = RuntimeException.class) 
    public void testPublicAPI() throws Exception { 
     MyClient classUnderTest = PowerMockito.spy(new MyClient()); 

     // Change to this 

     PowerMockito.doReturn(20).when(classUnderTest, "privateApi", anyString(), anyInt()); 

     classUnderTest.publicApi(); 
    } 
} 

консоли след

In publicApi 
result : 20