123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228 |
- /*
- * SonarQube
- * Copyright (C) 2009-2020 SonarSource SA
- * mailto:info AT sonarsource DOT com
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 3 of the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public License
- * along with this program; if not, write to the Free Software Foundation,
- * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
- */
- package org.sonar.db.rule;
-
- import java.util.Arrays;
- import java.util.Random;
- import java.util.function.Consumer;
- import org.sonar.api.rule.RuleKey;
- import org.sonar.api.rules.RuleType;
- import org.sonar.api.server.rule.RuleParamType;
- import org.sonar.core.util.Uuids;
- import org.sonar.db.DbTester;
- import org.sonar.db.organization.OrganizationDto;
- import org.sonar.db.user.UserDto;
-
- import static java.util.Arrays.asList;
- import static org.apache.commons.lang.RandomStringUtils.randomAlphabetic;
- import static org.sonar.api.rules.RuleType.SECURITY_HOTSPOT;
- import static org.sonar.db.rule.RuleTesting.newDeprecatedRuleKey;
- import static org.sonar.db.rule.RuleTesting.newRule;
- import static org.sonar.db.rule.RuleTesting.newRuleDto;
-
- public class RuleDbTester {
- private static final RuleType[] RULE_TYPES_EXCEPT_HOTSPOTS = Arrays.stream(RuleType.values())
- .filter(ruleType -> SECURITY_HOTSPOT != ruleType).toArray(RuleType[]::new);
-
- private final DbTester db;
-
- public RuleDbTester(DbTester db) {
- this.db = db;
- }
-
- public RuleDefinitionDto insert() {
- return insert(newRule());
- }
-
- public RuleDefinitionDto insert(RuleKey key) {
- return insert(newRule(key));
- }
-
- @SafeVarargs
- public final RuleDefinitionDto insert(Consumer<RuleDefinitionDto>... populaters) {
- RuleDefinitionDto rule = newRule();
- asList(populaters).forEach(populater -> populater.accept(rule));
- return insert(rule);
- }
-
- public RuleDefinitionDto insert(RuleKey key, Consumer<RuleDefinitionDto> populater) {
- RuleDefinitionDto rule = newRule(key);
- populater.accept(rule);
- return insert(rule);
- }
-
- public RuleDefinitionDto insertIssueRule() {
- return insert(newIssueRule());
- }
-
- public RuleDefinitionDto insertIssueRule(RuleKey key) {
- return insert(newIssueRule(key));
- }
-
- @SafeVarargs
- public final RuleDefinitionDto insertIssueRule(Consumer<RuleDefinitionDto>... populaters) {
- RuleDefinitionDto rule = newIssueRule();
- asList(populaters).forEach(populater -> populater.accept(rule));
- return insert(rule);
- }
-
- public RuleDefinitionDto insertIssueRule(RuleKey key, Consumer<RuleDefinitionDto> populater) {
- RuleDefinitionDto rule = newIssueRule(key);
- populater.accept(rule);
- return insert(rule);
- }
-
- private static RuleDefinitionDto newIssueRule(RuleKey key) {
- return newRule(key).setType(RULE_TYPES_EXCEPT_HOTSPOTS[new Random().nextInt(RULE_TYPES_EXCEPT_HOTSPOTS.length)]);
- }
-
- private static RuleDefinitionDto newIssueRule() {
- return newRule().setType(RULE_TYPES_EXCEPT_HOTSPOTS[new Random().nextInt(RULE_TYPES_EXCEPT_HOTSPOTS.length)]);
- }
-
- public RuleDefinitionDto insertHotspotRule() {
- return insert(newHotspotRule());
- }
-
- @SafeVarargs
- public final RuleDefinitionDto insertHotspotRule(Consumer<RuleDefinitionDto>... populaters) {
- RuleDefinitionDto rule = newHotspotRule();
- asList(populaters).forEach(populater -> populater.accept(rule));
- return insert(rule);
- }
-
- private static RuleDefinitionDto newHotspotRule() {
- return newRule().setType(SECURITY_HOTSPOT);
- }
-
- public RuleDefinitionDto insert(RuleDefinitionDto rule) {
- if (rule.getUuid() == null) {
- rule.setUuid(Uuids.createFast());
- }
-
- db.getDbClient().ruleDao().insert(db.getSession(), rule);
- db.commit();
- return rule;
- }
-
- public RuleDefinitionDto update(RuleDefinitionDto rule) {
- db.getDbClient().ruleDao().update(db.getSession(), rule);
- db.commit();
- return rule;
- }
-
- @SafeVarargs
- public final RuleMetadataDto insertOrUpdateMetadata(RuleDefinitionDto rule, OrganizationDto organization, Consumer<RuleMetadataDto>... populaters) {
- RuleMetadataDto dto = RuleTesting.newRuleMetadata(rule, organization);
- asList(populaters).forEach(populater -> populater.accept(dto));
- return insertOrUpdateMetadata(dto);
- }
-
- @SafeVarargs
- public final RuleMetadataDto insertOrUpdateMetadata(RuleDefinitionDto rule, UserDto noteUser, OrganizationDto organization, Consumer<RuleMetadataDto>... populaters) {
- RuleMetadataDto dto = RuleTesting.newRuleMetadata(rule, noteUser, organization);
- asList(populaters).forEach(populater -> populater.accept(dto));
- return insertOrUpdateMetadata(dto);
- }
-
- public RuleMetadataDto insertOrUpdateMetadata(RuleMetadataDto metadata) {
- db.getDbClient().ruleDao().insertOrUpdate(db.getSession(), metadata);
- db.commit();
- return metadata;
- }
-
- public RuleParamDto insertRuleParam(RuleDefinitionDto rule) {
- return insertRuleParam(rule, p -> {
- });
- }
-
- @SafeVarargs
- public final RuleParamDto insertRuleParam(RuleDefinitionDto rule, Consumer<RuleParamDto>... populaters) {
- RuleParamDto param = RuleTesting.newRuleParam(rule);
- asList(populaters).forEach(populater -> populater.accept(param));
- db.getDbClient().ruleDao().insertRuleParam(db.getSession(), rule, param);
- db.commit();
- return param;
- }
-
- public RuleDto insertRule(RuleDto ruleDto) {
- if (ruleDto.getUuid() == null) {
- ruleDto.setUuid(Uuids.createFast());
- }
-
- insert(ruleDto.getDefinition());
- RuleMetadataDto metadata = ruleDto.getMetadata();
- if (metadata.getOrganizationUuid() != null) {
- db.getDbClient().ruleDao().insertOrUpdate(db.getSession(), metadata.setRuleUuid(ruleDto.getUuid()));
- db.commit();
- }
- return ruleDto;
- }
-
- /**
- * Create and persist a rule with random values.
- */
- public RuleDto insertRule() {
- return insertRule(rule -> {
- });
- }
-
- @SafeVarargs
- public final RuleDto insertRule(OrganizationDto organization, Consumer<RuleDto>... populaters) {
- RuleDto ruleDto = newRuleDto(organization);
- asList(populaters).forEach(populater -> populater.accept(ruleDto));
- return insertRule(ruleDto);
- }
-
- public RuleDto insertRule(Consumer<RuleDto> populateRuleDto) {
- RuleDto ruleDto = newRuleDto();
- populateRuleDto.accept(ruleDto);
-
- if (ruleDto.getUuid() == null) {
- ruleDto.setUuid(Uuids.createFast());
- }
-
- return insertRule(ruleDto);
- }
-
- @SafeVarargs
- public final DeprecatedRuleKeyDto insertDeprecatedKey(Consumer<DeprecatedRuleKeyDto>... deprecatedRuleKeyDtoConsumers) {
- DeprecatedRuleKeyDto deprecatedRuleKeyDto = newDeprecatedRuleKey();
- asList(deprecatedRuleKeyDtoConsumers).forEach(c -> c.accept(deprecatedRuleKeyDto));
- db.getDbClient().ruleDao().insert(db.getSession(), deprecatedRuleKeyDto);
- return deprecatedRuleKeyDto;
- }
-
- public RuleParamDto insertRuleParam(RuleDto rule) {
- RuleParamDto param = new RuleParamDto();
- param.setRuleUuid(rule.getUuid());
- param.setName(randomAlphabetic(10));
- param.setType(RuleParamType.STRING.type());
- db.getDbClient().ruleDao().insertRuleParam(db.getSession(), rule.getDefinition(), param);
- db.commit();
- return param;
- }
-
- public RuleDto insertRule(RuleDefinitionDto ruleDefinition, RuleMetadataDto ruleMetadata) {
- db.getDbClient().ruleDao().insertOrUpdate(db.getSession(), ruleMetadata.setRuleUuid(ruleDefinition.getUuid()));
- db.commit();
- return new RuleDto(ruleDefinition, ruleMetadata);
- }
- }
|