JCredStashTest.java 文件源码

@Test
public void testPutSecretNewVersion() {
    String version = "foover";
    final PutItemRequest[] putItemRequest = new PutItemRequest[1];
    Mockito.when(dynamoDBClient.putItem(Mockito.any(PutItemRequest.class))).thenAnswer(invocationOnMock -> {
        Object[] args = invocationOnMock.getArguments();
        putItemRequest[0] = (PutItemRequest) args[0];
        return new PutItemResult();
    });

    JCredStash credStash = new JCredStash(dynamoDBClient, awskmsClient);
    credStash.putSecret("table", "mysecret", "foo", "alias/foo", new HashMap<>(), version);

    Mockito.verify(dynamoDBClient, VerificationModeFactory.times(1)).putItem(Mockito.any(PutItemRequest.class));
    Assert.assertEquals(putItemRequest[0].getItem().get("version").getS(), version);
}