Project

General

Profile

1 55329 alessia.ba
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
}