Revision 46586
Added by Alessia Bardi almost 7 years ago
OAIFeedMapperTest.java | ||
---|---|---|
1 | 1 |
package eu.dnetlib.data.mapreduce.hbase.oai; |
2 | 2 |
|
3 |
import static org.junit.Assert.assertFalse; |
|
4 |
import static org.junit.Assert.assertNotNull; |
|
5 |
import static org.junit.Assert.assertTrue; |
|
6 |
|
|
7 | 3 |
import java.io.IOException; |
8 | 4 |
import java.text.ParseException; |
9 | 5 |
import java.util.Date; |
10 | 6 |
|
7 |
import com.google.common.collect.ArrayListMultimap; |
|
8 |
import com.google.common.collect.Lists; |
|
9 |
import com.google.common.collect.Multimap; |
|
10 |
import com.mongodb.DBObject; |
|
11 |
import com.mongodb.client.MongoCollection; |
|
12 |
import eu.dnetlib.data.mapreduce.hbase.oai.config.OAIConfiguration; |
|
13 |
import eu.dnetlib.data.mapreduce.hbase.oai.config.OAIConfigurationReader; |
|
14 |
import eu.dnetlib.data.mapreduce.hbase.oai.config.OAIConfigurationStringReader; |
|
15 |
import eu.dnetlib.data.mapreduce.hbase.oai.utils.MongoSetCollection; |
|
16 |
import eu.dnetlib.data.mapreduce.hbase.oai.utils.RecordFieldsExtractor; |
|
17 |
import eu.dnetlib.miscutils.datetime.DateUtils; |
|
11 | 18 |
import org.apache.commons.io.IOUtils; |
12 | 19 |
import org.apache.hadoop.mapreduce.Counter; |
13 | 20 |
import org.apache.solr.common.util.DateUtil; |
... | ... | |
21 | 28 |
import org.springframework.core.io.ClassPathResource; |
22 | 29 |
import org.springframework.core.io.Resource; |
23 | 30 |
|
24 |
import com.google.common.collect.ArrayListMultimap; |
|
25 |
import com.google.common.collect.Lists; |
|
26 |
import com.google.common.collect.Multimap; |
|
27 |
import com.mongodb.DBCollection; |
|
28 |
import com.mongodb.DBObject; |
|
31 |
import static org.junit.Assert.*; |
|
29 | 32 |
|
30 |
import eu.dnetlib.data.mapreduce.hbase.oai.config.OAIConfiguration; |
|
31 |
import eu.dnetlib.data.mapreduce.hbase.oai.config.OAIConfigurationReader; |
|
32 |
import eu.dnetlib.data.mapreduce.hbase.oai.config.OAIConfigurationStringReader; |
|
33 |
import eu.dnetlib.data.mapreduce.hbase.oai.utils.MongoSetCollection; |
|
34 |
import eu.dnetlib.data.mapreduce.hbase.oai.utils.RecordFieldsExtractor; |
|
35 |
import eu.dnetlib.miscutils.datetime.DateUtils; |
|
36 |
|
|
37 | 33 |
public class OAIFeedMapperTest { |
38 | 34 |
|
39 | 35 |
private OaiFeedMapper oaiFeedMapper; |
... | ... | |
47 | 43 |
@Mock |
48 | 44 |
private MongoSetCollection mongoSetCollection; |
49 | 45 |
@Mock |
50 |
private DBCollection discardedCollection;
|
|
46 |
private MongoCollection<DBObject> discardedCollection;
|
|
51 | 47 |
@SuppressWarnings("rawtypes") |
52 | 48 |
@Mock |
53 | 49 |
private org.apache.hadoop.mapreduce.Mapper.Context context; |
... | ... | |
93 | 89 |
Mockito.when(mongoSetCollection.normalizeSetSpec(Matchers.anyString())).thenReturn("aNormalisedSetName"); |
94 | 90 |
Mockito.doNothing().when(counter).increment(Matchers.anyLong()); |
95 | 91 |
Mockito.when(context.getCounter(Matchers.anyString(), Matchers.anyString())).thenReturn(counter); |
96 |
Mockito.when(discardedCollection.insert((DBObject) Matchers.any())).thenReturn(null); |
|
92 |
//TODO: fix mock |
|
93 |
//Mockito.when(discardedCollection.insertOne((DBObject) Matchers.any());).thenReturn(null); |
|
97 | 94 |
|
98 | 95 |
} |
99 | 96 |
|
Also available in: Unified diff
just to let the test compile