how to mock keystore.getInstance() which returns the keystore object?

Question

I have tried to mock the KeyStore.getInstance() which returns the keyStore object here the below code which I had tired,

Junit:

KeyStore keyStoreMock = PowerMockito.mock(KeyStore.class); 
PowerMockito.when(KeyStore.getInstance(any())).thenReturn(keyStoreMock); 
PowerMockito.doNothing().when(keyStoreMock).load(any(InputStream.class),Mockito.any(char[].class)); 
X509Certificate cert = Mockito.mock(X509Certificate.class); 
when(keyStoreMock.getCertificate(any())).thenReturn(cert); 
when(cert.getNotAfter()).thenReturn(mockDate);

unable to mock keyStore.getInstance() which returns the of KeyStore class mock object.

Source Code:

KeyStore keystore = KeyStore.getInstance(KeyStore.getDefaultType());
keystore.load(new FileInputStream(filename), password.toCharArray());
X509Certificate certificate = (X509Certificate)keystore.getCertificate("Cert_name");  
endate = certificate.getNotAfter();

I think KeyStore.getInstance() returns null, how to overcome this issue.

Colud any one plz help me out to mock the keystore.getInstance() method.

can be marked as duplicate, this question is already answered here https://stackoverflow.com/questions/36354270/how-to-mock-keystore-class-and-assign-mock-behavior-to-its-methods — Praveen Kumar Mekala, Sep 26 '19 at 08:49
@PraveenKumarMekala actullay i have tried with ref link but i am unable to mock the KeyStore.getInstance() method — user12061625, Sep 26 '19 at 09:00
change @PrepareTest to PrepareForTest and PowerMockito.mock(keystore.class) to PowerMockito.mockStatic(keystore.class) also before PrepareForTestadd RunWith(PowerMockRunner.class) annotation and try — Praveen Kumar Mekala, Sep 26 '19 at 09:05
@PraveenKumarMekala i am updated as per the above update but also i am not able to mock the keystore.getInstance() method which returns keystore object — user12061625, Sep 26 '19 at 11:35
@PraveenKumarMekala thnaks for you update But i am not able to mock the KeyStore.getInstance(). Here the below snnipet i have tried with your updates — user12061625, Sep 27 '19 at 06:47
can you update question with same details and also error trace if you're getting any(I'm assuming keystore.getInstance is returning null) — Praveen Kumar Mekala, Sep 27 '19 at 07:10
@PraveenKumarMekala yes, it is returning null.i think that the reason i am unable to mock Could you pls help me out. — user12061625, Sep 27 '19 at 09:35
try with below solution, if it not working try with any instead of anyString — Praveen Kumar Mekala, Sep 27 '19 at 09:51

Praveen Kumar Mekala · Answer 1 · 2019-09-27T11:17:20.763

-2

@Runwith(PowerMockRunner.class)
@PrepareForTest(KeyStore.class)
public class TestClass {

    @InjectMock
    //Actual Class
    @Mock
    KeyStore mockKeyStore;

    @Test
    public void test() {
        KeyStore keyStoreMock = PowerMockito.mockStatic(KeyStore.class);
        PowerMockito.when(keyStore.getInstance(any())).thenReturn(mockKeyStore);//use this line or
        //use below two lines
        KeyStore powerMockKey = PowerMockito.mock(KeyStore.class);
        Mockito.when(KeyStore.getInstance(anyString())).thenReturn(powerMockKey));        
    }         
}

edited Sep 27 '19 at 11:17

answered Sep 27 '19 at 09:50

Praveen Kumar Mekala

628
1
10
26

1

Tried with both the above Snippet of code but unable to mock the KeyStore.getInstance.it returning original keyStore Object. – user12061625 Sep 27 '19 at 10:58

Ramachandran.A.G · Answer 2 · 2019-09-27T11:05:07.717

-2

Just executed this with power mock, this seems to work as well.

import org.junit.Assert;
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.PowerMockIgnore;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import java.io.InputStream;
import java.security.KeyStore;
import java.security.cert.X509Certificate;
import java.util.Date;

import static org.mockito.Matchers.any;
import static org.powermock.api.mockito.PowerMockito.mock;
import static org.powermock.api.mockito.PowerMockito.when;

@PrepareForTest(KeyStore.class)
@RunWith(PowerMockRunner.class)
@PowerMockIgnore({"javax.management.*", "javax.security.*"})
public class MyKeyTest {
  @Test
  public void mockKeyStore() throws Exception{
    KeyStore keyStoreMock = mock(KeyStore.class);
    PowerMockito.mockStatic(KeyStore.class, invocation -> keyStoreMock);
    when(KeyStore.getInstance(any())).thenReturn(keyStoreMock);
    PowerMockito.doNothing().when(keyStoreMock).load(any(InputStream.class), any(char[].class));
    X509Certificate cert = Mockito.mock(X509Certificate.class);
    when(keyStoreMock.getCertificate(any())).thenReturn(cert);
    when(cert.getNotAfter()).thenReturn(new Date());
    Assert.assertEquals(keyStoreMock,KeyStore.getInstance("x"));
    Assert.assertEquals(keyStoreMock,KeyStore.getInstance("y"));
    Assert.assertEquals(keyStoreMock,KeyStore.getInstance("MarbleCake"));
  }
}

edited Sep 27 '19 at 11:05

answered Sep 27 '19 at 09:57

Ramachandran.A.G

4,788
1
12
24

1

thanks for your update, But it is also not working.It returns the original keyStore object – user12061625 Sep 27 '19 at 10:53
did not understand that part , what does original key store mean – Ramachandran.A.G Sep 27 '19 at 10:56
1

i mean keyStore.getInstance is not returning keyStoreMock – user12061625 Sep 27 '19 at 11:00
can you show me how that was determined , just so that i can look at it ? – Ramachandran.A.G Sep 27 '19 at 11:02
Just so that i was sure , i added assertions. They return the mock dont they ?. They pass for me! – Ramachandran.A.G Sep 27 '19 at 11:05
Could you pls check out my updated post with Source code – user12061625 Sep 27 '19 at 12:24
When i was debugging it is not returning keyStoreMock for KeyStore.getInstance() , but Finally i want to get the mock endDate of a certificate – user12061625 Sep 27 '19 at 12:44
did you find the solution for mocking KeyStore? I'm facing the same problem with KeyFactory.class – Vinit Pillai May 12 '21 at 06:20

how to mock keystore.getInstance() which returns the keystore object?

2 Answers2