JCredStashTest.java 文件源码

@Test
public void testGetSecret() {
    final QueryRequest[] queryRequest = new QueryRequest[1];
    Mockito.when(dynamoDBClient.query(Mockito.any(QueryRequest.class))).thenAnswer(invocationOnMock -> {
        Object[] args = invocationOnMock.getArguments();
        queryRequest[0] = (QueryRequest) args[0];
        return new QueryResult().withCount(1).withItems(Arrays.asList(
                mockItem("mysecret", padVersion(1), new byte[]{}, new byte[]{}, new byte[]{})
        ));
    });


    JCredStash credStash = Mockito.spy(new JCredStash(dynamoDBClient, awskmsClient));

    Mockito.doReturn("foo").when(credStash).getSecret(Mockito.any(JCredStash.StoredSecret.class), Mockito.any(Map.class));

    String secret = credStash.getSecret("table", "mysecret", new HashMap<>());

    Mockito.verify(dynamoDBClient, VerificationModeFactory.times(1)).query(Mockito.any(QueryRequest.class));
    Assert.assertEquals("foo", secret);
}