commit 98d08f2d11165c39bf37c9d9a7d7489ec3e60c13
author Jason Tedor <jasontedor@apache.org> Sat Mar 28 21:05:20 2015 -0400
committer Jason Tedor <jasontedor@apache.org> Sat Mar 28 21:05:20 2015 -0400
tree 6f548e864a0f66738ee8ab0d814cf454921198d6
parent c68c9e45aaf1d17a0e9b4c37a0f739e42def4842

MRUNIT-221: Wrong Generic for setKeyGroupingComparator/setKeyOrderComparator in MultipleInputsMapReduceDriverBase (Contributed by Mathias Rüdiger via Jason Tedor)

src/main/java/org/apache/hadoop/mrunit/internal/driver/MultipleInputsMapReduceDriverBase.java

diff --git a/src/main/java/org/apache/hadoop/mrunit/internal/driver/MultipleInputsMapReduceDriverBase.java b/src/main/java/org/apache/hadoop/mrunit/internal/driver/MultipleInputsMapReduceDriverBase.java
index 726aaeb..bac9a80 100644
--- a/src/main/java/org/apache/hadoop/mrunit/internal/driver/MultipleInputsMapReduceDriverBase.java
+++ b/src/main/java/org/apache/hadoop/mrunit/internal/driver/MultipleInputsMapReduceDriverBase.java
@@ -113,7 +113,7 @@
    */
   @SuppressWarnings("unchecked")
   public void setKeyGroupingComparator(
-      final RawComparator<K2> groupingComparator) {
+      final RawComparator<K1> groupingComparator) {
     keyGroupComparator = ReflectionUtils.newInstance(
         groupingComparator.getClass(), getConfiguration());
   }
@@ -126,7 +126,7 @@
    *          Comparator to use in the shuffle stage for key grouping
    * @return this
    */
-  public T withKeyGroupingComparator(final RawComparator<K2> groupingComparator) {
+  public T withKeyGroupingComparator(final RawComparator<K1> groupingComparator) {
     setKeyGroupingComparator(groupingComparator);
     return thisAsMapReduceDriver();
   }
@@ -149,7 +149,7 @@
    * @param orderComparator
    */
   @SuppressWarnings("unchecked")
-  public void setKeyOrderComparator(final RawComparator<K2> orderComparator) {
+  public void setKeyOrderComparator(final RawComparator<K1> orderComparator) {
     keyValueOrderComparator = ReflectionUtils.newInstance(
         orderComparator.getClass(), getConfiguration());
   }
@@ -162,7 +162,7 @@
    *          Comparator to use in the shuffle stage for key value ordering
    * @return this
    */
-  public T withKeyOrderComparator(final RawComparator<K2> orderComparator) {
+  public T withKeyOrderComparator(final RawComparator<K1> orderComparator) {
     setKeyOrderComparator(orderComparator);
     return thisAsMapReduceDriver();
   }