Examples with ListTablesRequest com.amazonaws.athena.connector.lambda.metadata.ListTablesRequest used on opensource projects

Example 1 with ListTablesRequest

use of com.amazonaws.athena.connector.lambda.metadata.ListTablesRequest in project aws-athena-query-federation by awslabs.

the class CompositeHandlerTest method doListTables.

@Test
public void doListTables() throws Exception {
    ListTablesRequest req = mock(ListTablesRequest.class);
    when(req.getRequestType()).thenReturn(MetadataRequestType.LIST_TABLES);
    compositeHandler.handleRequest(allocator, req, new ByteArrayOutputStream(), objectMapper);
    verify(mockMetadataHandler, times(1)).doListTables(any(BlockAllocatorImpl.class), any(ListTablesRequest.class));
}

Example 2 with ListTablesRequest

use of com.amazonaws.athena.connector.lambda.metadata.ListTablesRequest in project aws-athena-query-federation by awslabs.

the class GlueMetadataHandlerTest method doListTablesWithLargePageSize.

@Test
public void doListTablesWithLargePageSize() throws Exception {
    ListTablesRequest req = new ListTablesRequest(IdentityUtil.fakeIdentity(), queryId, catalog, schema, null, GET_TABLES_REQUEST_MAX_RESULTS + 50);
    logger.info("Request - {}", req);
    ListTablesResponse actualResponse = handler.doListTables(allocator, req);
    logger.info("Response - {}", actualResponse);
    assertEquals("Lists do not match.", fullListResponse, actualResponse);
}

Example 3 with ListTablesRequest

use of com.amazonaws.athena.connector.lambda.metadata.ListTablesRequest in project aws-athena-query-federation by awslabs.

the class GlueMetadataHandlerTest method doListTablesWithPagination.

@Test
public void doListTablesWithPagination() throws Exception {
    logger.info("First paginated request");
    ListTablesRequest req = new ListTablesRequest(IdentityUtil.fakeIdentity(), queryId, catalog, schema, null, 3);
    logger.info("Request - {}", req);
    ListTablesResponse expectedResponse = new ListTablesResponse(req.getCatalogName(), new ImmutableList.Builder<TableName>().add(new TableName(req.getSchemaName(), "table1")).add(new TableName(req.getSchemaName(), "table2")).add(new TableName(req.getSchemaName(), "table3")).build(), "table4");
    ListTablesResponse actualResponse = handler.doListTables(allocator, req);
    logger.info("Response - {}", actualResponse);
    assertEquals("Lists do not match.", expectedResponse, actualResponse);
    logger.info("Second paginated request");
    req = new ListTablesRequest(IdentityUtil.fakeIdentity(), queryId, catalog, schema, actualResponse.getNextToken(), 3);
    logger.info("Request - {}", req);
    expectedResponse = new ListTablesResponse(req.getCatalogName(), new ImmutableList.Builder<TableName>().add(new TableName(req.getSchemaName(), "table4")).add(new TableName(req.getSchemaName(), "table5")).build(), null);
    actualResponse = handler.doListTables(allocator, req);
    logger.info("Response - {}", actualResponse);
    assertEquals("Lists do not match.", expectedResponse, actualResponse);
}

Example 4 with ListTablesRequest

use of com.amazonaws.athena.connector.lambda.metadata.ListTablesRequest in project aws-athena-query-federation by awslabs.

the class GlueMetadataHandlerTest method doListTablesWithUnlimitedPageSize.

@Test
public void doListTablesWithUnlimitedPageSize() throws Exception {
    ListTablesRequest req = new ListTablesRequest(IdentityUtil.fakeIdentity(), queryId, catalog, schema, null, UNLIMITED_PAGE_SIZE_VALUE);
    logger.info("Request - {}", req);
    ListTablesResponse actualResponse = handler.doListTables(allocator, req);
    logger.info("Response - {}", actualResponse);
    assertEquals("Lists do not match.", fullListResponse, actualResponse);
}

Example 5 with ListTablesRequest

use of com.amazonaws.athena.connector.lambda.metadata.ListTablesRequest in project aws-athena-query-federation by awslabs.

the class MySqlMuxJdbcMetadataHandlerTest method doListTables.

@Test
public void doListTables() {
    ListTablesRequest listTablesRequest = Mockito.mock(ListTablesRequest.class);
    Mockito.when(listTablesRequest.getCatalogName()).thenReturn("fakedatabase");
    this.jdbcMetadataHandler.doListTables(this.allocator, listTablesRequest);
    Mockito.verify(this.mySqlMetadataHandler, Mockito.times(1)).doListTables(Mockito.eq(this.allocator), Mockito.eq(listTablesRequest));
}