D-Net

package eu.dnetlib.data.mdstore;

import com.google.common.collect.Lists;

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

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

import eu.dnetlib.data.mdstore.repository.TransactionInfoRepository;

import org.junit.Assert;

import org.junit.Before;

import org.junit.Test;

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

/**

 * Created by claudio on 30/03/2017.

 */

public class TransactionInfoTest extends AbstractMDStoreTest {

	@Autowired

	private TransactionInfoRepository transaction;

	@Before

	public void setUp() {

		transaction.deleteAll();

	}

	@Test

	public void testReadWrite() {

		transaction.save(ti("1", "1"));

		Assert.assertNotNull(transaction.findOne("1"));

		transaction.save(ti("1", "1"));

		Assert.assertNotNull(transaction.findOne("1"));

		transaction.save(ti("2", "2"));

		Assert.assertNotNull(transaction.findOne("1"));

		Assert.assertEquals(2, Lists.newArrayList(transaction.findAll()).size());

		Assert.assertEquals(0, transaction.findOne("1").getTransactions().size());

		transaction.addTransaction("1", tx("t1"));

		Assert.assertEquals(1, transaction.findOne("1").getTransactions().size());

	}

	private TransactionInfo ti(final String mdId, final String currentId) {

		return TransactionInfo.create()

				.setMdId(mdId)

				.setCurrentId(currentId);

	}

	private Transaction tx(final String id) {

		return Transaction.create()

				.setId(id)

				.setRefresh(true)

				.setSize(0);

	}

}