Skip to content
This repository was archived by the owner on May 12, 2021. It is now read-only.

#186 method context.getNumReduceTasks() does not work #1

Open
lukasnalezenec wants to merge 2 commits into
base: trunk
Choose a base branch
from
Open

#186 method context.getNumReduceTasks() does not work #1

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
6876f84
#186 method context.getNumReduceTasks() does not work
lukasnalezenec Dec 28, 2013
6140104
Deep Mocks
lukasnalezenec Feb 17, 2014
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 0 additions & 2 deletions src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/AbstractMockContextWrapper.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -131,15 +131,13 @@ public Path[] answer(InvocationOnMock invocation) throws Throwable {
return DistributedCache.getLocalCacheFiles(configuration);
}
});

when(context.getNumReduceTasks()).thenAnswer(new Answer<Integer>() {
@Override
@SuppressWarnings("deprecation")
public Integer answer(InvocationOnMock invocation) throws Throwable {
return configuration.getInt("mapred.reduce.tasks", 1);
}
});

} catch (IOException e) {
throw new RuntimeException(e);
} catch (InterruptedException e) {
Expand Down
3 changes: 2 additions & 1 deletion src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/MockMapContextWrapper.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@
import org.apache.hadoop.mrunit.internal.output.MockOutputCreator;
import org.apache.hadoop.mrunit.mapreduce.MapDriver;
import org.apache.hadoop.mrunit.types.Pair;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

Expand Down Expand Up @@ -71,7 +72,7 @@ public MockMapContextWrapper(
@Override
@SuppressWarnings({ "unchecked" })
protected Mapper<KEYIN, VALUEIN, KEYOUT, VALUEOUT>.Context create() {
final Mapper<KEYIN, VALUEIN, KEYOUT, VALUEOUT>.Context context = mock(org.apache.hadoop.mapreduce.Mapper.Context.class);
final Mapper<KEYIN, VALUEIN, KEYOUT, VALUEOUT>.Context context = mock(org.apache.hadoop.mapreduce.Mapper.Context.class, Mockito.RETURNS_DEEP_STUBS);

createCommon(context, driver, mockOutputCreator);

Expand Down
3 changes: 2 additions & 1 deletion src/main/java/org/apache/hadoop/mrunit/internal/mapreduce/MockReduceContextWrapper.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@
import org.apache.hadoop.mrunit.internal.output.MockOutputCreator;
import org.apache.hadoop.mrunit.mapreduce.ReduceDriver;
import org.apache.hadoop.mrunit.types.KeyValueReuseList;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

Expand Down Expand Up @@ -80,7 +81,7 @@ public MockReduceContextWrapper(
@SuppressWarnings({ "unchecked" })
protected Reducer<KEYIN, VALUEIN, KEYOUT, VALUEOUT>.Context create() {

final Reducer<KEYIN, VALUEIN, KEYOUT, VALUEOUT>.Context context = mock(org.apache.hadoop.mapreduce.Reducer.Context.class);
final Reducer<KEYIN, VALUEIN, KEYOUT, VALUEOUT>.Context context = mock(org.apache.hadoop.mapreduce.Reducer.Context.class, Mockito.RETURNS_DEEP_STUBS);

createCommon(context, driver, mockOutputCreator);
try {
Expand Down