Skip to content

Commit 39e375b

Browse files
authored
Fix connection read object building (#13568)
Make sure the sourceCatalogId field is set with the value stored in the database.
1 parent 5e1f968 commit 39e375b

File tree

4 files changed

+17
-8
lines changed

4 files changed

+17
-8
lines changed

airbyte-server/src/main/java/io/airbyte/server/converters/ApiPojoConverters.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,8 @@ public static ConnectionRead internalToConnectionRead(final StandardSync standar
130130
.namespaceDefinition(Enums.convertTo(standardSync.getNamespaceDefinition(), io.airbyte.api.model.generated.NamespaceDefinitionType.class))
131131
.namespaceFormat(standardSync.getNamespaceFormat())
132132
.prefix(standardSync.getPrefix())
133-
.syncCatalog(CatalogConverter.toApi(standardSync.getCatalog()));
133+
.syncCatalog(CatalogConverter.toApi(standardSync.getCatalog()))
134+
.sourceCatalogId(standardSync.getSourceCatalogId());
134135

135136
if (standardSync.getResourceRequirements() != null) {
136137
connectionRead.resourceRequirements(resourceRequirementsToApi(standardSync.getResourceRequirements()));

airbyte-server/src/main/java/io/airbyte/server/handlers/helpers/ConnectionMatcher.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ public ConnectionRead match(final ConnectionRead query) {
4040
fromSearch.resourceRequirements(query.getResourceRequirements());
4141
fromSearch.syncCatalog(query.getSyncCatalog());
4242
fromSearch.operationIds(query.getOperationIds());
43+
fromSearch.sourceCatalogId(query.getSourceCatalogId());
4344

4445
return fromSearch;
4546
}

airbyte-server/src/test/java/io/airbyte/server/handlers/ConnectionsHandlerTest.java

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,8 @@ void setUp() throws IOException, JsonValidationException, ConfigNotFoundExceptio
115115
.withOperationIds(List.of(operationId))
116116
.withManual(false)
117117
.withSchedule(ConnectionHelpers.generateBasicSchedule())
118-
.withResourceRequirements(ConnectionHelpers.TESTING_RESOURCE_REQUIREMENTS);
118+
.withResourceRequirements(ConnectionHelpers.TESTING_RESOURCE_REQUIREMENTS)
119+
.withSourceCatalogId(UUID.randomUUID());
119120
standardSyncDeleted = new StandardSync()
120121
.withConnectionId(connectionId)
121122
.withName("presto to hudi2")
@@ -194,7 +195,8 @@ void testCreateConnection() throws JsonValidationException, ConfigNotFoundExcept
194195
.cpuRequest(standardSync.getResourceRequirements().getCpuRequest())
195196
.cpuLimit(standardSync.getResourceRequirements().getCpuLimit())
196197
.memoryRequest(standardSync.getResourceRequirements().getMemoryRequest())
197-
.memoryLimit(standardSync.getResourceRequirements().getMemoryLimit()));
198+
.memoryLimit(standardSync.getResourceRequirements().getMemoryLimit()))
199+
.sourceCatalogId(standardSync.getSourceCatalogId());
198200

199201
final ConnectionRead actualConnectionRead = connectionsHandler.createConnection(connectionCreate);
200202

@@ -343,7 +345,8 @@ void testUpdateConnection() throws JsonValidationException, ConfigNotFoundExcept
343345
standardSync.getConnectionId(),
344346
standardSync.getSourceId(),
345347
standardSync.getDestinationId(),
346-
standardSync.getOperationIds())
348+
standardSync.getOperationIds(),
349+
newSourceCatalogId)
347350
.schedule(null)
348351
.syncCatalog(catalog)
349352
.status(ConnectionStatus.INACTIVE);

airbyte-server/src/test/java/io/airbyte/server/helpers/ConnectionHelpers.java

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,8 @@ public static Schedule generateBasicSchedule() {
9999
public static ConnectionRead generateExpectedConnectionRead(final UUID connectionId,
100100
final UUID sourceId,
101101
final UUID destinationId,
102-
final List<UUID> operationIds) {
102+
final List<UUID> operationIds,
103+
final UUID sourceCatalogId) {
103104

104105
return new ConnectionRead()
105106
.connectionId(connectionId)
@@ -117,15 +118,17 @@ public static ConnectionRead generateExpectedConnectionRead(final UUID connectio
117118
.cpuRequest(TESTING_RESOURCE_REQUIREMENTS.getCpuRequest())
118119
.cpuLimit(TESTING_RESOURCE_REQUIREMENTS.getCpuLimit())
119120
.memoryRequest(TESTING_RESOURCE_REQUIREMENTS.getMemoryRequest())
120-
.memoryLimit(TESTING_RESOURCE_REQUIREMENTS.getMemoryLimit()));
121+
.memoryLimit(TESTING_RESOURCE_REQUIREMENTS.getMemoryLimit()))
122+
.sourceCatalogId(sourceCatalogId);
121123
}
122124

123125
public static ConnectionRead generateExpectedConnectionRead(final StandardSync standardSync) {
124126
final ConnectionRead connectionRead = generateExpectedConnectionRead(
125127
standardSync.getConnectionId(),
126128
standardSync.getSourceId(),
127129
standardSync.getDestinationId(),
128-
standardSync.getOperationIds());
130+
standardSync.getOperationIds(),
131+
standardSync.getSourceCatalogId());
129132

130133
if (standardSync.getSchedule() == null) {
131134
connectionRead.schedule(null);
@@ -147,7 +150,8 @@ public static ConnectionRead connectionReadFromStandardSync(final StandardSync s
147150
.operationIds(standardSync.getOperationIds())
148151
.name(standardSync.getName())
149152
.namespaceFormat(standardSync.getNamespaceFormat())
150-
.prefix(standardSync.getPrefix());
153+
.prefix(standardSync.getPrefix())
154+
.sourceCatalogId(standardSync.getSourceCatalogId());
151155

152156
if (standardSync.getNamespaceDefinition() != null) {
153157
connectionRead

0 commit comments

Comments
 (0)