use of com.google.cloud.spring.data.datastore.core.convert.TestItemWithEmbeddedEntity.EmbeddedEntity in project spring-cloud-gcp by GoogleCloudPlatform.
the class DefaultDatastoreEntityConverterTests method testEmbeddedEntity.
@Test
void testEmbeddedEntity() {
EmbeddedEntity embeddedEntityA = new EmbeddedEntity("item 0");
EmbeddedEntity embeddedEntityB = new EmbeddedEntity("item 1");
List<EmbeddedEntity> embeddedEntities = Arrays.asList(embeddedEntityA, embeddedEntityB);
Map<String, String> mapSimpleValues = new HashMap<>();
mapSimpleValues.put("a", "valueA");
mapSimpleValues.put("b", "valueB");
Map<String, String[]> mapListValues = new HashMap<>();
mapListValues.put("a", new String[] { "valueA" });
mapListValues.put("b", new String[] { "valueB" });
Map<String, EmbeddedEntity> embeddedEntityMapEmbeddedEntity = new HashMap<>();
embeddedEntityMapEmbeddedEntity.put("a", embeddedEntityA);
embeddedEntityMapEmbeddedEntity.put("b", embeddedEntityB);
Map<String, List<EmbeddedEntity>> embeddedEntityMapListOfEmbeddedEntities = new HashMap<>();
embeddedEntityMapListOfEmbeddedEntities.put("a", Arrays.asList(embeddedEntityA));
embeddedEntityMapListOfEmbeddedEntities.put("b", Arrays.asList(embeddedEntityB));
Map<String, Map<Long, Map<String, String>>> nestedEmbeddedMap = new HashMap<>();
Map<Long, Map<String, String>> nestedInnerEmbeddedMap = new HashMap<>();
nestedInnerEmbeddedMap.put(1L, mapSimpleValues);
nestedEmbeddedMap.put("outer1", nestedInnerEmbeddedMap);
Map<TestDatastoreItem.Color, String> enumKeysMap = new HashMap<>();
enumKeysMap.put(TestDatastoreItem.Color.BLACK, "black");
enumKeysMap.put(TestDatastoreItem.Color.WHITE, "white");
CustomMap customMap = new CustomMap();
customMap.put("key1", "val1");
TestItemWithEmbeddedEntity item = new TestItemWithEmbeddedEntity(123, new EmbeddedEntity("abc"), embeddedEntities, mapSimpleValues, mapListValues, embeddedEntityMapEmbeddedEntity, embeddedEntityMapListOfEmbeddedEntities, enumKeysMap, customMap);
item.setNestedEmbeddedMaps(nestedEmbeddedMap);
DatastoreEntityConverter entityConverter = new DefaultDatastoreEntityConverter(new DatastoreMappingContext(), new DatastoreServiceObjectToKeyFactory(() -> this.datastore));
Entity.Builder builder = getEntityBuilder();
entityConverter.write(item, builder);
Entity entity = builder.build();
assertThat(entity.getList("listOfEmbeddedEntities").stream().map(val -> ((BaseEntity<?>) val.get()).getString("stringField")).collect(Collectors.toList())).as("validate embedded entity").isEqualTo(Arrays.asList("item 0", "item 1"));
assertThat(entity.getEntity("embeddedEntityField").getString("stringField")).as("validate embedded entity").isEqualTo("abc");
assertThat(entity.getLong("intField")).as("validate int field").isEqualTo(123L);
assertThat(entity.getEntity("nestedEmbeddedMaps").getEntity("outer1").getEntity("1").getString("a")).isEqualTo("valueA");
assertThat(entity.getEntity("nestedEmbeddedMaps").getEntity("outer1").getEntity("1").getString("b")).isEqualTo("valueB");
assertThat(entity.getEntity("embeddedMapSimpleValues").getString("a")).isEqualTo("valueA");
assertThat(entity.getEntity("embeddedMapSimpleValues").getString("b")).isEqualTo("valueB");
assertThat(entity.getEntity("embeddedMapListOfValues").getList("a")).contains(StringValue.of("valueA"));
assertThat(entity.getEntity("embeddedMapListOfValues").getList("b")).contains(StringValue.of("valueB"));
assertThat(entity.getEntity("embeddedEntityMapEmbeddedEntity").getEntity("a").getString("stringField")).isEqualTo("item 0");
assertThat(entity.getEntity("embeddedEntityMapEmbeddedEntity").getEntity("b").getString("stringField")).isEqualTo("item 1");
List<Value> embeddedMapValuesEmbeddedEntityA = entity.getEntity("embeddedEntityMapListOfEmbeddedEntities").getList("a");
List<Value> embeddedMapValuesEmbeddedEntityB = entity.getEntity("embeddedEntityMapListOfEmbeddedEntities").getList("b");
assertThat(((BaseEntity) embeddedMapValuesEmbeddedEntityA.get(0).get()).getString("stringField")).isEqualTo("item 0");
assertThat(embeddedMapValuesEmbeddedEntityA).hasSize(1);
assertThat(((BaseEntity) embeddedMapValuesEmbeddedEntityB.get(0).get()).getString("stringField")).isEqualTo("item 1");
assertThat(embeddedMapValuesEmbeddedEntityB).hasSize(1);
TestItemWithEmbeddedEntity read = entityConverter.read(TestItemWithEmbeddedEntity.class, entity);
assertThat(read.getNestedEmbeddedMaps().get("outer1").get(1L)).containsEntry("a", "valueA");
assertThat(read.getNestedEmbeddedMaps().get("outer1").get(1L)).containsEntry("b", "valueB");
assertThat(entity.getEntity("customMap").getString("key1")).isEqualTo("val1");
assertThat(read).as("read objects equals the original one").isEqualTo(item);
}
Aggregations