SGF-448 - GemfireRepositoryConfigurationExtension needs to override the RepositoryConfigurationExtensionSupport postProcess(:BeanDefinitionBuilder, :AnnotationRepositoryConfigurationSource) method.

(cherry picked from commit cf64807b095637fea43ec00acfaa1204cdedd8cb) Signed-off-by: John Blum <jblum@pivotal.io>
2015-12-03 15:59:49 -08:00
parent 251199a5cc
commit c7b45af48e
3 changed files with 171 additions and 13 deletions
--- a/src/test/java/org/springframework/data/gemfire/repository/config/GemfireRepositoryConfigurationExtensionTest.java
+++ b/src/test/java/org/springframework/data/gemfire/repository/config/GemfireRepositoryConfigurationExtensionTest.java
@@ -0,0 +1,139 @@
+package org.springframework.data.gemfire.repository.config;
+
+import static org.hamcrest.CoreMatchers.equalTo;
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.hamcrest.CoreMatchers.is;
+import static org.hamcrest.CoreMatchers.nullValue;
+import static org.junit.Assert.assertThat;
+import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.springframework.beans.factory.config.RuntimeBeanReference;
+import org.springframework.beans.factory.support.BeanDefinitionBuilder;
+import org.springframework.data.gemfire.repository.support.GemfireRepositoryFactoryBean;
+import org.springframework.data.repository.config.AnnotationRepositoryConfigurationSource;
+import org.springframework.data.repository.config.XmlRepositoryConfigurationSource;
+import org.w3c.dom.Element;
+
+/**
+ * The GemfireRepositoryConfigurationExtensionTest class is a test suite of test cases testing the contract
+ * and functionality of the GemfireRepositoryConfigurationExtension class.
+ *
+ * @author John Blum
+ * @see org.junit.Test
+ * @see org.mockito.Mockito
+ * @see org.springframework.data.gemfire.repository.config.GemfireRepositoryConfigurationExtension
+ * @since 1.6.3
+ */
+public class GemfireRepositoryConfigurationExtensionTest {
+
+	private GemfireRepositoryConfigurationExtension repositoryConfigurationExtension;
+
+	@Before
+	public void setup() {
+		repositoryConfigurationExtension = new GemfireRepositoryConfigurationExtension();
+	}
+
+	@Test
+	public void repositoryFactoryClassNameIsGemfireRepositoryFactoryBean() {
+		assertThat(repositoryConfigurationExtension.getRepositoryFactoryClassName(),
+			is(equalTo(GemfireRepositoryFactoryBean.class.getName())));
+	}
+
+	@Test
+	public void modulePrefixIsGemFire() {
+		assertThat(repositoryConfigurationExtension.getModulePrefix(), is(equalTo("gemfire")));
+	}
+
+	@Test
+	public void postProcessWithAnnotationRepositoryConfigurationSource() {
+		AnnotationRepositoryConfigurationSource mockRepositoryConfigurationSource =
+			mock(AnnotationRepositoryConfigurationSource.class);
+
+		when(mockRepositoryConfigurationSource.getAttribute(eq("mappingContextRef"))).thenReturn("testMappingContext");
+
+		BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition();
+
+		repositoryConfigurationExtension.postProcess(beanDefinitionBuilder, mockRepositoryConfigurationSource);
+
+		Object mappingContextRef = beanDefinitionBuilder.getRawBeanDefinition().getPropertyValues()
+			.getPropertyValue("gemfireMappingContext").getValue();
+
+		assertThat(mappingContextRef, is(instanceOf(RuntimeBeanReference.class)));
+		assertThat(((RuntimeBeanReference) mappingContextRef).getBeanName(), is(equalTo("testMappingContext")));
+
+		verify(mockRepositoryConfigurationSource, times(1)).getAttribute(eq("mappingContextRef"));
+	}
+
+	@Test
+	public void postProcessWithAnnotationRepositoryConfigurationSourceHaingNoMappingContextRefAttribute() {
+		AnnotationRepositoryConfigurationSource mockRepositoryConfigurationSource =
+			mock(AnnotationRepositoryConfigurationSource.class);
+
+		when(mockRepositoryConfigurationSource.getAttribute(eq("mappingContextRef"))).thenReturn("  ");
+
+		BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition();
+
+		repositoryConfigurationExtension.postProcess(beanDefinitionBuilder, mockRepositoryConfigurationSource);
+
+		Object mappingContextRef = beanDefinitionBuilder.getRawBeanDefinition().getPropertyValues()
+			.getPropertyValue("gemfireMappingContext");
+
+		assertThat(mappingContextRef, is(nullValue()));
+
+		verify(mockRepositoryConfigurationSource, times(1)).getAttribute(eq("mappingContextRef"));
+	}
+
+	@Test
+	public void postProcessWithXmlRepositoryConfigurationSourceHavingNoMappingContextRefAttribute() {
+		Element mockElement = mock(Element.class);
+
+		XmlRepositoryConfigurationSource mockRepositoryConfigurationSource =
+			mock(XmlRepositoryConfigurationSource.class);
+
+		when(mockRepositoryConfigurationSource.getElement()).thenReturn(mockElement);
+		when(mockElement.getAttribute(eq("mapping-context-ref"))).thenReturn(null);
+
+		BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition();
+
+		repositoryConfigurationExtension.postProcess(beanDefinitionBuilder, mockRepositoryConfigurationSource);
+
+		Object mappingContextRef = beanDefinitionBuilder.getRawBeanDefinition().getPropertyValues()
+			.getPropertyValue("gemfireMappingContext");
+
+		assertThat(mappingContextRef, is(nullValue()));
+
+		verify(mockRepositoryConfigurationSource, times(1)).getElement();
+		verify(mockElement, times(1)).getAttribute(eq("mapping-context-ref"));
+	}
+
+	@Test
+	public void postProcessWithXmlRepositoryConfigurationSourceWith() {
+		Element mockElement = mock(Element.class);
+
+		XmlRepositoryConfigurationSource mockRepositoryConfigurationSource =
+			mock(XmlRepositoryConfigurationSource.class);
+
+		when(mockRepositoryConfigurationSource.getElement()).thenReturn(mockElement);
+		when(mockElement.getAttribute(eq("mapping-context-ref"))).thenReturn("testMappingContext");
+
+		BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition();
+
+		repositoryConfigurationExtension.postProcess(beanDefinitionBuilder, mockRepositoryConfigurationSource);
+
+		Object mappingContextRef = beanDefinitionBuilder.getRawBeanDefinition().getPropertyValues()
+			.getPropertyValue("gemfireMappingContext").getValue();
+
+		assertThat(mappingContextRef, is(instanceOf(RuntimeBeanReference.class)));
+		assertThat(((RuntimeBeanReference) mappingContextRef).getBeanName(), is(equalTo("testMappingContext")));
+
+		verify(mockRepositoryConfigurationSource, times(1)).getElement();
+		verify(mockElement, times(1)).getAttribute(eq("mapping-context-ref"));
+	}
+
+}