Search in sources :

Example 1 with BuildStatus

use of com.atlassian.stash.build.BuildStatus in project stashbot by palantir.

the class BuildSuccessReportingServletTest method testReportingInprogress.

@Test
public void testReportingInprogress() throws ServletException, IOException {
    Mockito.when(req.getPathInfo()).thenReturn(buildPathInfo(REPO_ID, JobType.VERIFY_COMMIT, INPROGRESS, BUILD_NUMBER, HEAD, null, null));
    bsrs.doGet(req, res);
    ArgumentCaptor<BuildStatus> buildStatusCaptor = ArgumentCaptor.forClass(BuildStatus.class);
    Mockito.verify(bss).add(Mockito.eq(HEAD), buildStatusCaptor.capture());
    Mockito.verify(res).setStatus(200);
    String output = mockWriter.toString();
    Assert.assertTrue(output.contains("Status Updated"));
    BuildStatus bs = buildStatusCaptor.getValue();
    Assert.assertEquals(bs.getState(), INPROGRESS);
    Assert.assertTrue(bs.getKey().contains(JobType.VERIFY_COMMIT.toString()));
    Assert.assertTrue(bs.getName().contains(JobType.VERIFY_COMMIT.toString()));
}
Also used : BuildStatus(com.atlassian.stash.build.BuildStatus) Test(org.junit.Test)

Example 2 with BuildStatus

use of com.atlassian.stash.build.BuildStatus in project stashbot by palantir.

the class BuildSuccessReportingServlet method getSuccessStatus.

private BuildStatus getSuccessStatus(Repository repo, JobTemplate jt, State state, long buildNumber, String buildHead) throws SQLException {
    Date now = new Date(java.lang.System.currentTimeMillis());
    DateFormat df = DateFormat.getDateInstance();
    // key will be the jenkins name
    String key = jt.getBuildNameFor(repo);
    String name = key + " (build " + Long.toString(buildNumber) + ")";
    String description = "Build " + Long.toString(buildNumber) + " " + state.toString() + " at " + df.format(now);
    String url = getJenkinsUrl(repo, jt, buildNumber);
    BuildStatus bs = new InternalBuildStatus(state, name, name, url, description, now);
    return bs;
}
Also used : DateFormat(java.text.DateFormat) BuildStatus(com.atlassian.stash.build.BuildStatus) InternalBuildStatus(com.atlassian.stash.internal.build.InternalBuildStatus) Date(java.sql.Date) InternalBuildStatus(com.atlassian.stash.internal.build.InternalBuildStatus)

Example 3 with BuildStatus

use of com.atlassian.stash.build.BuildStatus in project stashbot by palantir.

the class BuildSuccessReportingServletTest method testReportingSuccess.

@Test
public void testReportingSuccess() throws ServletException, IOException {
    Mockito.when(req.getPathInfo()).thenReturn(buildPathInfo(REPO_ID, JobType.VERIFY_COMMIT, SUCCESSFUL, BUILD_NUMBER, HEAD, null, null));
    bsrs.doGet(req, res);
    ArgumentCaptor<BuildStatus> buildStatusCaptor = ArgumentCaptor.forClass(BuildStatus.class);
    Mockito.verify(bss).add(Mockito.eq(HEAD), buildStatusCaptor.capture());
    Mockito.verify(res).setStatus(200);
    String output = mockWriter.toString();
    Assert.assertTrue(output.contains("Status Updated"));
    BuildStatus bs = buildStatusCaptor.getValue();
    Assert.assertEquals(bs.getState(), SUCCESSFUL);
    Assert.assertTrue(bs.getKey().contains(JobType.VERIFY_COMMIT.toString()));
    Assert.assertTrue(bs.getName().contains(JobType.VERIFY_COMMIT.toString()));
}
Also used : BuildStatus(com.atlassian.stash.build.BuildStatus) Test(org.junit.Test)

Example 4 with BuildStatus

use of com.atlassian.stash.build.BuildStatus in project stashbot by palantir.

the class BuildSuccessReportingServletTest method testReportingFailure.

@Test
public void testReportingFailure() throws ServletException, IOException {
    Mockito.when(req.getPathInfo()).thenReturn(buildPathInfo(REPO_ID, JobType.VERIFY_COMMIT, FAILED, BUILD_NUMBER, HEAD, null, null));
    bsrs.doGet(req, res);
    ArgumentCaptor<BuildStatus> buildStatusCaptor = ArgumentCaptor.forClass(BuildStatus.class);
    Mockito.verify(bss).add(Mockito.eq(HEAD), buildStatusCaptor.capture());
    Mockito.verify(res).setStatus(200);
    String output = mockWriter.toString();
    Assert.assertTrue(output.contains("Status Updated"));
    BuildStatus bs = buildStatusCaptor.getValue();
    Assert.assertEquals(bs.getState(), FAILED);
    Assert.assertTrue(bs.getKey().contains(JobType.VERIFY_COMMIT.toString()));
    Assert.assertTrue(bs.getName().contains(JobType.VERIFY_COMMIT.toString()));
}
Also used : BuildStatus(com.atlassian.stash.build.BuildStatus) Test(org.junit.Test)

