diff --git a/airbyte-integrations/bases/standard-source-test/src/main/java/io/airbyte/integrations/standardtest/source/AbstractSourceConnectorTest.java b/airbyte-integrations/bases/standard-source-test/src/main/java/io/airbyte/integrations/standardtest/source/AbstractSourceConnectorTest.java index 488bc46deb3755..26d786fa784e09 100644 --- a/airbyte-integrations/bases/standard-source-test/src/main/java/io/airbyte/integrations/standardtest/source/AbstractSourceConnectorTest.java +++ b/airbyte-integrations/bases/standard-source-test/src/main/java/io/airbyte/integrations/standardtest/source/AbstractSourceConnectorTest.java @@ -66,6 +66,8 @@ public abstract class AbstractSourceConnectorTest { private static final String JOB_ID = String.valueOf(0L); private static final int JOB_ATTEMPT = 0; + private static final UUID SOURCE_ID = UUID.randomUUID(); + private static final String CPU_REQUEST_FIELD_NAME = "cpuRequest"; private static final String CPU_LIMIT_FIELD_NAME = "cpuLimit"; private static final String MEMORY_REQUEST_FIELD_NAME = "memoryRequest"; @@ -162,7 +164,8 @@ protected UUID runDiscover() throws Exception { final UUID toReturn = new DefaultDiscoverCatalogWorker( mConfigRepository, new AirbyteIntegrationLauncher(JOB_ID, JOB_ATTEMPT, getImageName(), processFactory, workerConfigs.getResourceRequirements())) - .run(new StandardDiscoverCatalogInput().withConnectionConfiguration(getConfig()), jobRoot).getDiscoverCatalogId(); + .run(new StandardDiscoverCatalogInput().withSourceId(SOURCE_ID.toString()).withConnectionConfiguration(getConfig()), jobRoot) + .getDiscoverCatalogId(); verify(mConfigRepository).writeActorCatalogFetchEvent(lastPersistedCatalog.capture(), any(), any(), any()); return toReturn; } diff --git a/airbyte-workers/src/main/java/io/airbyte/workers/general/DefaultDiscoverCatalogWorker.java b/airbyte-workers/src/main/java/io/airbyte/workers/general/DefaultDiscoverCatalogWorker.java index 4c1eb16ab73964..209fdbcdb55905 100644 --- a/airbyte-workers/src/main/java/io/airbyte/workers/general/DefaultDiscoverCatalogWorker.java +++ b/airbyte-workers/src/main/java/io/airbyte/workers/general/DefaultDiscoverCatalogWorker.java @@ -87,7 +87,10 @@ public ConnectorJobOutput run(final StandardDiscoverCatalogInput discoverSchemaI } final UUID catalogId = - configRepository.writeActorCatalogFetchEvent(catalog.get(), UUID.fromString(discoverSchemaInput.getSourceId()), + configRepository.writeActorCatalogFetchEvent(catalog.get(), + // NOTE: sourceId is marked required in the OpenAPI config but the code generator doesn't enforce + // it, so we check again here. + discoverSchemaInput.getSourceId() == null ? null : UUID.fromString(discoverSchemaInput.getSourceId()), discoverSchemaInput.getConnectorVersion(), discoverSchemaInput.getConfigHash()); return new ConnectorJobOutput().withOutputType(OutputType.DISCOVER_CATALOG_ID).withDiscoverCatalogId(catalogId);