### Question:
AwsCmdbMetadataHandler extends MetadataHandler { @Override public void getPartitions(BlockWriter blockWriter, GetTableLayoutRequest request, QueryStatusChecker queryStatusChecker) throws Exception { TableProvider tableProvider = tableProviders.get(request.getTableName()); if (tableProvider == null) { throw new RuntimeException("Unknown table " + request.getTableName()); } tableProvider.getPartitions(blockWriter, request); } AwsCmdbMetadataHandler(); @VisibleForTesting protected  AwsCmdbMetadataHandler(TableProviderFactory tableProviderFactory,
            EncryptionKeyFactory keyFactory,
            AWSSecretsManager secretsManager,
            AmazonAthena athena,
            String spillBucket,
            String spillPrefix); @Override ListSchemasResponse doListSchemaNames(BlockAllocator blockAllocator, ListSchemasRequest listSchemasRequest); @Override ListTablesResponse doListTables(BlockAllocator blockAllocator, ListTablesRequest listTablesRequest); @Override GetTableResponse doGetTable(BlockAllocator blockAllocator, GetTableRequest getTableRequest); @Override void enhancePartitionSchema(SchemaBuilder partitionSchemaBuilder, GetTableLayoutRequest request); @Override void getPartitions(BlockWriter blockWriter, GetTableLayoutRequest request, QueryStatusChecker queryStatusChecker); @Override GetSplitsResponse doGetSplits(BlockAllocator blockAllocator, GetSplitsRequest getSplitsRequest);  }### Answer:
@Test public void doGetTableLayout() throws Exception { GetTableLayoutRequest request = new GetTableLayoutRequest(identity, queryId, catalog, new TableName("schema1", "table1"), mockConstraints, SchemaBuilder.newBuilder().build(), Collections.EMPTY_SET); GetTableLayoutResponse response = handler.doGetTableLayout(blockAllocator, request); assertNotNull(response); assertEquals(1, response.getPartitions().getRowCount()); }