SONAR-6993 Add branch in analysis metadata holder

server/sonar-server/src/main/java/org/sonar/server/computation/analysis/AnalysisMetadataHolder.java

@@ -50,4 +50,10 @@ public interface AnalysisMetadataHolder {
    */
   boolean isCrossProjectDuplicationEnabled();
 
+  /**
+   * @throws IllegalStateException if branch has not been set
+   */
+  @CheckForNull
+  String getBranch();
+
 }

server/sonar-server/src/main/java/org/sonar/server/computation/analysis/AnalysisMetadataHolderImpl.java

@@ -24,30 +24,31 @@
 import javax.annotation.Nullable;
 import org.sonar.server.computation.snapshot.Snapshot;
 
+import static com.google.common.base.Preconditions.checkNotNull;
 import static com.google.common.base.Preconditions.checkState;
 
 public class AnalysisMetadataHolderImpl implements MutableAnalysisMetadataHolder {
   @CheckForNull
-  private Long analysisDate;
+  private InitializedProperty<Long> analysisDate = new InitializedProperty<>();
 
-  private boolean baseProjectSnapshotInitialized = false;
+  private InitializedProperty<Snapshot> baseProjectSnapshot = new InitializedProperty<>();
 
   @CheckForNull
-  private Snapshot baseProjectSnapshot;
+  private InitializedProperty<Boolean> crossProjectDuplicationEnabled = new InitializedProperty<>();
 
-  @CheckForNull
-  private Boolean isCrossProjectDuplicationEnabled;
+  private InitializedProperty<String> branch = new InitializedProperty<>();
 
   @Override
   public void setAnalysisDate(Date date) {
-    checkState(analysisDate == null, "Analysis date has already been set");
-    this.analysisDate = date.getTime();
+    checkNotNull(date, "Date must not be null");
+    checkState(!analysisDate.isInitialized(), "Analysis date has already been set");
+    this.analysisDate.setProperty(date.getTime());
   }
 
   @Override
   public Date getAnalysisDate() {
-    checkState(analysisDate != null, "Analysis date has not been set");
-    return new Date(this.analysisDate);
+    checkState(analysisDate.isInitialized(), "Analysis date has not been set");
+    return new Date(this.analysisDate.getProperty());
   }
 
   @Override
@@ -57,27 +58,58 @@ public boolean isFirstAnalysis() {
 
   @Override
   public void setBaseProjectSnapshot(@Nullable Snapshot baseProjectSnapshot) {
-    checkState(!baseProjectSnapshotInitialized, "Base project snapshot has already been set");
-    this.baseProjectSnapshot = baseProjectSnapshot;
-    this.baseProjectSnapshotInitialized = true;
+    checkState(!this.baseProjectSnapshot.isInitialized(), "Base project snapshot has already been set");
+    this.baseProjectSnapshot.setProperty(baseProjectSnapshot);
   }
 
   @Override
   @CheckForNull
   public Snapshot getBaseProjectSnapshot() {
-    checkState(baseProjectSnapshotInitialized, "Base project snapshot has not been set");
-    return baseProjectSnapshot;
+    checkState(baseProjectSnapshot.isInitialized(), "Base project snapshot has not been set");
+    return baseProjectSnapshot.getProperty();
   }
 
   @Override
-  public void setIsCrossProjectDuplicationEnabled(boolean isCrossProjectDuplicationEnabled) {
-    checkState(this.isCrossProjectDuplicationEnabled == null, "Cross project duplication flag has already been set");
-    this.isCrossProjectDuplicationEnabled = isCrossProjectDuplicationEnabled;
+  public void setCrossProjectDuplicationEnabled(boolean isCrossProjectDuplicationEnabled) {
+    checkState(!this.crossProjectDuplicationEnabled.isInitialized(), "Cross project duplication flag has already been set");
+    this.crossProjectDuplicationEnabled.setProperty(isCrossProjectDuplicationEnabled);
   }
 
   @Override
   public boolean isCrossProjectDuplicationEnabled() {
-    checkState(isCrossProjectDuplicationEnabled != null, "Cross project duplication flag has not been set");
-    return isCrossProjectDuplicationEnabled;
+    checkState(crossProjectDuplicationEnabled.isInitialized(), "Cross project duplication flag has not been set");
+    return crossProjectDuplicationEnabled.getProperty();
+  }
+
+  @Override
+  public void setBranch(@Nullable String branch) {
+    checkState(!this.branch.isInitialized(), "Branch has already been set");
+    this.branch.setProperty(branch);
+  }
+
+  @Override
+  public String getBranch() {
+    checkState(branch.isInitialized(), "Branch has not been set");
+    return branch.getProperty();
+  }
+
+  private static class InitializedProperty<E> {
+    private E property;
+    private boolean initialized = false;
+
+    public InitializedProperty setProperty(@Nullable E property) {
+      this.property = property;
+      this.initialized = true;
+      return this;
+    }
+
+    @CheckForNull
+    public E getProperty() {
+      return property;
+    }
+
+    public boolean isInitialized() {
+      return initialized;
+    }
   }
 }

server/sonar-server/src/main/java/org/sonar/server/computation/analysis/MutableAnalysisMetadataHolder.java

@@ -39,6 +39,11 @@ public interface MutableAnalysisMetadataHolder extends AnalysisMetadataHolder {
   /**
    * @throws IllegalStateException if cross project duplication flag has already been set
    */
-  void setIsCrossProjectDuplicationEnabled(boolean isCrossProjectDuplicationEnabled);
+  void setCrossProjectDuplicationEnabled(boolean isCrossProjectDuplicationEnabled);
+
+  /**
+   * @throws IllegalStateException if branch has already been set
+   */
+  void setBranch(@Nullable String branch);
 
 }

server/sonar-server/src/test/java/org/sonar/server/computation/analysis/AnalysisMetadataHolderImplTest.java

@@ -60,6 +60,16 @@ public void getAnalysisDate_throws_ISE_when_holder_is_not_initialized() {
     new AnalysisMetadataHolderImpl().getAnalysisDate();
   }
 
+  @Test
+  public void setAnalysisDate_throws_NPE_when_date_is_null() {
+    AnalysisMetadataHolderImpl underTest = new AnalysisMetadataHolderImpl();
+
+    expectedException.expect(NullPointerException.class);
+    expectedException.expectMessage("Date must not be null");
+
+    underTest.setAnalysisDate(null);
+  }
+
   @Test
   public void setAnalysisDate_throws_ISE_when_called_twice() {
     AnalysisMetadataHolderImpl underTest = new AnalysisMetadataHolderImpl();
@@ -117,7 +127,7 @@ public void setBaseProjectSnapshot_throws_ISE_when_called_twice() {
   public void isCrossProjectDuplicationEnabled_return_true() {
     AnalysisMetadataHolderImpl underTest = new AnalysisMetadataHolderImpl();
 
-    underTest.setIsCrossProjectDuplicationEnabled(true);
+    underTest.setCrossProjectDuplicationEnabled(true);
 
     assertThat(underTest.isCrossProjectDuplicationEnabled()).isEqualTo(true);
   }
@@ -126,7 +136,7 @@ public void isCrossProjectDuplicationEnabled_return_true() {
   public void isCrossProjectDuplicationEnabled_return_false() {
     AnalysisMetadataHolderImpl underTest = new AnalysisMetadataHolderImpl();
 
-    underTest.setIsCrossProjectDuplicationEnabled(false);
+    underTest.setCrossProjectDuplicationEnabled(false);
 
     assertThat(underTest.isCrossProjectDuplicationEnabled()).isEqualTo(false);
   }
@@ -142,10 +152,46 @@ public void isCrossProjectDuplicationEnabled_throws_ISE_when_holder_is_not_initi
   @Test
   public void setIsCrossProjectDuplicationEnabled_throws_ISE_when_called_twice() {
     AnalysisMetadataHolderImpl underTest = new AnalysisMetadataHolderImpl();
-    underTest.setIsCrossProjectDuplicationEnabled(true);
+    underTest.setCrossProjectDuplicationEnabled(true);
 
     expectedException.expect(IllegalStateException.class);
     expectedException.expectMessage("Cross project duplication flag has already been set");
-    underTest.setIsCrossProjectDuplicationEnabled(false);
+    underTest.setCrossProjectDuplicationEnabled(false);
+  }
+
+  @Test
+  public void set_branch() {
+    AnalysisMetadataHolderImpl underTest = new AnalysisMetadataHolderImpl();
+
+    underTest.setBranch("origin/master");
+
+    assertThat(underTest.getBranch()).isEqualTo("origin/master");
+  }
+
+  @Test
+  public void set_no_branch() {
+    AnalysisMetadataHolderImpl underTest = new AnalysisMetadataHolderImpl();
+
+    underTest.setBranch(null);
+
+    assertThat(underTest.getBranch()).isNull();
+  }
+
+  @Test
+  public void branch_throws_ISE_when_holder_is_not_initialized() {
+    expectedException.expect(IllegalStateException.class);
+    expectedException.expectMessage("Branch has not been set");
+
+    new AnalysisMetadataHolderImpl().getBranch();
+  }
+
+  @Test
+  public void setBranch_throws_ISE_when_called_twice() {
+    AnalysisMetadataHolderImpl underTest = new AnalysisMetadataHolderImpl();
+    underTest.setBranch("origin/master");
+
+    expectedException.expect(IllegalStateException.class);
+    expectedException.expectMessage("Branch has already been set");
+    underTest.setBranch("origin/master");
   }
 }

server/sonar-server/src/test/java/org/sonar/server/computation/analysis/MutableAnalysisMetadataHolderRule.java

@@ -66,7 +66,17 @@ public boolean isCrossProjectDuplicationEnabled() {
   }
 
   @Override
-  public void setIsCrossProjectDuplicationEnabled(boolean isCrossProjectDuplicationEnabled) {
-    delegate.setIsCrossProjectDuplicationEnabled(isCrossProjectDuplicationEnabled);
+  public void setCrossProjectDuplicationEnabled(boolean isCrossProjectDuplicationEnabled) {
+    delegate.setCrossProjectDuplicationEnabled(isCrossProjectDuplicationEnabled);
+  }
+
+  @Override
+  public String getBranch() {
+    return delegate.getBranch();
+  }
+
+  @Override
+  public void setBranch(@Nullable String branch) {
+    delegate.setBranch(branch);
   }
 }