如何使用Mockito模拟DynamoDB映射器加载函数？

import org.mockito.Mockito;
import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapper;

DynamoDBMapper dynamoDBMapperMock = Mockito.mock(DynamoDBMapper.class);

// 定义模拟的返回值
YourEntity yourEntity = new YourEntity();
yourEntity.setId("yourId");
yourEntity.setName("yourName");

// 定义模拟的行为
Mockito.when(dynamoDBMapperMock.load(YourEntity.class, "yourId")).thenReturn(yourEntity);

YourEntity loadedEntity = dynamoDBMapperMock.load(YourEntity.class, "yourId");