Fixes #441 : Change return type of Extractors.toStringMethod() to Ext…

…ractor<Object, String>
assertj · Jul 4, 2015 · 6464c05 · 6464c05
1 parent f588104
commit 6464c05
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/main/java/org/assertj/core/extractor/Extractors.java b/src/main/java/org/assertj/core/extractor/Extractors.java
@@ -32,7 +32,7 @@ public class Extractors {
   /**
    * Provides extractor for extracting {@link java.lang.Object#toString} from any object
    */
-  public static Extractor<?, String> toStringMethod() {
+  public static Extractor<Object, String> toStringMethod() {
     return new ToStringExtractor();
   }
 

diff --git a/src/test/java/org/assertj/core/api/iterable/IterableAssert_extracting_Test.java b/src/test/java/org/assertj/core/api/iterable/IterableAssert_extracting_Test.java
@@ -13,6 +13,7 @@
 package org.assertj.core.api.iterable;
 
 import org.assertj.core.api.AbstractIterableAssert;
+import org.assertj.core.extractor.Extractors;
 import org.assertj.core.groups.Tuple;
 import org.assertj.core.test.Employee;
 import org.assertj.core.test.ExpectedException;
@@ -170,5 +171,12 @@ public Tuple extract(Employee input) {
       }
     }).containsOnly(tuple("Yoda", 800, 1L), tuple("Luke", 26, 2L));
   }
+
+  @Test
+  public void should_allow_extracting_by_toString_method() {
+    assertThat(employees).extracting(Extractors.toStringMethod()).containsOnly(
+        "Employee[id=1, name=Name[first='Yoda', last='null'], age=800]",
+        "Employee[id=2, name=Name[first='Luke', last='Skywalker'], age=26]");
+  }
 
 }