Fixes for #637, making theory parameters assignment more consistent across sources

src/main/java/org/junit/experimental/theories/ParameterSignature.java

@@ -5,9 +5,35 @@
 import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 public class ParameterSignature {
+
+    private static final Map<Class<?>, Class<?>> CONVERTABLE_TYPES_MAP = buildConvertableTypesMap();
+
+    private static Map<Class<?>, Class<?>> buildConvertableTypesMap() {
+        Map<Class<?>, Class<?>> map = new HashMap<Class<?>, Class<?>>();
+
+        putSymmetrically(map, boolean.class, Boolean.class);
+        putSymmetrically(map, byte.class, Byte.class);
+        putSymmetrically(map, short.class, Short.class);
+        putSymmetrically(map, char.class, Character.class);
+        putSymmetrically(map, int.class, Integer.class);
+        putSymmetrically(map, long.class, Long.class);
+        putSymmetrically(map, float.class, Float.class);
+        putSymmetrically(map, double.class, Double.class);
+
+        return Collections.unmodifiableMap(map);
+    }
+
+    private static <T> void putSymmetrically(Map<T, T> map, T a, T b) {
+        map.put(a, b);
+        map.put(b, a);
+    }
+
     public static ArrayList<ParameterSignature> signatures(Method method) {
         return signatures(method.getParameterTypes(), method
                 .getParameterAnnotations());
@@ -42,21 +68,33 @@ public boolean canAcceptValue(Object candidate) {
     }
 
     public boolean canAcceptType(Class<?> candidate) {
-        return type.isAssignableFrom(candidate);
+        return type.isAssignableFrom(candidate) ||
+                isAssignableViaTypeConversion(type, candidate);
+    }
+
+    public boolean canPotentiallyAcceptType(Class<?> candidate) {
+        return candidate.isAssignableFrom(type) ||
+                isAssignableViaTypeConversion(candidate, type) ||
+                canAcceptType(candidate);
+    }
+
+    private boolean isAssignableViaTypeConversion(Class<?> targetType, Class<?> candidate) {
+        if (CONVERTABLE_TYPES_MAP.containsKey(candidate)) {
+            Class<?> wrapperClass = CONVERTABLE_TYPES_MAP.get(candidate);
+            return targetType.isAssignableFrom(wrapperClass);
+        } else {
+            return false;
+        }
     }
 
-    public Class<?> getType() {
+	public Class<?> getType() {
         return type;
     }
 
     public List<Annotation> getAnnotations() {
         return Arrays.asList(annotations);
     }
 
-    public boolean canAcceptArrayType(Class<?> type) {
-        return type.isArray() && canAcceptType(type.getComponentType());
-    }
-
     public boolean hasAnnotation(Class<? extends Annotation> type) {
         return getAnnotation(type) != null;
     }

src/main/java/org/junit/experimental/theories/internal/AllMembersSupplier.java

@@ -40,14 +40,13 @@ public Object getValue() throws CouldNotGenerateValueException {
                 throw new CouldNotGenerateValueException();
                 // do nothing, just look for more values
             }
-        }
-
+        }    
+    
         @Override
         public String getDescription() throws CouldNotGenerateValueException {
             return fMethod.getName();
         }
-    }   
-
+    }           
     private final TestClass fClass;
 
     /**
@@ -71,60 +70,48 @@ public List<PotentialAssignment> getValueSources(ParameterSignature sig) {
 
     private void addMultiPointMethods(ParameterSignature sig, List<PotentialAssignment> list) {
         for (FrameworkMethod dataPointsMethod : getDataPointsMethods(sig)) {
-            try {
-                addMultiPointArrayValues(sig, dataPointsMethod.getName(), list, dataPointsMethod.invokeExplosively(null));
-            } catch (Throwable e) {
-                // ignore and move on
+            Class<?> returnType = dataPointsMethod.getReturnType();
+
+            if (returnType.isArray() && sig.canPotentiallyAcceptType(returnType.getComponentType())) {
+                try {
+                    addArrayValues(sig, dataPointsMethod.getName(), list, dataPointsMethod.invokeExplosively(null));
+                } catch (Throwable e) {
+                    // ignore and move on
+                }
             }
         }
     }
 
-    private void addSinglePointMethods(ParameterSignature sig,
-            List<PotentialAssignment> list) {
+    private void addSinglePointMethods(ParameterSignature sig, List<PotentialAssignment> list) {
         for (FrameworkMethod dataPointMethod : getSingleDataPointMethods(sig)) {
             if (sig.canAcceptType(dataPointMethod.getType())) {
                 list.add(new MethodParameterValue(dataPointMethod));
             }
         }
     }
 
-    private void addMultiPointFields(ParameterSignature sig,
-            List<PotentialAssignment> list) {
+    private void addMultiPointFields(ParameterSignature sig, List<PotentialAssignment> list) {
         for (final Field field : getDataPointsFields(sig)) {
-            Class<?> type = field.getType();
-            if (sig.canAcceptArrayType(type)) {
-                try {
-                    addArrayValues(field.getName(), list, getStaticFieldValue(field));
-                } catch (Throwable e) {
-                    // ignore and move on
-                }
-            }
+            addArrayValues(sig, field.getName(), list, getStaticFieldValue(field));
         }
-    }    
+    }
 
-    private void addSinglePointFields(ParameterSignature sig,
-            List<PotentialAssignment> list) {
+    private void addSinglePointFields(ParameterSignature sig, List<PotentialAssignment> list) {
         for (final Field field : getSingleDataPointFields(sig)) {
-            Class<?> type = field.getType();
-            if (sig.canAcceptType(type)) {
-                list.add(PotentialAssignment.forValue(field.getName(), getStaticFieldValue(field)));
+            Object value = getStaticFieldValue(field);
+
+            if (sig.canAcceptValue(value)) {
+                list.add(PotentialAssignment.forValue(field.getName(), value));
             }
         }
     }
 
-    private void addArrayValues(String name, List<PotentialAssignment> list, Object array) {
-        for (int i = 0; i < Array.getLength(array); i++) {
-            list.add(PotentialAssignment.forValue(name + "[" + i + "]", Array.get(array, i)));
-        }
-    }
-
-    private void addMultiPointArrayValues(ParameterSignature sig, String name, List<PotentialAssignment> list,
-            Object array) throws Throwable {
+    private void addArrayValues(ParameterSignature sig, String name, List<PotentialAssignment> list, Object array) {
         for (int i = 0; i < Array.getLength(array); i++) {
-            if (!sig.canAcceptValue(Array.get(array, i))) {
-                return;
+            Object value = Array.get(array, i);
+            if (sig.canAcceptValue(value)) {
+                list.add(PotentialAssignment.forValue(name + "[" + i + "]", value));
             }
-            list.add(PotentialAssignment.forValue(name + "[" + i + "]", Array.get(array, i)));
         }
     }
 

src/test/java/org/junit/tests/experimental/theories/AssumingInTheoriesTest.java

@@ -1,47 +1,42 @@
 package org.junit.tests.experimental.theories;
 
+import static org.junit.tests.experimental.theories.TheoryTestUtils.runTheoryClass;
 import org.junit.Assert;
 import org.junit.Assume;
 import org.junit.Test;
 import org.junit.experimental.theories.DataPoint;
 import org.junit.experimental.theories.Theories;
 import org.junit.experimental.theories.Theory;
-import org.junit.runner.JUnitCore;
-import org.junit.runner.Request;
 import org.junit.runner.Result;
 import org.junit.runner.RunWith;
-import org.junit.runner.Runner;
 import org.junit.runners.model.InitializationError;
 
 @RunWith(Theories.class)
 public class AssumingInTheoriesTest {
 
-	@Test
-	public void noTheoryAnnotationMeansAssumeShouldIgnore() {
-		Assume.assumeTrue(false);
-	}
-
-	@Test
-	public void theoryMeansOnlyAssumeShouldFail() throws InitializationError {
-		JUnitCore junitRunner = new JUnitCore();
-		Runner theoryRunner = new Theories(TheoryWithNoUnassumedParameters.class);
-		Request request = Request.runner(theoryRunner);
-		Result result = junitRunner.run(request);
-		Assert.assertEquals(1, result.getFailureCount());
-	}
-
-	/**
-	 * Simple class that SHOULD fail because no parameters are met.
-	 */
-	public static class TheoryWithNoUnassumedParameters {
-
-		@DataPoint 
-		public final static boolean FALSE = false;
-
-		@Theory
-		public void theoryWithNoUnassumedParameters(boolean value) {
-			Assume.assumeTrue(value);
-		}
-	}
+    @Test
+    public void noTheoryAnnotationMeansAssumeShouldIgnore() {
+        Assume.assumeTrue(false);
+    }
+
+    @Test
+    public void theoryMeansOnlyAssumeShouldFail() throws InitializationError {
+        Result result = runTheoryClass(TheoryWithNoUnassumedParameters.class);
+        Assert.assertEquals(1, result.getFailureCount());
+    }
+
+    /**
+     * Simple class that SHOULD fail because no parameters are met.
+     */
+    public static class TheoryWithNoUnassumedParameters {
+
+        @DataPoint
+        public final static boolean FALSE = false;
+
+        @Theory
+        public void theoryWithNoUnassumedParameters(boolean value) {
+            Assume.assumeTrue(value);
+        }
+    }
 
 }

