Fix up

Signed-off-by: BoykoAlex <alex.boyko@broadcom.com>

Author: BoykoAlex

headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/java/data/test/DataRepositoryAotMetadataCodeLensProviderJdbcTest.java

@@ -245,11 +245,11 @@ void turnIntoQueryUsesCompactStyleByDefault() throws Exception {
 	private String extractValueFromAttributes(CodeLens codeLens) {
 		Object args = codeLens.getCommand().getArguments().get(1);
 		if (args instanceof JsonObject) {
-			JsonObject params = (JsonObject) args;
-			if (params.has("parameters") && params.get("parameters").isJsonObject()) {
-				JsonObject parameters = params.getAsJsonObject("parameters");
-				if (parameters.has("attributes") && parameters.get("attributes").isJsonArray()) {
-					JsonArray attributes = parameters.getAsJsonArray("attributes");
+			JsonObject root = (JsonObject) args;
+			if (root.has("refactoring") && root.get("refactoring").isJsonObject()) {
+				JsonObject refactoring = root.getAsJsonObject("refactoring");
+				if (refactoring.has("attributes") && refactoring.get("attributes").isJsonArray()) {
+					JsonArray attributes = refactoring.getAsJsonArray("attributes");
 					for (JsonElement element : attributes) {
 						if (element.isJsonObject()) {
 							JsonObject attr = element.getAsJsonObject();

headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/java/data/test/QueryMethodCodeActionProviderJpaTest.java

@@ -84,7 +84,7 @@ void convertToQueryCodeAction() throws Exception {
 		Command cmd = ca.getCommand();
 		assertEquals(JdtRefactorings.JDT_QUICKFIX, cmd.getArguments().get(0));
 		QuickfixEdit qfEdit = jdtRefactorings.createEdits((JsonElement) cmd.getArguments().get(1)).block();
-		WorkspaceEdit edit = qfEdit.getWorkspaceEdit();
+		WorkspaceEdit edit = qfEdit.workspaceEdit;
 		TextDocumentEdit docEdit = edit.getDocumentChanges().get(0).getLeft();
 		String rawText = docEdit.getEdits().get(0).getLeft().getNewText();
 

headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/java/data/test/QueryMethodCodeActionProviderMongoDbTest.java

@@ -84,7 +84,7 @@ void convertToQueryCodeAction() throws Exception {
 		Command cmd = ca.getCommand();
 		assertEquals(JdtRefactorings.JDT_QUICKFIX, cmd.getArguments().get(0));
 		QuickfixEdit qfEdit = jdtRefactorings.createEdits((JsonElement) cmd.getArguments().get(1)).block();
-		WorkspaceEdit edit = qfEdit.getWorkspaceEdit();
+		WorkspaceEdit edit = qfEdit.workspaceEdit;
 		TextDocumentEdit docEdit = edit.getDocumentChanges().get(0).getLeft();
 		String rawText = docEdit.getEdits().get(0).getLeft().getNewText().trim();