should have workspaceid for update connection (#22512)

* should have workspaceid for update connection * add test
airbytehq · Feb 7, 2023 · 6f0c0be · 6f0c0be
1 parent 85a6026
commit 6f0c0be
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/...yte-commons-server/src/main/java/io/airbyte/commons/server/handlers/SchedulerHandler.java b/...yte-commons-server/src/main/java/io/airbyte/commons/server/handlers/SchedulerHandler.java
@@ -192,7 +192,8 @@ public CheckConnectionRead checkSourceConnectionFromSourceIdForUpdate(final Sour
     final SourceCoreConfig sourceCoreConfig = new SourceCoreConfig()
         .sourceId(updatedSource.getSourceId())
         .connectionConfiguration(updatedSource.getConfiguration())
-        .sourceDefinitionId(updatedSource.getSourceDefinitionId());
+        .sourceDefinitionId(updatedSource.getSourceDefinitionId())
+        .workspaceId(updatedSource.getWorkspaceId());
 
     return checkSourceConnectionFromSourceCreate(sourceCoreConfig);
   }

diff --git a/...commons-server/src/test/java/io/airbyte/commons/server/handlers/SchedulerHandlerTest.java b/...commons-server/src/test/java/io/airbyte/commons/server/handlers/SchedulerHandlerTest.java
@@ -265,7 +265,8 @@ void testCheckSourceConnectionFromUpdate() throws IOException, JsonValidationExc
     final SourceConnection submittedSource = new SourceConnection()
         .withSourceId(source.getSourceId())
         .withSourceDefinitionId(source.getSourceDefinitionId())
-        .withConfiguration(source.getConfiguration());
+        .withConfiguration(source.getConfiguration())
+        .withWorkspaceId(source.getWorkspaceId());
     when(synchronousSchedulerClient.createSourceCheckConnectionJob(submittedSource, DESTINATION_DOCKER_IMAGE, protocolVersion, false))
         .thenReturn((SynchronousResponse<StandardCheckConnectionOutput>) jobResponse);
     when(secretsRepositoryWriter.statefulSplitEphemeralSecrets(