Search in sources :

Example 1 with CustomMap

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);
}
Also used : FullEntity(com.google.cloud.datastore.FullEntity) Entity(com.google.cloud.datastore.Entity) EmbeddedEntity(org.springframework.cloud.gcp.data.datastore.core.convert.TestItemWithEmbeddedEntity.EmbeddedEntity) BaseEntity(com.google.cloud.datastore.BaseEntity) DatastoreMappingContext(org.springframework.cloud.gcp.data.datastore.core.mapping.DatastoreMappingContext) HashMap(java.util.HashMap) CustomMap(org.springframework.cloud.gcp.data.datastore.entities.CustomMap) ArrayList(java.util.ArrayList) List(java.util.List) BaseEntity(com.google.cloud.datastore.BaseEntity) EmbeddedEntity(org.springframework.cloud.gcp.data.datastore.core.convert.TestItemWithEmbeddedEntity.EmbeddedEntity) DiscriminatorValue(org.springframework.cloud.gcp.data.datastore.core.mapping.DiscriminatorValue) NullValue(com.google.cloud.datastore.NullValue) EntityValue(com.google.cloud.datastore.EntityValue) Value(com.google.cloud.datastore.Value) StringValue(com.google.cloud.datastore.StringValue) ListValue(com.google.cloud.datastore.ListValue) HashMap(java.util.HashMap) Map(java.util.Map) CustomMap(org.springframework.cloud.gcp.data.datastore.entities.CustomMap) Test(org.junit.Test)

Example 2 with CustomMap

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);
}
Also used : ServiceConfiguration(org.springframework.cloud.gcp.data.datastore.entities.ServiceConfiguration) CustomMap(org.springframework.cloud.gcp.data.datastore.entities.CustomMap) Test(org.junit.Test)

Aggregations

Test (org.junit.Test)2 CustomMap (org.springframework.cloud.gcp.data.datastore.entities.CustomMap)2 BaseEntity (com.google.cloud.datastore.BaseEntity)1 Entity (com.google.cloud.datastore.Entity)1 EntityValue (com.google.cloud.datastore.EntityValue)1 FullEntity (com.google.cloud.datastore.FullEntity)1 ListValue (com.google.cloud.datastore.ListValue)1 NullValue (com.google.cloud.datastore.NullValue)1 StringValue (com.google.cloud.datastore.StringValue)1 Value (com.google.cloud.datastore.Value)1 ArrayList (java.util.ArrayList)1 HashMap (java.util.HashMap)1 List (java.util.List)1 Map (java.util.Map)1 EmbeddedEntity (org.springframework.cloud.gcp.data.datastore.core.convert.TestItemWithEmbeddedEntity.EmbeddedEntity)1 DatastoreMappingContext (org.springframework.cloud.gcp.data.datastore.core.mapping.DatastoreMappingContext)1 DiscriminatorValue (org.springframework.cloud.gcp.data.datastore.core.mapping.DiscriminatorValue)1 ServiceConfiguration (org.springframework.cloud.gcp.data.datastore.entities.ServiceConfiguration)1