1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.hadoop.hbase.mapred;
20
21 import static org.mockito.Mockito.mock;
22 import static org.mockito.Mockito.times;
23 import static org.mockito.Mockito.verify;
24
25 import java.io.IOException;
26
27 import org.apache.hadoop.hbase.SmallTests;
28 import org.apache.hadoop.hbase.client.Result;
29 import org.apache.hadoop.hbase.io.ImmutableBytesWritable;
30 import org.apache.hadoop.mapred.OutputCollector;
31 import org.apache.hadoop.mapred.Reporter;
32 import org.junit.Test;
33 import org.junit.experimental.categories.Category;
34 import org.mockito.Mockito;
35
36 @Category(SmallTests.class)
37 public class TestIdentityTableMap {
38
39 @Test
40 @SuppressWarnings({ "deprecation", "unchecked" })
41 public void shouldCollectPredefinedTimes() throws IOException {
42 int recordNumber = 999;
43 Result resultMock = mock(Result.class);
44 IdentityTableMap identityTableMap = null;
45 try {
46 Reporter reporterMock = mock(Reporter.class);
47 identityTableMap = new IdentityTableMap();
48 ImmutableBytesWritable bytesWritableMock = mock(ImmutableBytesWritable.class);
49 OutputCollector<ImmutableBytesWritable, Result> outputCollectorMock =
50 mock(OutputCollector.class);
51
52 for (int i = 0; i < recordNumber; i++)
53 identityTableMap.map(bytesWritableMock, resultMock, outputCollectorMock,
54 reporterMock);
55
56 verify(outputCollectorMock, times(recordNumber)).collect(
57 Mockito.any(ImmutableBytesWritable.class), Mockito.any(Result.class));
58 } finally {
59 if (identityTableMap != null)
60 identityTableMap.close();
61 }
62 }
63 }