Examples with FakeItem.createUniqueFakeItem software.amazon.awssdk.enhanced.dynamodb.functionaltests.models.FakeItem.createUniqueFakeItem used on opensource projects

Example 1 with FakeItem.createUniqueFakeItem

use of software.amazon.awssdk.enhanced.dynamodb.functionaltests.models.FakeItem.createUniqueFakeItem in project aws-sdk-java-v2 by aws.

the class UpdateItemOperationTest method transformResponse_mapsAttributesReturnedInResponse.

@Test
public void transformResponse_mapsAttributesReturnedInResponse() {
    FakeItem fakeItem1 = FakeItem.createUniqueFakeItem();
    FakeItem fakeItem2 = FakeItem.createUniqueFakeItem();
    Map<String, AttributeValue> fakeItem2Attributes = FakeItem.getTableSchema().itemToMap(fakeItem2, true);
    UpdateItemOperation<FakeItem> updateItemOperation = UpdateItemOperation.create(UpdateItemEnhancedRequest.builder(FakeItem.class).item(fakeItem1).build());
    FakeItem result = updateItemOperation.transformResponse(UpdateItemResponse.builder().attributes(fakeItem2Attributes).build(), FakeItem.getTableSchema(), PRIMARY_CONTEXT, null).attributes();
    assertThat(result, is(fakeItem2));
}

Example 2 with FakeItem.createUniqueFakeItem

use of software.amazon.awssdk.enhanced.dynamodb.functionaltests.models.FakeItem.createUniqueFakeItem in project aws-sdk-java-v2 by aws.

the class UpdateItemOperationTest method transformResponse_afterReadThrowsException_throwsIllegalStateException.

@Test(expected = IllegalStateException.class)
public void transformResponse_afterReadThrowsException_throwsIllegalStateException() {
    when(mockDynamoDbEnhancedClientExtension.afterRead(any(DynamoDbExtensionContext.AfterRead.class))).thenThrow(RuntimeException.class);
    UpdateItemOperation<FakeItem> updateItemOperation = UpdateItemOperation.create(UpdateItemEnhancedRequest.builder(FakeItem.class).item(createUniqueFakeItem()).build());
    UpdateItemResponse response = UpdateItemResponse.builder().attributes(FakeItem.getTableSchema().itemToMap(FakeItem.createUniqueFakeItem(), true)).build();
    updateItemOperation.transformResponse(response, FakeItem.getTableSchema(), PRIMARY_CONTEXT, mockDynamoDbEnhancedClientExtension);
}