Update tests to TaskAttemptContextImpl (Contributed by Henning)

kambatla · Aug 20, 2012 · 501b976 · 501b976
1 parent 2182b3b
commit 501b976
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/test/com/hadoop/mapreduce/TestLzoTextInputFormat.java b/src/test/com/hadoop/mapreduce/TestLzoTextInputFormat.java
@@ -43,6 +43,7 @@
 import org.apache.hadoop.mapreduce.RecordWriter;
 import org.apache.hadoop.mapreduce.TaskAttemptContext;
 import org.apache.hadoop.mapreduce.TaskAttemptID;
+import org.apache.hadoop.mapreduce.task.TaskAttemptContextImpl;
 import org.apache.hadoop.mapreduce.lib.input.FileSplit;
 import org.apache.hadoop.mapreduce.lib.input.TextInputFormat;
 import org.apache.hadoop.mapreduce.lib.output.TextOutputFormat;
@@ -165,8 +166,8 @@ private void runTest(boolean testWithIndex, int charsToOutput) throws IOExceptio
     TextOutputFormat.setOutputCompressorClass(job, LzopCodec.class);
     TextOutputFormat.setOutputPath(job, outputDir);
 
-    TaskAttemptContext attemptContext = new TaskAttemptContext(job.getConfiguration(),
-        new TaskAttemptID("123", 0, false, 1, 2));
+    TaskAttemptContext attemptContext = new TaskAttemptContextImpl(
+        job.getConfiguration(), new TaskAttemptID("123", 0, false, 1, 2));
 
     // create some input data
     byte[] expectedMd5 = createTestInput(outputDir, localFs, attemptContext, charsToOutput);
@@ -312,8 +313,8 @@ private void runTestIgnoreNonLzo(boolean testWithIndex, int charsToOutput,
     TextOutputFormat.setOutputCompressorClass(job, LzopCodec.class);
     TextOutputFormat.setOutputPath(job, outputDir);
 
-    TaskAttemptContext attemptContext = new TaskAttemptContext(job.getConfiguration(),
-        new TaskAttemptID("123", 0, false, 1, 2));
+    TaskAttemptContext attemptContext = new TaskAttemptContextImpl(
+        job.getConfiguration(), new TaskAttemptID("123", 0, false, 1, 2));
 
     // create some input data
     byte[] expectedMd5 = createTestInput(outputDir, localFs, attemptContext, charsToOutput);