ROASTER-5: Added addParameter(JavaType, String) method

forge · Apr 7, 2014 · 1f98f72 · 1f98f72
1 parent f66b742
commit 1f98f72
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/api/src/main/java/org/jboss/forge/roaster/model/source/MethodSource.java b/api/src/main/java/org/jboss/forge/roaster/model/source/MethodSource.java
@@ -95,6 +95,11 @@ public interface MethodSource<O extends JavaSource<O>> extends Method<O, MethodS
     */
    ParameterSource<O> addParameter(String type, String name);
 
+   /**
+    * Add a parameter with the specified {@link JavaType} type and name to this method
+    */
+   ParameterSource<O> addParameter(JavaType<?> type, String name);
+
    /**
     * Remove a parameter from this method
     */

diff --git a/impl/src/main/java/org/jboss/forge/roaster/model/impl/MethodImpl.java b/impl/src/main/java/org/jboss/forge/roaster/model/impl/MethodImpl.java
@@ -653,6 +653,12 @@ public ParameterSource<O> addParameter(Class<?> type, String name)
       return addParameter(type.getName(), name);
    }
 
+   @Override
+   public ParameterSource<O> addParameter(JavaType<?> type, String name)
+   {
+      return addParameter(type.getQualifiedName(), name);
+   }
+
    @SuppressWarnings("unchecked")
    @Override
    public ParameterSource<O> addParameter(String type, String name)

diff --git a/impl/src/test/java/org/jboss/forge/test/roaster/model/JavaClassMethodTest.java b/impl/src/test/java/org/jboss/forge/test/roaster/model/JavaClassMethodTest.java
@@ -142,6 +142,16 @@ public void testAddParameterStringType() throws Exception
       Assert.assertEquals(3, method.getParameters().size());
    }
 
+   @Test
+   public void testAddParameterJavaType() throws Exception
+   {
+      JavaClassSource type = Roaster.create(JavaClassSource.class).setName("Mock").setPackage("mock.pkg");
+      ParameterSource<JavaClassSource> param = method.addParameter(type, "mock");
+      Assert.assertNotNull(param);
+      Assert.assertEquals(3, method.getParameters().size());
+      Assert.assertTrue(method.getOrigin().hasImport(type));
+   }
+
    @Test
    public void testRemoveParameter() throws Exception
    {