MRUNIT-163: getTaskAttemptID is missing from context
Fix for hadoop2 as interface of TaskAttemptID has been changed
diff --git a/src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/MockMapContextWrapper.java b/src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/MockMapContextWrapper.java
index b61f8e6..0febe85 100644
--- a/src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/MockMapContextWrapper.java
+++ b/src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/MockMapContextWrapper.java
@@ -105,7 +105,7 @@
@Override
public TaskAttemptID answer(InvocationOnMock invocation)
throws Throwable {
- return new TaskAttemptID("", 0, true, 0, 0);
+ return TaskAttemptID.forName("attempt__0000_m_000000_0");
}
});
} catch (IOException e) {
diff --git a/src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/MockReduceContextWrapper.java b/src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/MockReduceContextWrapper.java
index dbd06e8..4996074 100644
--- a/src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/MockReduceContextWrapper.java
+++ b/src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/MockReduceContextWrapper.java
@@ -105,7 +105,7 @@
@Override
public TaskAttemptID answer(InvocationOnMock invocation)
throws Throwable {
- return new TaskAttemptID("", 0, false, 0, 0);
+ return TaskAttemptID.forName("attempt__0000_r_000000_0");
}
});
} catch (IOException e) {