Example 5 with BuildStatus

use of com.atlassian.stash.build.BuildStatus in project stashbot by palantir.

the class BuildSuccessReportingServlet method doGet.

@Override
public void doGet(HttpServletRequest req, HttpServletResponse res) throws ServletException, IOException {
    try {
        // Look at JenkinsManager class if you change this:
        // final two arguments could be empty...
        final String URL_FORMAT = "BASE_URL/REPO_ID/TYPE/STATE/BUILD_NUMBER/BUILD_HEAD[/MERGE_HEAD/PULLREQUEST_ID]";
        final String pathInfo = req.getPathInfo();
        final String[] parts = pathInfo.split("/");
        if (parts.length != 6 && parts.length != 8) {
            throw new IllegalArgumentException("The format of the URL is " + URL_FORMAT);
        }
        final int repoId;
        final RepositoryConfiguration rc;
        try {
            repoId = Integer.valueOf(parts[1]);
        } catch (NumberFormatException e) {
            throw new IllegalArgumentException("The format of the URL is " + URL_FORMAT, e);
        }
        // This is necessary if we want unauthenticated users to be able to call this.  *sigh*
        RepoIdFetcherOperation getRepoId = new RepoIdFetcherOperation(repositoryService, repoId);
        ss.withPermission(Permission.REPO_READ, "BUILD SUCCESS REPORT").call(getRepoId);
        final Repository repo = getRepoId.getRepo();
        rc = configurationPersistanceManager.getRepositoryConfigurationForRepository(repo);
        if (repo == null) {
            throw new IllegalArgumentException("Unable to get a repository for id " + repoId);
        }
        JobTemplate jt = jtm.fromString(rc, parts[2].toLowerCase());
        if (jt == null) {
            throw new IllegalArgumentException("Unable to get a valid JobTemplate from " + parts[2]);
        }
        final State state = BuildStatus.State.fromString(parts[3]);
        if (state == null) {
            throw new IllegalArgumentException("The state must be 'successful', 'failed', or 'inprogress'");
        }
        final long buildNumber;
        try {
            buildNumber = Long.parseLong(parts[4]);
        } catch (NumberFormatException e) {
            throw new IllegalArgumentException("Unable to parse build number", e);
        }
        // TODO: ensure this hash actually exists?
        final String buildHead = parts[5];
        final String mergeHead;
        final long pullRequestId;
        final PullRequest pullRequest;
        final String retUrl;
        if (parts.length == 8 && !parts[6].isEmpty() && !parts[7].isEmpty()) {
            mergeHead = parts[6];
            try {
                // This is a pull request, so add a comment
                pullRequestId = Long.parseLong(parts[7]);
                PullRequestFetcherOperation prfo = new PullRequestFetcherOperation(pullRequestService, repoId, pullRequestId);
                ss.withPermission(Permission.REPO_READ, "BUILD SUCCESS REPORT").call(prfo);
                pullRequest = prfo.getPullRequest();
                if (pullRequest == null) {
                    throw new IllegalArgumentException("Unable to find pull request for repo id " + repo.getId().toString() + " pr id " + Long.toString(pullRequestId));
                }
            } catch (NumberFormatException e) {
                throw new IllegalArgumentException("Unable to parse pull request id " + parts[7], e);
            }
            retUrl = ub.getJenkinsTriggerUrl(repo, jt.getJobType(), buildHead, pullRequest);
        } else {
            mergeHead = null;
            pullRequestId = 0;
            pullRequest = null;
            retUrl = ub.getJenkinsTriggerUrl(repo, jt.getJobType(), buildHead, null);
        }
        if (mergeHead == null) {
            BuildStatus bs;
            bs = getSuccessStatus(repo, jt, state, buildNumber, buildHead);
            log.debug("Registering build status for buildHead " + buildHead + " " + bsToString(bs));
            BuildStatusAddOperation bssAdder = new BuildStatusAddOperation(buildStatusService, buildHead, bs);
            // Yeah, I know what you are thinking... "Admin permission?  To add a build status?"
            // I tried REPO_WRITE and REPO_ADMIN and neither was enough, but this worked!
            ss.withPermission(Permission.SYS_ADMIN, "BUILD SUCCESS REPORT").call(bssAdder);
            printOutput(req, res);
            return;
        }
        // arg order for bools is started, success, override, failed
        if (state.equals(State.SUCCESSFUL)) {
            configurationPersistanceManager.setPullRequestMetadata(pullRequest, mergeHead, buildHead, null, true, null, false);
        } else if (state.equals(State.INPROGRESS)) {
            configurationPersistanceManager.setPullRequestMetadata(pullRequest, mergeHead, buildHead, true, false, null, null);
        } else if (state.equals(State.FAILED)) {
            configurationPersistanceManager.setPullRequestMetadata(pullRequest, mergeHead, buildHead, null, false, null, true);
        }
        // mergeHead is not null *and* pullRequest is not null if we reach
        // here.
        final StringBuffer sb = new StringBuffer();
        final String url = getJenkinsUrl(repo, jt, buildNumber);
        /* NOTE: mergeHead and buildHead are the reverse of what you might
             * think, because we have to check out the "toRef" becasue it is
             * the ref that is guaranteed to be in the correct repo.
             * Nonetheless, buildHead is the commit that is being merged "into"
             * the target branch, which is the mergeHead variable here.
             */
        final int hashLength = 4;
        final String shortMergeHead = mergeHead.substring(0, hashLength);
        final String shortBuildHead = buildHead.substring(0, hashLength);
        final String mergeHeadUrl = ub.buildStashCommitUrl(repo, mergeHead);
        final String buildHeadUrl = ub.buildStashCommitUrl(repo, buildHead);
        final String mergeHeadLink = "[" + shortMergeHead + "](" + mergeHeadUrl + ")";
        final String buildHeadLink = "[" + shortBuildHead + "](" + buildHeadUrl + ")";
        final String consoleUrl = url + "/console";
        sb.append("*[Build #" + buildNumber + "](" + url + ") ");
        sb.append("(merging " + mergeHeadLink + " into " + buildHeadLink + ") ");
        switch(state) {
            case INPROGRESS:
                sb.append("is in progress...*");
                break;
            case SUCCESSFUL:
                sb.append("has **passed &#x2713;**.*");
                break;
            case FAILED:
                sb.append("has* **FAILED &#x2716;**. ");
                sb.append("([*Retrigger this build* &#x27f3;](" + retUrl + ") *or* [*view console output* &#x2630;](" + consoleUrl + ").)");
                break;
        }
        log.debug("Registering comment on pr for buildHead " + buildHead + " mergeHead " + mergeHead);
        // Still make comment so users can see links to build
        PullRequestCommentAddOperation prcao = new PullRequestCommentAddOperation(pullRequestService, repo.getId(), pullRequest.getId(), sb.toString());
        // So in order to create comments, we have to do it AS some user.  ss.doAsUser rather than ss.doWithPermission is the magic sauce here.
        JenkinsServerConfiguration jsc = configurationPersistanceManager.getJenkinsServerConfiguration(rc.getJenkinsServerName());
        StashUser user = us.findUserByNameOrEmail(jsc.getStashUsername());
        ss.impersonating(user, "BUILD SUCCESS REPORT").call(prcao);
        printOutput(req, res);
    } catch (SQLException e) {
        throw new RuntimeException("Unable to get configuration", e);
    } catch (Exception e) {
        throw new RuntimeException("Unable to report build status", e);
    }
}
Also used : SQLException(java.sql.SQLException) PullRequest(com.atlassian.stash.pull.PullRequest) JobTemplate(com.palantir.stash.stashbot.persistence.JobTemplate) ServletException(javax.servlet.ServletException) SQLException(java.sql.SQLException) IOException(java.io.IOException) BuildStatusAddOperation(com.palantir.stash.stashbot.util.BuildStatusAddOperation) Repository(com.atlassian.stash.repository.Repository) StashUser(com.atlassian.stash.user.StashUser) State(com.atlassian.stash.build.BuildStatus.State) BuildStatus(com.atlassian.stash.build.BuildStatus) InternalBuildStatus(com.atlassian.stash.internal.build.InternalBuildStatus) PullRequestCommentAddOperation(com.palantir.stash.stashbot.util.PullRequestCommentAddOperation) RepoIdFetcherOperation(com.palantir.stash.stashbot.util.RepoIdFetcherOperation) PullRequestFetcherOperation(com.palantir.stash.stashbot.util.PullRequestFetcherOperation) RepositoryConfiguration(com.palantir.stash.stashbot.persistence.RepositoryConfiguration) JenkinsServerConfiguration(com.palantir.stash.stashbot.persistence.JenkinsServerConfiguration)

