SONAR-7961 Refactoring of add_project and remove_project actions

1 files changed, 13 insertions, 0 deletions

diff --git a/sonar-db/src/test/java/org/sonar/db/qualityprofile/QualityProfileDbTester.java b/sonar-db/src/test/java/org/sonar/db/qualityprofile/QualityProfileDbTester.java
index 0dd2ed189dd..d6e0a12d31c 100644
--- a/sonar-db/src/test/java/org/sonar/db/qualityprofile/QualityProfileDbTester.java
+++ b/sonar-db/src/test/java/org/sonar/db/qualityprofile/QualityProfileDbTester.java
@@ -37,6 +37,11 @@ public class QualityProfileDbTester {
     dbClient.qualityProfileDao().insert(dbSession, qualityProfile, qualityProfiles);
   }
 
+  public QualityProfileDto insertQualityProfile(QualityProfileDto qualityProfile) {
+    dbClient.qualityProfileDao().insert(dbSession, qualityProfile);
+    return qualityProfile;
+  }
+
   public void insertProjectWithQualityProfileAssociations(ComponentDto project, QualityProfileDto... qualityProfiles) {
     dbClient.componentDao().insert(dbSession, project);
     for (QualityProfileDto qualityProfile : qualityProfiles) {
@@ -44,4 +49,12 @@ public class QualityProfileDbTester {
     }
   }
 
+  public void associateProjectWithQualityProfile(ComponentDto project, QualityProfileDto... qualityProfiles) {
+    for (QualityProfileDto qualityProfile : qualityProfiles) {
+      dbClient.qualityProfileDao().insertProjectProfileAssociation(project.uuid(), qualityProfile.getKey(), dbSession);
+    }
+
+    dbSession.commit();
+  }
+
 }