diff --git a/testing/trino-testing/src/main/java/io/trino/testing/BaseConnectorTest.java b/testing/trino-testing/src/main/java/io/trino/testing/BaseConnectorTest.java index 294ba9508552..280465a85803 100644 --- a/testing/trino-testing/src/main/java/io/trino/testing/BaseConnectorTest.java +++ b/testing/trino-testing/src/main/java/io/trino/testing/BaseConnectorTest.java @@ -164,7 +164,14 @@ public abstract class BaseConnectorTest private final ConcurrentMap>> mockTableListings = new ConcurrentHashMap<>(); @BeforeClass - public void addMockCatalog() + public void initMockCatalog() + { + QueryRunner queryRunner = getQueryRunner(); + queryRunner.installPlugin(buildMockConnectorPlugin()); + queryRunner.createCatalog("mock_dynamic_listing", "mock", Map.of()); + } + + protected MockConnectorPlugin buildMockConnectorPlugin() { MockConnectorFactory connectorFactory = MockConnectorFactory.builder() .withListSchemaNames(session -> ImmutableList.copyOf(mockTableListings.keySet())) @@ -177,9 +184,7 @@ public void addMockCatalog() .apply(session); }) .build(); - QueryRunner queryRunner = getQueryRunner(); - queryRunner.installPlugin(new MockConnectorPlugin(connectorFactory)); - queryRunner.createCatalog("mock_dynamic_listing", "mock", Map.of()); + return new MockConnectorPlugin(connectorFactory); } /**