AerospikeRepositoryFactoryTest.java

/**
 * 
 */
package org.springframework.data.aerospike.repository.support;

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.CoreMatchers.notNullValue;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.io.Serializable;

import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.runners.MockitoJUnitRunner;
import org.springframework.context.ApplicationContext;
import org.springframework.data.aerospike.convert.AerospikeConverter;
import org.springframework.data.aerospike.core.AerospikeOperations;
import org.springframework.data.aerospike.core.AerospikeTemplate;
import org.springframework.data.aerospike.core.Person;
import org.springframework.data.aerospike.mapping.AerospikePersistentEntity;
import org.springframework.data.keyvalue.repository.support.SimpleKeyValueRepository;
import org.springframework.data.mapping.context.MappingContext;
import org.springframework.data.repository.core.EntityInformation;
import org.springframework.data.repository.core.RepositoryInformation;
import org.springframework.data.repository.core.RepositoryMetadata;
import org.springframework.data.repository.core.support.PersistentEntityInformation;

/**
 *
 *
 * @author Peter Milne
 * @author Jean Mercier
 *
 */
@RunWith(MockitoJUnitRunner.class)
public class AerospikeRepositoryFactoryTest {

	@Mock ApplicationContext applicationContext;
	@Mock AerospikeTemplate template;
	@Mock RepositoryInformation repositoryInformation;
	@Mock AerospikeConverter converter;
	@SuppressWarnings("rawtypes")
	@Mock MappingContext context;
	@Mock AerospikeRepositoryFactory aerospikeRepositoryFactoryMock;
	@SuppressWarnings("rawtypes")
	@Mock AerospikePersistentEntity entity;
	@Mock AerospikeOperations aerospikeOperations;

	@Rule public ExpectedException exception = ExpectedException.none();

	/**
	 * @throws java.lang.Exception
	 */
	@SuppressWarnings("unchecked")
	@Before
	public void setUp() throws Exception {
			when(aerospikeOperations.getMappingContext()).thenReturn(context);

	}

	/**
	 * @throws java.lang.Exception
	 */
	@After
	public void tearDown() throws Exception {
	}

	/**
	 * Test method for {@link org.springframework.data.aerospike.repository.support.AerospikeRepositoryFactory#getEntityInformation(java.lang.Class)}.
	 */
	@SuppressWarnings("unchecked")
	@Test
	public void testGetEntityInformationClassOfT() {
		when(context.getPersistentEntity(Person.class)).thenReturn(entity);
		when(entity.getType()).thenReturn(Person.class);

		AerospikeRepositoryFactory factory = new AerospikeRepositoryFactory(aerospikeOperations);
		EntityInformation<Person, Serializable> entityInformation = factory.getEntityInformation(Person.class);
		assertTrue(entityInformation instanceof PersistentEntityInformation);
	}

	/**
	 * Test method for {@link org.springframework.data.aerospike.repository.support.AerospikeRepositoryFactory#getTargetRepository(org.springframework.data.repository.core.RepositoryInformation)}.
	 */
	@SuppressWarnings("unchecked")
	@Test
	public void testGetTargetRepositoryRepositoryInformation() {
		when(context.getPersistentEntity(Person.class)).thenReturn(entity);
		when(entity.getType()).thenReturn(Person.class);
		Mockito.<Class<?>>when(repositoryInformation.getDomainType()).thenReturn(Person.class);
		Mockito.<Class<?>>when(repositoryInformation.getRepositoryBaseClass()).thenReturn(Person.class);
		when(aerospikeRepositoryFactoryMock.getTargetRepository(repositoryInformation)).thenReturn(new Object());

		Person.class.getDeclaredConstructors();

			Object repository = aerospikeRepositoryFactoryMock.getTargetRepository(repositoryInformation);
		assertThat(repository, is(notNullValue()));
	}

	/**
	 * Test method for {@link org.springframework.data.aerospike.repository.support.AerospikeRepositoryFactory#getRepositoryBaseClass(org.springframework.data.repository.core.RepositoryMetadata)}.
	 */
	@Test
	public void testGetRepositoryBaseClassRepositoryMetadata() {
		RepositoryMetadata metadata = mock(RepositoryMetadata.class);
		Mockito.<Class<?>>when(metadata.getRepositoryInterface()).thenReturn(SimpleKeyValueRepository.class);
		AerospikeRepositoryFactory factory = new AerospikeRepositoryFactory(aerospikeOperations);
		Class<?> repbaseClass = factory.getRepositoryBaseClass(metadata);
		assertTrue(repbaseClass.getSimpleName().equals(SimpleKeyValueRepository.class.getSimpleName()));
	}

}