source.dussan.org Git - sonarqube.git/blob

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2019 SonarSource SA
   4  * mailto:info AT sonarsource DOT com
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  */
  20 package org.sonar.ce.task.projectanalysis.webhook;
  21
  22 import com.google.common.collect.ImmutableMap;
  23 import java.util.Collections;
  24 import java.util.Date;
  25 import java.util.Map;
  26 import java.util.Random;
  27 import java.util.function.Supplier;
  28 import javax.annotation.Nullable;
  29 import org.junit.Before;
  30 import org.junit.Test;
  31 import org.mockito.ArgumentCaptor;
  32 import org.sonar.api.ce.posttask.Branch;
  33 import org.sonar.api.ce.posttask.CeTask;
  34 import org.sonar.api.ce.posttask.PostProjectAnalysisTaskTester;
  35 import org.sonar.api.ce.posttask.Project;
  36 import org.sonar.api.ce.posttask.QualityGate;
  37 import org.sonar.api.config.Configuration;
  38 import org.sonar.api.measures.Metric;
  39 import org.sonar.ce.task.projectanalysis.component.ConfigurationRepository;
  40 import org.sonar.server.qualitygate.Condition;
  41 import org.sonar.server.qualitygate.EvaluatedCondition;
  42 import org.sonar.server.qualitygate.EvaluatedQualityGate;
  43 import org.sonar.server.webhook.Analysis;
  44 import org.sonar.server.webhook.ProjectAnalysis;
  45 import org.sonar.server.webhook.WebHooks;
  46 import org.sonar.server.webhook.WebhookPayload;
  47 import org.sonar.server.webhook.WebhookPayloadFactory;
  48
  49 import static org.apache.commons.lang.RandomStringUtils.randomAlphanumeric;
  50 import static org.assertj.core.api.Assertions.assertThat;
  51 import static org.mockito.ArgumentMatchers.any;
  52 import static org.mockito.ArgumentMatchers.eq;
  53 import static org.mockito.Mockito.mock;
  54 import static org.mockito.Mockito.verify;
  55 import static org.mockito.Mockito.when;
  56 import static org.sonar.api.ce.posttask.PostProjectAnalysisTaskTester.newBranchBuilder;
  57 import static org.sonar.api.ce.posttask.PostProjectAnalysisTaskTester.newCeTaskBuilder;
  58 import static org.sonar.api.ce.posttask.PostProjectAnalysisTaskTester.newConditionBuilder;
  59 import static org.sonar.api.ce.posttask.PostProjectAnalysisTaskTester.newProjectBuilder;
  60 import static org.sonar.api.ce.posttask.PostProjectAnalysisTaskTester.newQualityGateBuilder;
  61 import static org.sonar.api.ce.posttask.PostProjectAnalysisTaskTester.newScannerContextBuilder;
  62
  63 public class WebhookPostTaskTest {
  64
  65   private final Random random = new Random();
  66   private final Configuration configuration = mock(Configuration.class);
  67   private final WebhookPayload webhookPayload = mock(WebhookPayload.class);
  68   private final WebhookPayloadFactory payloadFactory = mock(WebhookPayloadFactory.class);
  69   private final WebHooks webHooks = mock(WebHooks.class);
  70   private final ConfigurationRepository configurationRepository = mock(ConfigurationRepository.class);
  71   private WebhookPostTask underTest = new WebhookPostTask(payloadFactory, webHooks);
  72
  73   @Before
  74   public void wireMocks() {
  75     when(payloadFactory.create(any(ProjectAnalysis.class))).thenReturn(webhookPayload);
  76     when(configurationRepository.getConfiguration()).thenReturn(configuration);
  77   }
  78
  79   @Test
  80   public void call_webhooks_when_no_analysis_not_qualitygate() {
  81     callWebHooks(null, null);
  82   }
  83
  84   @Test
  85   public void call_webhooks_with_analysis_and_qualitygate() {
  86     QualityGate.Condition condition = newConditionBuilder()
  87       .setMetricKey(randomAlphanumeric(96))
  88       .setOperator(QualityGate.Operator.values()[random.nextInt(QualityGate.Operator.values().length)])
  89       .setErrorThreshold(randomAlphanumeric(22))
  90       .setWarningThreshold(randomAlphanumeric(23))
  91       .setOnLeakPeriod(random.nextBoolean())
  92       .build(QualityGate.EvaluationStatus.OK, randomAlphanumeric(33));
  93     QualityGate qualityGate = newQualityGateBuilder()
  94       .setId(randomAlphanumeric(23))
  95       .setName(randomAlphanumeric(66))
  96       .setStatus(QualityGate.Status.values()[random.nextInt(QualityGate.Status.values().length)])
  97       .add(condition)
  98       .build();
  99
 100     callWebHooks(randomAlphanumeric(40), qualityGate);
 101   }
 102
 103   private void callWebHooks(@Nullable String analysisUUid, @Nullable QualityGate qualityGate) {
 104     Project project = newProjectBuilder()
 105       .setUuid(randomAlphanumeric(3))
 106       .setKey(randomAlphanumeric(4))
 107       .setName(randomAlphanumeric(5))
 108       .build();
 109     CeTask ceTask = newCeTaskBuilder()
 110       .setStatus(CeTask.Status.values()[random.nextInt(CeTask.Status.values().length)])
 111       .setId(randomAlphanumeric(6))
 112       .build();
 113     Date date = new Date();
 114     Map<String, String> properties = ImmutableMap.of(randomAlphanumeric(17), randomAlphanumeric(18));
 115     Branch branch = newBranchBuilder()
 116       .setIsMain(random.nextBoolean())
 117       .setType(Branch.Type.values()[random.nextInt(Branch.Type.values().length)])
 118       .setName(randomAlphanumeric(29))
 119       .build();
 120
 121     PostProjectAnalysisTaskTester.of(underTest)
 122       .at(date)
 123       .withCeTask(ceTask)
 124       .withProject(project)
 125       .withBranch(branch)
 126       .withQualityGate(qualityGate)
 127       .withScannerContext(newScannerContextBuilder()
 128         .addProperties(properties)
 129         .build())
 130       .withAnalysisUuid(analysisUUid)
 131       .withQualityGate(qualityGate)
 132       .execute();
 133
 134     ArgumentCaptor<Supplier> supplierCaptor = ArgumentCaptor.forClass(Supplier.class);
 135     verify(webHooks)
 136       .sendProjectAnalysisUpdate(
 137         eq(new WebHooks.Analysis(project.getUuid(),
 138           analysisUUid,
 139           ceTask.getId())),
 140         supplierCaptor.capture());
 141
 142     assertThat(supplierCaptor.getValue().get()).isSameAs(webhookPayload);
 143
 144     EvaluatedQualityGate webQualityGate = null;
 145     if (qualityGate != null) {
 146       QualityGate.Condition condition = qualityGate.getConditions().iterator().next();
 147       Condition qgCondition = new Condition(
 148         condition.getMetricKey(),
 149         Condition.Operator.valueOf(condition.getOperator().name()),
 150         condition.getErrorThreshold(),
 151         condition.getWarningThreshold()
 152       );
 153       webQualityGate = EvaluatedQualityGate.newBuilder()
 154         .setQualityGate(new org.sonar.server.qualitygate.QualityGate(qualityGate.getId(), qualityGate.getName(), Collections.singleton(qgCondition)))
 155         .setStatus(Metric.Level.valueOf(qualityGate.getStatus().name()))
 156         .addCondition(qgCondition, EvaluatedCondition.EvaluationStatus.valueOf(condition.getStatus().name()), condition.getValue())
 157         .build();
 158     }
 159
 160     verify(payloadFactory).create(new ProjectAnalysis(
 161       new org.sonar.server.webhook.Project(project.getUuid(), project.getKey(), project.getName()),
 162       new org.sonar.server.webhook.CeTask(ceTask.getId(), org.sonar.server.webhook.CeTask.Status.valueOf(ceTask.getStatus().name())),
 163       analysisUUid == null ? null : new Analysis(analysisUUid, date.getTime()),
 164       new org.sonar.server.webhook.Branch(branch.isMain(), branch.getName().get(), org.sonar.server.webhook.Branch.Type.valueOf(branch.getType().name())),
 165       webQualityGate,
 166       analysisUUid == null ? null : date.getTime(),
 167       properties));
 168   }
 169 }