SONAR-19839 branch renaming gets propagated to new code periods branch references

2 files changed, 57 insertions, 0 deletions

diff --git a/server/sonar-db-dao/src/it/java/org/sonar/db/newcodeperiod/NewCodePeriodDaoIT.java b/server/sonar-db-dao/src/it/java/org/sonar/db/newcodeperiod/NewCodePeriodDaoIT.java
index 06777152e46..01c710e2c57 100644
--- a/server/sonar-db-dao/src/it/java/org/sonar/db/newcodeperiod/NewCodePeriodDaoIT.java
+++ b/server/sonar-db-dao/src/it/java/org/sonar/db/newcodeperiod/NewCodePeriodDaoIT.java
@@ -29,6 +29,7 @@ import org.sonar.core.util.UuidFactory;
 import org.sonar.db.DbSession;
 import org.sonar.db.DbTester;
 import org.sonar.db.component.BranchDto;
+import org.sonar.db.component.BranchType;
 import org.sonar.db.component.ProjectData;
 import org.sonar.db.project.ProjectDto;
 
@@ -119,6 +120,52 @@ public class NewCodePeriodDaoIT {
   }
 
   @Test
+  public void update_referenceBranchPeriod_value() {
+    insert("1", "proj-uuid", null, REFERENCE_BRANCH, "oldBranchName", null);
+    insert("2", "proj-uuid-2", null, REFERENCE_BRANCH, "anotherBranch", null);
+
+    underTest.updateBranchReferenceValues(dbSession, new BranchDto()
+      .setBranchType(BranchType.BRANCH)
+      .setUuid("branch-uuid")
+      .setProjectUuid("proj-uuid")
+      .setKey("oldBranchName"), "newBranchName");
+
+    //second project reference branch renaming should not affect the reference branch
+    underTest.updateBranchReferenceValues(dbSession, new BranchDto()
+      .setBranchType(BranchType.BRANCH)
+      .setUuid("branch-uuid")
+      .setProjectUuid("proj-uuid-2")
+      .setKey("oldBranchName"), "newBranchName");
+
+    db.commit();
+
+    Optional<NewCodePeriodDto> updatedNewCodePeriod = underTest.selectByUuid(dbSession, "1");
+    Optional<NewCodePeriodDto> unmodifiedNewCodePeriod = underTest.selectByUuid(dbSession, "2");
+
+    assertThat(updatedNewCodePeriod).isNotNull()
+      .isNotEmpty();
+
+    assertThat(unmodifiedNewCodePeriod).isNotNull()
+      .isNotEmpty();
+
+    assertNewCodePeriodRowCount(2);
+
+    NewCodePeriodDto updatedResult = updatedNewCodePeriod.get();
+    assertThat(updatedResult.getUuid()).isEqualTo("1");
+    assertThat(updatedResult.getProjectUuid()).isEqualTo("proj-uuid");
+    assertThat(updatedResult.getBranchUuid()).isNull();
+    assertThat(updatedResult.getType()).isEqualTo(REFERENCE_BRANCH);
+    assertThat(updatedResult.getValue()).isEqualTo("newBranchName");
+
+    NewCodePeriodDto unmodifiedResult = unmodifiedNewCodePeriod.get();
+    assertThat(unmodifiedResult.getUuid()).isEqualTo("2");
+    assertThat(unmodifiedResult.getProjectUuid()).isEqualTo("proj-uuid-2");
+    assertThat(unmodifiedResult.getBranchUuid()).isNull();
+    assertThat(unmodifiedResult.getType()).isEqualTo(REFERENCE_BRANCH);
+    assertThat(unmodifiedResult.getValue()).isEqualTo("anotherBranch");
+  }
+
+  @Test
   public void insert_with_upsert() {
     insert("1", "proj-uuid", "branch-uuid", NUMBER_OF_DAYS, "5", null);
 
diff --git a/server/sonar-db-dao/src/main/java/org/sonar/db/newcodeperiod/NewCodePeriodDao.java b/server/sonar-db-dao/src/main/java/org/sonar/db/newcodeperiod/NewCodePeriodDao.java
index f3d680ed095..316d7fd7bed 100644
--- a/server/sonar-db-dao/src/main/java/org/sonar/db/newcodeperiod/NewCodePeriodDao.java
+++ b/server/sonar-db-dao/src/main/java/org/sonar/db/newcodeperiod/NewCodePeriodDao.java
@@ -27,6 +27,7 @@ import org.sonar.api.utils.System2;
 import org.sonar.core.util.UuidFactory;
 import org.sonar.db.Dao;
 import org.sonar.db.DbSession;
+import org.sonar.db.component.BranchDto;
 
 import static java.util.Objects.requireNonNull;
 import static java.util.Optional.ofNullable;
@@ -74,6 +75,15 @@ public class NewCodePeriodDao implements Dao {
     mapper(dbSession).update(dto.setUpdatedAt(system2.now()));
   }
 
+  public void updateBranchReferenceValues(DbSession dbSession, BranchDto branchDto, String newBranchName) {
+    requireNonNull(branchDto, "Original referenced branch must be specified.");
+    requireNonNull(branchDto.getProjectUuid(), MSG_PROJECT_UUID_NOT_SPECIFIED);
+    requireNonNull(newBranchName, "New branch name must be specified.");
+    selectAllByProject(dbSession, branchDto.getProjectUuid()).stream()
+      .filter(newCP -> NewCodePeriodType.REFERENCE_BRANCH.equals(newCP.getType()) && branchDto.getBranchKey().equals(newCP.getValue()))
+      .forEach(newCodePeriodDto -> update(dbSession, newCodePeriodDto.setValue(newBranchName)));
+  }
+
   public Optional<NewCodePeriodDto> selectByProject(DbSession dbSession, String projectUuid) {
     requireNonNull(projectUuid, MSG_PROJECT_UUID_NOT_SPECIFIED);
     return ofNullable(mapper(dbSession).selectByProject(projectUuid));