fix: test coverage (#3)

Co-authored-by: milan.horvath <[email protected]>
onecx · Apr 3, 2024 · c3c53d7 · c3c53d7
1 parent 2fde360
commit c3c53d7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/main/java/org/tkit/onecx/shell/bff/rs/controllers/WorkspaceConfigRestController.java b/src/main/java/org/tkit/onecx/shell/bff/rs/controllers/WorkspaceConfigRestController.java
@@ -61,6 +61,7 @@ public Response getWorkspaceConfig(GetWorkspaceConfigRequestDTO getWorkspaceConf
 
         //get base workspace info
         try (Response response = workspaceClient.searchWorkspaces(mapper.map(getWorkspaceConfigRequestDTO))) {
+            Response.ResponseBuilder responseBuilder = null;
             if (!response.readEntity(WorkspacePageResult.class).getStream().isEmpty()) {
                 var workspaceInfo = response.readEntity(WorkspacePageResult.class).getStream().get(0);
                 responseDTO.setWorkspace(mapper.map(workspaceInfo, getWorkspaceConfigRequestDTO));
@@ -90,11 +91,12 @@ public Response getWorkspaceConfig(GetWorkspaceConfigRequestDTO getWorkspaceConf
                 }
                 //call remoteComponent Mocks => should be removed after implementation
                 responseDTO = mockRemoteComponents(responseDTO);
-                return Response.status(Response.Status.OK).entity(responseDTO).build();
+                responseBuilder = Response.status(Response.Status.OK).entity(responseDTO);
             } else {
-                return Response.status(Response.Status.NOT_FOUND.getStatusCode(),
-                        "No workspace with matching url found").build();
+                responseBuilder = Response.status(Response.Status.NOT_FOUND.getStatusCode(),
+                        "No workspace with matching url found");
             }
+            return responseBuilder.build();
         }
     }