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

Example 6 with ListTablesRequest

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

the class JdbcMetadataHandlerTest method doListTablesSQLException.

@Test(expected = RuntimeException.class)
public void doListTablesSQLException() throws SQLException {
    Mockito.when(this.connection.getMetaData().getTables(Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.any())).thenThrow(new SQLException());
    this.jdbcMetadataHandler.doListTables(this.blockAllocator, new ListTablesRequest(this.federatedIdentity, "testQueryId", "testCatalog", "testSchema", null, UNLIMITED_PAGE_SIZE_VALUE));
}

Example 7 with ListTablesRequest

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

the class JdbcMetadataHandlerTest method doListTables.

@Test
public void doListTables() throws SQLException {
    String[] schema = { "TABLE_SCHEM", "TABLE_NAME" };
    Object[][] values = { { "testSchema", "testTable" }, { "testSchema", "testtable2" } };
    TableName[] expected = { new TableName("testSchema", "testTable"), new TableName("testSchema", "testtable2") };
    AtomicInteger rowNumber = new AtomicInteger(-1);
    ResultSet resultSet = mockResultSet(schema, values, rowNumber);
    Mockito.when(connection.getMetaData().getTables("testCatalog", "testSchema", null, new String[] { "TABLE", "VIEW", "EXTERNAL TABLE" })).thenReturn(resultSet);
    Mockito.when(connection.getCatalog()).thenReturn("testCatalog");
    ListTablesResponse listTablesResponse = this.jdbcMetadataHandler.doListTables(this.blockAllocator, new ListTablesRequest(this.federatedIdentity, "testQueryId", "testCatalog", "testSchema", null, UNLIMITED_PAGE_SIZE_VALUE));
    Assert.assertArrayEquals(expected, listTablesResponse.getTables().toArray());
}

Example 8 with ListTablesRequest

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

the class MultiplexingJdbcMetadataHandlerTest 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.fakeDatabaseHandler, Mockito.times(1)).doListTables(Mockito.eq(this.allocator), Mockito.eq(listTablesRequest));
}

Example 9 with ListTablesRequest

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

the class PostGreSqlMuxJdbcMetadataHandlerTest method doListTables.

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

Example 10 with ListTablesRequest

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

the class NeptuneMetadataHandlerTest method doListTables.

@Test
public void doListTables() {
    logger.info("doListTables - enter");
    List<Table> tables = new ArrayList<Table>();
    Table table1 = new Table();
    table1.setName("table1");
    Table table2 = new Table();
    table2.setName("table2");
    Table table3 = new Table();
    table3.setName("table3");
    tables.add(table1);
    tables.add(table2);
    tables.add(table3);
    GetTablesResult tableResult = new GetTablesResult();
    tableResult.setTableList(tables);
    ListTablesRequest req = new ListTablesRequest(IDENTITY, "queryId", "default", "default", null, UNLIMITED_PAGE_SIZE_VALUE);
    when(glue.getTables(any(GetTablesRequest.class))).thenReturn(tableResult);
    ListTablesResponse res = handler.doListTables(allocator, req);
    logger.info("doListTables - {}", res.getTables());
    assertFalse(res.getTables().isEmpty());
    logger.info("doListTables - exit");
}