2014-10-17 77 views
1

我想用Mockito来模拟一些对象,但是我的代码不工作。我的代码包含一个最终的方法sslContext.init(..,..,..),我不知道如何解决这个问题。我怎样才能用mockito来模拟最后的方法

这是我的代码,我想嘲笑:

public class MYClass { 
private SSLSocketFactory sslSocketFactory; 
public MyClass() throws IOException{ 
    String trustStoreFilePath = System.getProperty("javax.net.ssl.trustStore"); 
    try { 
      TrustManagerFactory trustManagerFactory = CertManagerFactory.loadTrustStore(trustStoreFilePath); 
      SSLContext sslContext = SSLContext.getInstance("TLS"); 
      sslContext.init(null, trustManagerFactory.getTrustManagers(), null); 
      this.sslSocketFactory = sslContext.getSocketFactory(); 
     } 
     catch (NoSuchAlgorithmException e) {} 
     catch (KeyManagementException e) {} 
    } 
} 

,我的代码是:


@RunWith(PowerMockRunner.class) 
@PrepareForTest({CertManagerFactory.class, SSLContext.class}) 
public class MyClassTest{ 

@Before 
public void setUp() throws IOException, NoSuchAlgorithmException, KeyManagementException { 

    PowerMockito.mockStatic(CertManagerFactory.class, SSLContext.class);   
    SSLContext sslContextMock = mock(SSLContext.class); 
    SSLSocketFactory sSLSocketFactoryMock = mock(SSLSocketFactory.class); 
    TrustManagerFactory trustManagerFactoryMock = mock(TrustManagerFactory.class); PowerMockito.when(CertManagerFactory.loadTrustStore(anyString())).thenReturn(trustManagerFactoryMock); 
    PowerMockito.when(SSLContext.getInstance(anyString())).thenReturn(sslContextMock); 
    when(trustManagerFactoryMock.getTrustManagers()).thenReturn(any(TrustManager[].class)); 
    when(sslContextMock.getSocketFactory()).thenReturn(sSLSocketFactoryMock); 
    MyClass myClass= new MyClass(); 

} 

+0

是您对Mockito还是PowerMockito的问题? – 2014-10-17 02:37:33

+0

我不回答这个问题,但我的(谷歌带我到这里...) 我使用mockserver。我需要模拟一个SSLContext。 mockserver已经提供了一个SSLFactory,它允许创建一个模拟的SSLContext。 – mcoolive 2016-09-29 09:16:21

回答

2

我认为你需要模拟特定的最终方法。你可以尝试后SSLContext sslContextMock = mock(SSLContext.class);

PowerMockito.doNothing().when(sslContextMock).init(any(KeyManager[].class), any(TrustManager[].class), any(SecureRandom.class)); 

添加此行如果你想模拟KeyManagementException投掷,你可以做如下:

PowerMockito.doThrow(new KeyManagementException()).when(sslContextMock).init(any(KeyManager[].class), any(TrustManager[].class), any(SecureRandom.class)); 

希望它能帮助。

+0

非常感谢。顺便说一下,如何验证异常抛出的输出错误消息。 – houchker 2014-10-19 22:32:21