Move test data creation to class TestData

scm-it/src/test/java/sonia/scm/it/TestData.java

@@ -5,11 +5,15 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import sonia.scm.web.VndMediaType;
 
+import javax.json.Json;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 import static java.util.Arrays.asList;
 import static sonia.scm.it.RestUtil.createResourceUrl;
 import static sonia.scm.it.RestUtil.given;
+import static sonia.scm.it.ScmTypes.availableScmTypes;
 
 public class TestData {
 
@@ -17,13 +21,24 @@ public class TestData {
 
   private static final List<String> PROTECTED_USERS = asList("scmadmin", "anonymous");
 
+  private static Map<String, String> DEFAULT_REPOSITORIES = new HashMap<>();
+
+  public static void createDefault() {
+    cleanup();
+    createDefaultRepositories();
+  }
+
   public static void cleanup() {
     cleanupRepositories();
     cleanupGroups();
     cleanupUsers();
   }
 
-  public static void cleanupRepositories() {
+  public static String getDefaultRepositoryUrl(String repositoryType) {
+    return DEFAULT_REPOSITORIES.get(repositoryType);
+  }
+
+  private static void cleanupRepositories() {
     List<String> repositories = given(VndMediaType.REPOSITORY_COLLECTION)
       .when()
       .get(createResourceUrl("repositories"))
@@ -33,9 +48,10 @@ public class TestData {
       .body().jsonPath().getList("_embedded.repositories._links.self.href");
     LOG.info("about to delete {} repositories", repositories.size());
     repositories.forEach(TestData::delete);
+    DEFAULT_REPOSITORIES.clear();
   }
 
-  public static void cleanupGroups() {
+  private static void cleanupGroups() {
     List<String> groups = given(VndMediaType.GROUP_COLLECTION)
       .when()
       .get(createResourceUrl("groups"))
@@ -47,7 +63,7 @@ public class TestData {
     groups.forEach(TestData::delete);
   }
 
-  public static void cleanupUsers() {
+  private static void cleanupUsers() {
     List<String> users = given(VndMediaType.USER_COLLECTION)
       .when()
       .get(createResourceUrl("users"))
@@ -59,7 +75,7 @@ public class TestData {
     users.stream().filter(url -> PROTECTED_USERS.stream().noneMatch(url::contains)).forEach(TestData::delete);
   }
 
-  public static void delete(String url) {
+  private static void delete(String url) {
     given(VndMediaType.REPOSITORY)
       .when()
       .delete(url)
@@ -68,6 +84,32 @@ public class TestData {
     LOG.info("deleted {}", url);
   }
 
+  private static void createDefaultRepositories() {
+    for (String repositoryType : availableScmTypes()) {
+      String url = given(VndMediaType.REPOSITORY)
+        .body(repositoryJson(repositoryType))
+
+        .when()
+        .post(createResourceUrl("repositories"))
+
+        .then()
+        .statusCode(HttpStatus.SC_CREATED)
+        .extract()
+        .header("location");
+      DEFAULT_REPOSITORIES.put(repositoryType, url);
+    }
+  }
+
+  public static String repositoryJson(String repositoryType) {
+    return Json.createObjectBuilder()
+      .add("contact", "zaphod.beeblebrox@hitchhiker.com")
+      .add("description", "Heart of Gold")
+      .add("name", "HeartOfGold-" + repositoryType)
+      .add("archived", false)
+      .add("type", repositoryType)
+      .build().toString();
+  }
+
   public static void main(String[] args) {
     cleanup();
   }