D-Net

package eu.dnetlib.data.mdstore;

import java.util.List;

import com.google.common.collect.Lists;

import com.mongodb.BasicDBObject;

import com.mongodb.client.MongoCollection;

import com.mongodb.client.model.Filters;

import eu.dnetlib.data.mdstore.model.MDStoreManagerInfo;

import org.apache.commons.lang3.StringUtils;

import org.bson.Document;

import org.junit.Before;

import org.junit.Test;

import org.junit.Test.None;

import org.springframework.beans.factory.annotation.Autowired;

import static eu.dnetlib.data.mdstore.MDStoreConstants.*;

import static org.junit.Assert.*;

/**

 * Created by claudio on 22/03/2017.

 */

public class MDStoreTransactionManagerTest extends AbstractMDStoreTest {

	@Autowired

	private MDStoreTransactionManager transactionManager;

	@Before

	public void setUp() throws MDStoreException {

		getCollection(METADATA_NAME, MDStoreManagerInfo.class).deleteMany(new BasicDBObject());

		getCollection(METADATA_MANAGER_TABLE, MDStoreManagerInfo.class).deleteMany(new BasicDBObject());

	}

	@Test(expected = None.class)

	public void testInit() throws MDStoreException {

		final MongoCollection<MDStoreManagerInfo> managerTable = getCollection(METADATA_MANAGER_TABLE, MDStoreManagerInfo.class);

		final List<Document> indexes = Lists.newArrayList(managerTable.listIndexes());

		assertEquals(2, indexes.size());

		assertTrue(indexes.stream().anyMatch(idx -> ((Document) idx.get("key")).containsKey(MDID)));

		transactionManager.verifyConsistency();

	}

	@Test

	public void testCreateDrop() throws MDStoreException {

		final MongoCollection<MDStoreManagerInfo> managerTable = getCollection(METADATA_MANAGER_TABLE, MDStoreManagerInfo.class);

		final String mdId = mockMDID();

		final String internalId = transactionManager.createMDStore(mdId);

		assertTrue(StringUtils.isNotBlank(internalId));

		final MDStoreManagerInfo find = managerTable.find(Filters.eq(CURRENT_ID, internalId)).first();

		assertNotNull(find);

		assertEquals(internalId, find.getCurrentId());

		transactionManager.dropMDStore(mdId);

		assertNull(managerTable.find(Filters.eq(MDID, mdId)).first());

	}

	@Test

	public void testTransaction() throws MDStoreException {

		final String mdId = mockMDID();

		transactionManager.createMDStore(mdId);

		final String transactionId = transactionManager.startTransaction(mdId, true);

		final MongoCollection<MDStoreManagerInfo> managerTable = getCollection(METADATA_MANAGER_TABLE, MDStoreManagerInfo.class);

		final MDStoreManagerInfo find = managerTable.find(Filters.eq(MDID, mdId)).first();

		assertEquals(1, find.getTransactions().size());

		assertTrue(find.getTransactions().stream().anyMatch(t -> transactionId.equals(t.getId())));

	}

}