src/test/java/org/junit/tests/experimental/theories/ParameterSignatureTest.java

@@ -3,6 +3,7 @@
 import static org.hamcrest.CoreMatchers.isA;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
 import static org.junit.Assume.assumeTrue;
 
 import java.lang.annotation.Annotation;
@@ -40,17 +41,52 @@ public void getType(Method method, int index) {
                 .signatures(method).get(index).getType());
     }
 
-    public void foo(@TestedOn(ints = {1, 2, 3})
-    int x) {
+    public void foo(@TestedOn(ints = {1, 2, 3}) int x) {
     }
 
     @Test
     public void getAnnotations() throws SecurityException,
             NoSuchMethodException {
-        Method method = ParameterSignatureTest.class.getMethod("foo", int.class);
+        Method method = getClass().getMethod("foo", int.class);
         List<Annotation> annotations = ParameterSignature.signatures(method)
                 .get(0).getAnnotations();
         assertThat(annotations,
                 CoreMatchers.<TestedOn>hasItem(isA(TestedOn.class)));
     }
+
+    public void intMethod(int param) {
+    }
+
+    public void integerMethod(Integer param) {
+    }
+
+    public void numberMethod(Number param) {
+    }
+
+    @Test
+    public void primitiveTypesShouldBeAcceptedAsWrapperTypes() throws Exception {
+        List<ParameterSignature> signatures = ParameterSignature
+                .signatures(getClass().getMethod("integerMethod", Integer.class));
+        ParameterSignature integerSignature = signatures.get(0);
+
+        assertTrue(integerSignature.canAcceptType(int.class));
+    }
+
+    @Test
+    public void primitiveTypesShouldBeAcceptedAsWrapperTypeAssignables() throws Exception {
+        List<ParameterSignature> signatures = ParameterSignature
+                .signatures(getClass().getMethod("numberMethod", Number.class));
+        ParameterSignature numberSignature = signatures.get(0);
+
+        assertTrue(numberSignature.canAcceptType(int.class));
+    }
+
+    @Test
+    public void wrapperTypesShouldBeAcceptedAsPrimitiveTypes() throws Exception {
+        List<ParameterSignature> signatures = ParameterSignature
+                .signatures(getClass().getMethod("intMethod", int.class));
+        ParameterSignature intSignature = signatures.get(0);
+
+        assertTrue(intSignature.canAcceptType(Integer.class));
+    }
 }

src/test/java/org/junit/tests/experimental/theories/TheoryTestUtils.java

@@ -4,7 +4,13 @@
 import java.util.List;
 
 import org.junit.experimental.theories.PotentialAssignment;
+import org.junit.experimental.theories.Theories;
 import org.junit.experimental.theories.internal.Assignments;
+import org.junit.runner.JUnitCore;
+import org.junit.runner.Request;
+import org.junit.runner.Result;
+import org.junit.runner.Runner;
+import org.junit.runners.model.InitializationError;
 import org.junit.runners.model.TestClass;
 
 public final class TheoryTestUtils {
@@ -17,5 +23,11 @@ public static List<PotentialAssignment> potentialAssignments(Method method)
                 new TestClass(method.getDeclaringClass()))
                 .potentialsForNextUnassigned();
     }
+
+    public static Result runTheoryClass(Class<?> testClass) throws InitializationError {
+        Runner theoryRunner = new Theories(testClass);
+        Request request = Request.runner(theoryRunner);
+        return new JUnitCore().run(request);
+    }
 
 }