Aggregations

BuildStatus (com.atlassian.stash.build.BuildStatus)5 Test (org.junit.Test)3 InternalBuildStatus (com.atlassian.stash.internal.build.InternalBuildStatus)2 State (com.atlassian.stash.build.BuildStatus.State)1 PullRequest (com.atlassian.stash.pull.PullRequest)1 Repository (com.atlassian.stash.repository.Repository)1 StashUser (com.atlassian.stash.user.StashUser)1 JenkinsServerConfiguration (com.palantir.stash.stashbot.persistence.JenkinsServerConfiguration)1 JobTemplate (com.palantir.stash.stashbot.persistence.JobTemplate)1 RepositoryConfiguration (com.palantir.stash.stashbot.persistence.RepositoryConfiguration)1 BuildStatusAddOperation (com.palantir.stash.stashbot.util.BuildStatusAddOperation)1 PullRequestCommentAddOperation (com.palantir.stash.stashbot.util.PullRequestCommentAddOperation)1 PullRequestFetcherOperation (com.palantir.stash.stashbot.util.PullRequestFetcherOperation)1 RepoIdFetcherOperation (com.palantir.stash.stashbot.util.RepoIdFetcherOperation)1 IOException (java.io.IOException)1 Date (java.sql.Date)1 SQLException (java.sql.SQLException)1 DateFormat (java.text.DateFormat)1 ServletException (javax.servlet.ServletException)1