java/jenkins_plugins/build_monitor/src/test/java/com/smartcodeltd/jenkinsci/plugins/buildmonitor/viewmodel/syntacticsugar/BuildStateRecipe.java

package com.smartcodeltd.jenkinsci.plugins.buildmonitor.viewmodel.syntacticsugar;

import static com.smartcodeltd.jenkinsci.plugins.buildmonitor.viewmodel.plugins.bfa.AnalysedTest.failureCauseBuildAction;
import static com.smartcodeltd.jenkinsci.plugins.buildmonitor.viewmodel.plugins.claim.ClaimedTest.claimBuildAction;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import com.sonyericsson.jenkins.plugins.bfa.model.FailureCauseBuildAction;

import hudson.model.Result;
import hudson.model.AbstractBuild;
import hudson.model.AbstractProject;
import hudson.model.User;
import hudson.plugins.claim.ClaimBuildAction;
import hudson.scm.ChangeLogSet;

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Iterator;
import java.util.List;

/**
 * @author Jan Molak
 */
public class BuildStateRecipe {

  private AbstractBuild build;

  public BuildStateRecipe() {
    build = mock(AbstractBuild.class);

    AbstractProject<?, ?> parent = mock(AbstractProject.class);
    when(build.getParent()).thenReturn(parent);
  }

  public BuildStateRecipe numberIs(int number) {
    when(build.getNumber()).thenReturn(number);

    // see hudson.model.Run::getDisplayName
    return nameIs("#" + number);
  }

  public BuildStateRecipe nameIs(String name) {
    when(build.getDisplayName()).thenReturn(name);

    return this;
  }

  public BuildStateRecipe whichNumberIs(int number) {
    return numberIs(number);
  }

  public BuildStateRecipe finishedWith(Result result) {
    when(build.getResult()).thenReturn(result);

    return this;
  }

  public BuildStateRecipe withChangesFrom(String... authors) {
    ChangeLogSet changeSet = changeSetBasedOn(entriesBy(authors));
    when(build.getChangeSet()).thenReturn(changeSet);

    // any methods that use getChangeSet as their source of data should be called normally
    // (build is a partial mock in this case)
    when(build.getCulprits()).thenCallRealMethod();

    return this;
  }

  public BuildStateRecipe succeededThanksTo(String... authors) {
    finishedWith(Result.SUCCESS);
    withChangesFrom(authors);

    return this;
  }

  public BuildStateRecipe wasBrokenBy(String... culprits) {
    finishedWith(Result.FAILURE);
    withChangesFrom(culprits);

    return this;
  }

  public BuildStateRecipe hasntStartedYet() {
    when(build.hasntStartedYet()).thenReturn(true);

    return this;
  }

  public BuildStateRecipe isStillBuilding() {
    when(build.isBuilding()).thenReturn(true);

    return this;
  }

  public BuildStateRecipe isStillUpdatingTheLog() {
    when(build.isLogUpdated()).thenReturn(true);

    return this;
  }

  public BuildStateRecipe startedAt(String time) throws Exception {
    long startedAt = new SimpleDateFormat("H:m:s").parse(time).getTime();

    Calendar timestamp = mock(Calendar.class);
    when(build.getTimestamp()).thenReturn(timestamp);

    when(timestamp.getTimeInMillis()).thenReturn(startedAt);

    return this;
  }

  public BuildStateRecipe andTook(int minutes) throws Exception {
    long duration = (long) minutes * 60 * 1000;

    when(build.getDuration()).thenReturn(duration);

    return this;
  }

  public BuildStateRecipe andUsuallyTakes(int minutes) throws Exception {
    long duration = (long) minutes * 60 * 1000;

    when(build.getEstimatedDuration()).thenReturn(duration);

    return this;
  }

  public BuildStateRecipe andWasClaimedBy(String aPotentialHero, String reason) {
    final ClaimBuildAction action = claimBuildAction(aPotentialHero, reason);
    when(build.getAction(ClaimBuildAction.class)).thenReturn(action);

    return this;
  }

  public BuildStateRecipe andKnownFailures(String... failures) {
    final FailureCauseBuildAction action = failureCauseBuildAction(failures);
    when(build.getAction(FailureCauseBuildAction.class)).thenReturn(action);

    return this;
  }

  public AbstractBuild execute() {
    return build;
  }


  private List<ChangeLogSet.Entry> entriesBy(String... authors) {
    List<ChangeLogSet.Entry> entries = new ArrayList<ChangeLogSet.Entry>();

    for (String name : authors) {
      User author = mock(User.class);
      ChangeLogSet.Entry entry = mock(ChangeLogSet.Entry.class);

      when(author.getFullName()).thenReturn(name);
      when(entry.getAuthor()).thenReturn(author);

      entries.add(entry);
    }
    return entries;
  }

  private ChangeLogSet changeSetBasedOn(final List<ChangeLogSet.Entry> entries) {
    return new ChangeLogSet<ChangeLogSet.Entry>(null) {
      @Override
      public boolean isEmptySet() {
        return false;
      }

      @Override
      public Iterator<Entry> iterator() {
        return entries.iterator();
      }
    };
  }
}