From 4dccb9d33119402aa9f07411ccf359c11da4c976 Mon Sep 17 00:00:00 2001
From: Simon Verhoeven <verhoeven.simon@gmail.com>
Date: Wed, 6 Dec 2023 17:24:30 +0100
Subject: [PATCH] fix: fix issues related to new addProperty signature (#457)

---
 .../java/org/openrewrite/java/spring/AddSpringProperty.java   | 2 +-
 .../java/spring/boot2/MigrateDatabaseCredentials.java         | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/main/java/org/openrewrite/java/spring/AddSpringProperty.java b/src/main/java/org/openrewrite/java/spring/AddSpringProperty.java
index fc55c7ae8..c74432eba 100644
--- a/src/main/java/org/openrewrite/java/spring/AddSpringProperty.java
+++ b/src/main/java/org/openrewrite/java/spring/AddSpringProperty.java
@@ -91,7 +91,7 @@ public boolean isAcceptable(SourceFile sourceFile, ExecutionContext ctx) {
                 if (t instanceof Yaml.Documents && sourcePathMatches(((SourceFile) t).getSourcePath(), ctx)) {
                     t = createMergeYamlVisitor().getVisitor().visit(t, ctx);
                 } else if (t instanceof Properties.File && sourcePathMatches(((SourceFile) t).getSourcePath(), ctx)) {
-                    t = new AddProperty(property, value, null).getVisitor().visit(t, ctx);
+                    t = new AddProperty(property, value, null, null).getVisitor().visit(t, ctx);
                 }
                 return t;
             }
diff --git a/src/main/java/org/openrewrite/java/spring/boot2/MigrateDatabaseCredentials.java b/src/main/java/org/openrewrite/java/spring/boot2/MigrateDatabaseCredentials.java
index f7a562046..43615d3a4 100644
--- a/src/main/java/org/openrewrite/java/spring/boot2/MigrateDatabaseCredentials.java
+++ b/src/main/java/org/openrewrite/java/spring/boot2/MigrateDatabaseCredentials.java
@@ -129,8 +129,8 @@ public Properties visitFile(Properties.File file, ExecutionContext ctx) {
             }, new PropertiesVisitor<ExecutionContext>() {
                 @Override
                 public Properties visitFile(Properties.File file, ExecutionContext ctx) {
-                    doAfterVisit(new AddProperty("spring." + tool + ".username", "${spring.datasource.username}", null).getVisitor());
-                    doAfterVisit(new AddProperty("spring." + tool + ".password", "${spring.datasource.password}", null).getVisitor());
+                    doAfterVisit(new AddProperty("spring." + tool + ".username", "${spring.datasource.username}", null, null).getVisitor());
+                    doAfterVisit(new AddProperty("spring." + tool + ".password", "${spring.datasource.password}", null, null).getVisitor());
                     return file;
                 }
             });