Code: Select all
PageIterable pages = table.query(queryEnhancedRequest);
for(Page page : pages) {
if(page.items() != null) {
tableItems.addAll(page.items());
}
}
Code: Select all
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
import software.amazon.awssdk.enhanced.dynamodb.DynamoDBTable;
import software.amazon.awssdk.enhanced.dynamodb.Page;
import software.amazon.awssdk.enhanced.dynamodb.PageIterable;
import software.amazon.awssdk.enhanced.dynamodb.QueryEnhancedRequest;
@RunWith(MockitoJUnitRunner.class)
public class AccountRepositoryTest {
@Mock
private DynamoDbTable myTable;
@Mock
private PageIterable pages;
@Mock
private Iterator iterator;
@Mock
private Page page;
@Test
public void testGetAccountData_Success() {
when(myTable.query(queryEnhancedRequest)).thenReturn(pages);
when(pages.iterator()).thenReturn(iterator);
when(iterator.hasNext()).thenReturn(true, false);
when(iterator.next()).thenReturn(page);
...