Test GradleSourceUtil.insert/removeDirectDependency

forge · Aug 6, 2013 · df87146 · df87146
1 parent 9f132ba
commit df87146
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 0 deletions.
diff --git a/impl/src/main/java/org/jboss/forge/addon/gradle/parser/GradleSourceUtil.java b/impl/src/main/java/org/jboss/forge/addon/gradle/parser/GradleSourceUtil.java
@@ -97,6 +97,18 @@ public static String removeDependency(String source, String group, String name,
 
       throw new UnremovableElementException();
    }
+
+   public static String insertDirectDependency(String source, String group, String name)
+   {
+      // TODO
+      return source;
+   }
+
+   public static String removeDirectDependency(String source, String group, String name)
+   {
+      // TODO 
+      return source;
+   }
 
    public static String insertManagedDependency(String source, String group, String name, String version,
             String configuration)

diff --git a/impl/src/test/java/org/jboss/forge/addon/gradle/parser/GradleSourceUtilTest.java b/impl/src/test/java/org/jboss/forge/addon/gradle/parser/GradleSourceUtilTest.java
@@ -108,6 +108,48 @@ public void testRemoveDependencyForException() throws UnremovableElementExceptio
                "}";
       GradleSourceUtil.removeDependency(source, "a", "b", "1.0", "compile");
    }
+
+   @Test
+   public void testInsertDirectDependency()
+   {
+      String source = "" +
+               "dependencies {\n" +
+               "    compile 'a:b:1.0'\n" +
+               "}";
+      String expected = "" +
+               "dependencies {\n" +
+               "    compile 'a:b:1.0'\n" +
+               "    direct handler: it, group: 'x', name: 'y'\n" +
+               "}";
+      String result = GradleSourceUtil.insertDirectDependency(source, "x", "y");
+      assertEquals(expected, result);
+   }
+
+   @Test
+   public void testRemoveDirectDependency()
+   {
+      String source = "" +
+               "dependencies {\n" +
+               "    compile 'a:b:1.0'\n" +
+               "    direct handler: it, group: 'x', name: 'y'\n" +
+               "}";
+      String expected = "" +
+               "dependencies {\n" +
+               "    compile 'a:b:1.0'\n" +
+               "}";
+      String result = GradleSourceUtil.removeDirectDependency(source, "x", "y");
+      assertEquals(expected, result);
+   }
+
+   @Test(expected = UnremovableElementException.class)
+   public void testRemoveDirectDependencyForException()
+   {
+      String source = "" +
+               "dependencies {\n" +
+               "    compile 'a:b:1.0'\n" +
+               "}";
+      GradleSourceUtil.removeDirectDependency(source, "x", "y");
+   }
 
    @Test
    public void testInsertManagedDependency()