Add PdxInstanceWrapper class to wrap (decorate) existing PdxInstances and more appropriately handle JSON to Object mappings returned by getObject().

This commit is contained in:
John Blum
2020-05-12 19:01:17 -07:00
parent 22b01985b7
commit df466c0a62
2 changed files with 810 additions and 0 deletions

View File

@@ -0,0 +1,447 @@
/*
* Copyright 2020 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
* or implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
package org.springframework.geode.pdx;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyBoolean;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.atLeastOnce;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoInteractions;
import static org.mockito.Mockito.verifyNoMoreInteractions;
import java.io.DataOutput;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;
import java.util.Optional;
import com.fasterxml.jackson.core.JsonGenerationException;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.MapperFeature;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.junit.Test;
import org.apache.geode.internal.Sendable;
import org.apache.geode.pdx.JSONFormatter;
import org.apache.geode.pdx.PdxInstance;
import org.apache.geode.pdx.WritablePdxInstance;
/**
* Unit Tests for {@link PdxInstanceWrapper}.
*
* @author John Blum
* @see org.junit.Test
* @see org.mockito.Mockito
* @see com.fasterxml.jackson.core.JsonGenerator
* @see com.fasterxml.jackson.databind.ObjectMapper
* @see org.apache.geode.pdx.JSONFormatter
* @see org.apache.geode.pdx.PdxInstance
* @see org.apache.geode.pdx.WritablePdxInstance
* @since 1.3.0
*/
public class PdxInstanceWrapperUnitTests {
@Test
public void constructPdxInstanceWrapper() {
PdxInstance mockPdxInstance = mock(PdxInstance.class);
PdxInstanceWrapper wrapper = new PdxInstanceWrapper(mockPdxInstance);
assertThat(wrapper).isNotNull();
assertThat(wrapper.getDelegate()).isEqualTo(mockPdxInstance);
verifyNoInteractions(mockPdxInstance);
}
@Test(expected = IllegalArgumentException.class)
public void constructPdxInstanceWrapperWithNull() {
try {
new PdxInstanceWrapper(null);
}
catch (IllegalArgumentException expected) {
assertThat(expected).hasMessage("Argument must not be null");
assertThat(expected).hasNoCause();
throw expected;
}
}
@Test
public void fromObjectIsObject() {
assertThat(PdxInstanceWrapper.from("TEST")).isEqualTo("TEST");
}
@Test
public void fromNullIsNull() {
assertThat(PdxInstanceWrapper.from((Object) null)).isEqualTo(null);
}
@Test
public void fromPdxInstanceIsWrapper() {
PdxInstance mockPdxInstance = mock(PdxInstance.class);
PdxInstanceWrapper wrapper = PdxInstanceWrapper.from(mockPdxInstance);
assertThat(wrapper).isNotNull();
assertThat(wrapper.getDelegate()).isEqualTo(mockPdxInstance);
verifyNoInteractions(mockPdxInstance);
}
@Test
public void objectMapperConfigurationIsCorrect() {
PdxInstanceWrapper wrapper = spy(PdxInstanceWrapper.from(mock(PdxInstance.class)));
ObjectMapper mockObjectMapper = mock(ObjectMapper.class);
doReturn(mockObjectMapper).when(wrapper).newObjectMapper();
doReturn(mockObjectMapper).when(mockObjectMapper).configure(any(DeserializationFeature.class), anyBoolean());
doReturn(mockObjectMapper).when(mockObjectMapper).configure(any(MapperFeature.class), anyBoolean());
doReturn(mockObjectMapper).when(mockObjectMapper).findAndRegisterModules();
ObjectMapper objectMapper = wrapper.getObjectMapper().orElse(null);
assertThat(objectMapper).isNotNull();
verify(mockObjectMapper, times(1)).configure(eq(DeserializationFeature.FAIL_ON_IGNORED_PROPERTIES), eq(false));
verify(mockObjectMapper, times(1)).configure(eq(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES), eq(false));
verify(mockObjectMapper, times(1)).configure(eq(MapperFeature.ACCEPT_CASE_INSENSITIVE_ENUMS), eq(true));
verify(mockObjectMapper, times(1)).findAndRegisterModules();
verifyNoMoreInteractions(mockObjectMapper);
}
@Test
public void getClassNameCallsPdxInstanceGetClassName() {
PdxInstance mockPdxInstance = mock(PdxInstance.class);
doReturn("example.app.test.model.Type").when(mockPdxInstance).getClassName();
assertThat(PdxInstanceWrapper.from(mockPdxInstance).getClassName()).isEqualTo("example.app.test.model.Type");
verify(mockPdxInstance, times(1)).getClassName();
verifyNoMoreInteractions(mockPdxInstance);
}
@Test
public void isDeserializaleCallsPdxInstanceIsDeserializable() {
PdxInstance mockPdxInstance = mock(PdxInstance.class);
doReturn(true).when(mockPdxInstance).isDeserializable();
assertThat(PdxInstanceWrapper.from(mockPdxInstance).isDeserializable()).isTrue();
verify(mockPdxInstance, times(1)).isDeserializable();
verifyNoMoreInteractions(mockPdxInstance);
}
@Test
public void isEnumCallsPdxInstanceIsEnum() {
PdxInstance mockPdxInstance = mock(PdxInstance.class);
doReturn(false).when(mockPdxInstance).isEnum();
assertThat(PdxInstanceWrapper.from(mockPdxInstance).isEnum()).isFalse();
verify(mockPdxInstance, times(1)).isEnum();
verifyNoMoreInteractions(mockPdxInstance);
}
@Test
public void isIdentityFieldCallsPdxInstanceIsIdentityField() {
PdxInstance mockPdxInstance = mock(PdxInstance.class);
doReturn(false).when(mockPdxInstance).isIdentityField(anyString());
doReturn(true).when(mockPdxInstance).isIdentityField("id");
PdxInstanceWrapper wrapper = PdxInstanceWrapper.from(mockPdxInstance);
assertThat(wrapper.isIdentityField("id")).isTrue();
assertThat(wrapper.isIdentityField("randomField")).isFalse();
verify(mockPdxInstance, times(1)).isIdentityField(eq("id"));
verify(mockPdxInstance, times(1)).isIdentityField(eq("randomField"));
verifyNoMoreInteractions(mockPdxInstance);
}
@Test
public void getFieldCallPdxInstanceGetField() {
PdxInstance mockPdxInstance = mock(PdxInstance.class);
doReturn("TestValue").when(mockPdxInstance).getField(eq("TestField"));
assertThat(PdxInstanceWrapper.from(mockPdxInstance).getField("TestField")).isEqualTo("TestValue");
verify(mockPdxInstance, times(1)).getField(eq("TestField"));
verifyNoMoreInteractions(mockPdxInstance);
}
@Test
public void getFieldNameCallsPdxInstanceGetFieldNames() {
List<String> fieldNames = Arrays.asList("FieldOne", "FieldTwo");
PdxInstance mockPdxInstance = mock(PdxInstance.class);
doReturn(fieldNames).when(mockPdxInstance).getFieldNames();
assertThat(PdxInstanceWrapper.from(mockPdxInstance).getFieldNames()).isEqualTo(fieldNames);
verify(mockPdxInstance, times(1)).getFieldNames();
verifyNoMoreInteractions(mockPdxInstance);
}
@Test
public void getObjectReturnsObject() throws JsonProcessingException {
Account mockAccount = mock(Account.class);
ObjectMapper mockObjectMapper = mock(ObjectMapper.class);
PdxInstance mockPdxInstance = mock(PdxInstance.class);
String json = String.format("{ \"@type\": \"%s\", \"name\": \"Savings\"}", Account.class.getName());
doReturn(JSONFormatter.JSON_CLASSNAME).when(mockPdxInstance).getClassName();
doReturn(true).when(mockPdxInstance).hasField(eq(PdxInstanceWrapper.AT_TYPE_FIELD_NAME));
doReturn(Account.class.getName()).when(mockPdxInstance).getField(eq(PdxInstanceWrapper.AT_TYPE_FIELD_NAME));
PdxInstanceWrapper wrapper = spy(PdxInstanceWrapper.from(mockPdxInstance));
assertThat(wrapper).isNotNull();
assertThat(wrapper.getDelegate()).isEqualTo(mockPdxInstance);
doReturn(Optional.of(mockObjectMapper)).when(wrapper).getObjectMapper();
doReturn(json).when(wrapper).jsonFormatterToJson(eq(mockPdxInstance));
doReturn(mockAccount).when(mockObjectMapper).readValue(eq(json), eq(Account.class));
assertThat(wrapper.getObject()).isEqualTo(mockAccount);
verify(wrapper, atLeastOnce()).getDelegate();
verify(wrapper, times(1)).getObjectMapper();
verify(mockPdxInstance, times(1)).getClassName();
verify(mockPdxInstance, times(1)).hasField(eq(PdxInstanceWrapper.AT_TYPE_FIELD_NAME));
verify(mockPdxInstance, times(1)).getField(eq(PdxInstanceWrapper.AT_TYPE_FIELD_NAME));
verify(wrapper, times(1)).jsonFormatterToJson(eq(mockPdxInstance));
verify(mockObjectMapper, times(1)).readValue(eq(json), eq(Account.class));
verify(mockPdxInstance, never()).getObject();
verifyNoMoreInteractions(mockObjectMapper, mockPdxInstance);
}
@Test
@SuppressWarnings("unchecked")
public void getObjectCallsPdxInstanceGetObjectWhenAtTypeFieldIsNotPresent() throws JsonProcessingException {
Object value = new Object();
ObjectMapper mockObjectMapper = mock(ObjectMapper.class);
PdxInstance mockPdxInstance = mock(PdxInstance.class);
doReturn(JSONFormatter.JSON_CLASSNAME).when(mockPdxInstance).getClassName();
doReturn(false).when(mockPdxInstance).hasField(eq(PdxInstanceWrapper.AT_TYPE_FIELD_NAME));
doReturn(value).when(mockPdxInstance).getObject();
PdxInstanceWrapper wrapper = spy(PdxInstanceWrapper.from(mockPdxInstance));
assertThat(wrapper).isNotNull();
assertThat(wrapper.getDelegate()).isEqualTo(mockPdxInstance);
doReturn(Optional.of(mockObjectMapper)).when(wrapper).getObjectMapper();
assertThat(wrapper.getObject()).isEqualTo(value);
verify(wrapper, atLeastOnce()).getDelegate();
verify(wrapper, times(1)).getObjectMapper();
verify(mockPdxInstance, times(1)).getClassName();
verify(mockPdxInstance, times(1)).hasField(eq(PdxInstanceWrapper.AT_TYPE_FIELD_NAME));
verify(mockPdxInstance, never()).getField(anyString());
verify(wrapper, never()).jsonFormatterToJson(any());
verify(mockObjectMapper, never()).readValue(anyString(), any(Class.class));
verify(mockPdxInstance, times(1)).getObject();
verifyNoMoreInteractions(mockObjectMapper, mockPdxInstance);
}
@Test
@SuppressWarnings("unchecked")
public void getObjectCallsPdxInstanceGetObjectWhenClassNameIsNotGemFireJson() throws JsonProcessingException {
ObjectMapper mockObjectMapper = mock(ObjectMapper.class);
PdxInstance mockPdxInstance = mock(PdxInstance.class);
doReturn("non.existing.class.Name").when(mockPdxInstance).getClassName();
doReturn("TEST").when(mockPdxInstance).getObject();
PdxInstanceWrapper wrapper = spy(PdxInstanceWrapper.from(mockPdxInstance));
assertThat(wrapper).isNotNull();
assertThat(wrapper.getDelegate()).isEqualTo(mockPdxInstance);
doReturn(Optional.of(mockObjectMapper)).when(wrapper).getObjectMapper();
assertThat(wrapper.getObject()).isEqualTo("TEST");
verify(wrapper, atLeastOnce()).getDelegate();
verify(wrapper, times(1)).getObjectMapper();
verify(mockPdxInstance, times(1)).getClassName();
verify(mockPdxInstance, never()).hasField(anyString());
verify(mockPdxInstance, never()).getField(anyString());
verify(wrapper, never()).jsonFormatterToJson(any());
verify(mockObjectMapper, never()).readValue(anyString(), any(Class.class));
verify(mockPdxInstance, times(1)).getObject();
verifyNoMoreInteractions(mockObjectMapper, mockPdxInstance);
}
@Test
@SuppressWarnings("unchecked")
public void getObjectCallsPdxInstanceGetObjectWhenExceptionIsThrown() throws JsonProcessingException {
Object value = new Object();
ObjectMapper mockObjectMapper = mock(ObjectMapper.class);
PdxInstance mockPdxInstance = mock(PdxInstance.class);
String json = String.format("{ \"@type\": \"%s\", \"name\": \"Checking\"}", Account.class.getName());
doReturn(JSONFormatter.JSON_CLASSNAME).when(mockPdxInstance).getClassName();
doReturn(true).when(mockPdxInstance).hasField(eq(PdxInstanceWrapper.AT_TYPE_FIELD_NAME));
doReturn(Account.class.getName()).when(mockPdxInstance).getField(eq(PdxInstanceWrapper.AT_TYPE_FIELD_NAME));
doReturn(value).when(mockPdxInstance).getObject();
PdxInstanceWrapper wrapper = spy(PdxInstanceWrapper.from(mockPdxInstance));
assertThat(wrapper).isNotNull();
assertThat(wrapper.getDelegate()).isEqualTo(mockPdxInstance);
doReturn(Optional.of(mockObjectMapper)).when(wrapper).getObjectMapper();
doReturn(json).when(wrapper).jsonFormatterToJson(eq(mockPdxInstance));
doThrow(new JsonGenerationException("TEST", mock(JsonGenerator.class)))
.when(mockObjectMapper).readValue(anyString(), any(Class.class));
assertThat(wrapper.getObject()).isEqualTo(value);
verify(wrapper, times(1)).getObjectMapper();
verify(mockPdxInstance, times(1)).getClassName();
verify(mockPdxInstance, times(1)).hasField(eq(PdxInstanceWrapper.AT_TYPE_FIELD_NAME));
verify(mockPdxInstance, times(1)).getField(eq(PdxInstanceWrapper.AT_TYPE_FIELD_NAME));
verify(wrapper, atLeastOnce()).getDelegate();
verify(wrapper, times(1)).jsonFormatterToJson(eq(mockPdxInstance));
verify(mockObjectMapper, times(1)).readValue(eq(json), eq(Account.class));
verify(mockPdxInstance, times((1))).getObject();
verifyNoMoreInteractions(mockObjectMapper, mockPdxInstance);
}
@Test
public void getObjectCallsPdxInstanceGetObjectWhenObjectMapperIsNotPresent() {
PdxInstance mockPdxInstance = mock(PdxInstance.class);
doReturn("non.existing.class.Name").when(mockPdxInstance).getClassName();
doReturn("MOCK").when(mockPdxInstance).getObject();
PdxInstanceWrapper wrapper = spy(PdxInstanceWrapper.from(mockPdxInstance));
assertThat(wrapper).isNotNull();
assertThat(wrapper.getDelegate()).isEqualTo(mockPdxInstance);
doReturn(Optional.empty()).when(wrapper).getObjectMapper();
assertThat(wrapper.getObject()).isEqualTo("MOCK");
verify(wrapper, atLeastOnce()).getDelegate();
verify(wrapper, times(1)).getObjectMapper();
verify(wrapper, never()).jsonFormatterToJson(any());
verify(mockPdxInstance, times(1)).getObject();
verifyNoMoreInteractions(mockPdxInstance);
}
@Test
public void createWriterCallsPdxInstanceCreateWriter() {
PdxInstance mockPdxInstance = mock(PdxInstance.class);
WritablePdxInstance mockWritablePdxInstance = mock(WritablePdxInstance.class);
doReturn(mockWritablePdxInstance).when(mockPdxInstance).createWriter();
assertThat(PdxInstanceWrapper.from(mockPdxInstance).createWriter()).isEqualTo(mockWritablePdxInstance);
verify(mockPdxInstance, times(1)).createWriter();
verifyNoMoreInteractions(mockPdxInstance);
}
@Test
public void hasFieldCallsPdxInstanceHasField() {
PdxInstance mockPdxInstance = mock(PdxInstance.class);
doReturn(true).when(mockPdxInstance).hasField("name");
assertThat(PdxInstanceWrapper.from(mockPdxInstance).hasField("name")).isTrue();
verify(mockPdxInstance, times(1)).hasField(eq("name"));
verifyNoMoreInteractions(mockPdxInstance);
}
@Test
public void sendToCallsPdxInstanceSendTo() throws IOException {
SendablePdxInstance mockSendablePdxInstance = mock(SendablePdxInstance.class);
PdxInstanceWrapper wrapper = PdxInstanceWrapper.from(mockSendablePdxInstance);
DataOutput mockOut = mock(DataOutput.class);
assertThat(wrapper).isNotNull();
assertThat(wrapper.getDelegate()).isEqualTo(mockSendablePdxInstance);
wrapper.sendTo(mockOut);
verify(mockSendablePdxInstance, times(1)).sendTo(eq(mockOut));
verifyNoMoreInteractions(mockSendablePdxInstance);
verifyNoMoreInteractions(mockOut);
}
interface Account {
String getName();
}
interface SendablePdxInstance extends PdxInstance, Sendable { }
}