use of org.springframework.cloud.gcp.data.datastore.entities.CustomMap in project spring-cloud-gcp by spring-cloud.
the class DefaultDatastoreEntityConverterTests method testEmbeddedEntity.
@Test
public 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.size()).isEqualTo(1);
assertThat(((BaseEntity) embeddedMapValuesEmbeddedEntityB.get(0).get()).getString("stringField")).isEqualTo("item 1");
assertThat(embeddedMapValuesEmbeddedEntityB.size()).isEqualTo(1);
TestItemWithEmbeddedEntity read = entityConverter.read(TestItemWithEmbeddedEntity.class, entity);
assertThat(read.getNestedEmbeddedMaps().get("outer1").get(1L).get("a")).isEqualTo("valueA");
assertThat(read.getNestedEmbeddedMaps().get("outer1").get(1L).get("b")).isEqualTo("valueB");
assertThat(entity.getEntity("customMap").getString("key1")).isEqualTo("val1");
assertThat(read).as("read objects equals the original one").isEqualTo(item);
}
use of org.springframework.cloud.gcp.data.datastore.entities.CustomMap in project spring-cloud-gcp by spring-cloud.
the class Employee method mapSubclass.
@Test
public void mapSubclass() {
CustomMap customMap1 = new CustomMap();
customMap1.put("key1", "val1");
ServiceConfiguration service1 = new ServiceConfiguration("service1", customMap1);
CustomMap customMap2 = new CustomMap();
customMap2.put("key2", "val2");
ServiceConfiguration service2 = new ServiceConfiguration("service2", customMap2);
this.datastoreTemplate.saveAll(Arrays.asList(service1, service2));
waitUntilTrue(() -> this.datastoreTemplate.count(ServiceConfiguration.class) == 2);
Collection<ServiceConfiguration> events = this.datastoreTemplate.findAll(ServiceConfiguration.class);
assertThat(events).containsExactlyInAnyOrder(service1, service2);
}
Aggregations