Project

General

Profile

1
package eu.dnetlib.data.mdstore.modular.mongodb;
2

    
3
import java.util.Map;
4

    
5
import com.google.common.collect.Maps;
6
import com.mongodb.DBObject;
7
import com.mongodb.WriteConcern;
8
import com.mongodb.client.MongoCollection;
9
import org.junit.Before;
10
import org.junit.Test;
11
import org.junit.runner.RunWith;
12
import org.mockito.Mock;
13
import org.mockito.MockitoAnnotations;
14
import org.mockito.junit.MockitoJUnitRunner;
15

    
16
import static org.mockito.Mockito.when;
17

    
18
/**
19
 * Created by Alessia Bardi on 2019-04-12.
20
 *
21
 * @author Alessia Bardi
22
 */
23
@RunWith(MockitoJUnitRunner.class)
24
public class MongoBulkWritesManagerTest {
25

    
26
	private MongoBulkWritesManager mng;
27

    
28
	@Mock
29
	MongoCollection<DBObject> coll;
30

    
31
	@Before
32
	public void setUp(){
33
		MockitoAnnotations.initMocks(this);
34
		when(coll.withWriteConcern(WriteConcern.ACKNOWLEDGED)).thenReturn(coll);
35
		mng = new MongoBulkWritesManager(coll, coll, null, 10, null, false);
36
	}
37

    
38
	@Test
39
	public void buildDBObjectTest(){
40
		Map<String, String> props = Maps.newHashMap();
41
		props.put("timestamp", "1555078665140");
42
		props.put("id", "od______4301::5af4702a60ddf0615fd1dfd6ded104df");
43
		props.put("originalId", "x");
44
		props.put("body","<body/>");
45

    
46
		DBObject obj = mng.buildDBObject("<x/>", props, null);
47
		System.out.println(obj);
48
	}
49
}
(6-6/6)