Examples with TestEmbeddedColumns org.springframework.cloud.gcp.data.spanner.core.convert.TestEntities.TestEmbeddedColumns used on opensource projects

Example 1 with TestEmbeddedColumns

use of org.springframework.cloud.gcp.data.spanner.core.convert.TestEntities.TestEmbeddedColumns in project spring-cloud-gcp by spring-cloud.

the class ConverterAwareMappingSpannerEntityWriterTests method writeTest.

@Test
@SuppressWarnings("unchecked")
public void writeTest() {
    TestEntity t = new TestEntity();
    t.id = "key1";
    t.enumField = TestEntity.Color.WHITE;
    // any positive time value will do.
    t.commitTimestamp = Timestamp.ofTimeMicroseconds(1000);
    t.booleanField = true;
    t.intField = 123;
    t.longField = 3L;
    t.doubleField = 3.33;
    t.doubleArray = new double[] { 3.33, 3.33, 3.33 };
    t.doubleList = new ArrayList<>();
    t.doubleList.add(3.33);
    t.stringList = new ArrayList<>();
    t.stringList.add("stringstringstring");
    t.dateField = Date.fromYearMonthDay(2018, 11, 22);
    t.timestampField = Timestamp.ofTimeMicroseconds(333);
    t.bytes = ByteArray.copyFrom("333");
    t.booleanList = new ArrayList<>();
    t.booleanList.add(t.booleanField);
    t.longList = new ArrayList<>();
    t.longList.add(t.longField);
    t.dateList = new ArrayList<>();
    t.dateList.add(t.dateField);
    t.timestampList = new ArrayList<>();
    t.timestampList.add(t.timestampField);
    t.bytesList = new ArrayList<>();
    t.bytesList.add(t.bytes);
    // this property will be ignored in write mapping because it is a child relationship. no
    // exception will result even though it is an unsupported type for writing.
    t.childTestEntities = new ArrayList<>();
    t.childTestEntities.add(new ChildTestEntity());
    t.testEmbeddedColumns = new TestEmbeddedColumns();
    t.testEmbeddedColumns.id2 = "key2";
    t.testEmbeddedColumns.id3 = "key3";
    t.testEmbeddedColumns.intField2 = 123;
    Instant i1 = Instant.ofEpochSecond(111);
    Instant i2 = Instant.ofEpochSecond(222);
    Instant i3 = Instant.ofEpochSecond(333);
    t.momentsInTime = new ArrayList<>();
    t.momentsInTime.add(i1);
    t.momentsInTime.add(i2);
    t.momentsInTime.add(i3);
    Timestamp t1 = Timestamp.ofTimeSecondsAndNanos(111, 0);
    Timestamp t2 = Timestamp.ofTimeSecondsAndNanos(222, 0);
    Timestamp t3 = Timestamp.ofTimeSecondsAndNanos(333, 0);
    List<Timestamp> timestamps = new ArrayList<>();
    timestamps.add(t1);
    timestamps.add(t2);
    timestamps.add(t3);
    t.bigDecimalField = new BigDecimal("1000000.00001");
    t.bigDecimals = Arrays.asList(new BigDecimal("999999999.0025"), BigDecimal.ZERO);
    WriteBuilder writeBuilder = mock(WriteBuilder.class);
    ValueBinder<WriteBuilder> idBinder = mock(ValueBinder.class);
    when(idBinder.to(anyString())).thenReturn(null);
    when(writeBuilder.set(eq("id"))).thenReturn(idBinder);
    ValueBinder<WriteBuilder> id2Binder = mock(ValueBinder.class);
    when(id2Binder.to(anyString())).thenReturn(null);
    when(writeBuilder.set(eq("id2"))).thenReturn(id2Binder);
    ValueBinder<WriteBuilder> id3Binder = mock(ValueBinder.class);
    when(id3Binder.to(anyString())).thenReturn(null);
    when(writeBuilder.set(eq("id3"))).thenReturn(id3Binder);
    ValueBinder<WriteBuilder> id4Binder = mock(ValueBinder.class);
    when(id4Binder.to(anyString())).thenReturn(null);
    when(writeBuilder.set(eq("id4"))).thenReturn(id4Binder);
    ValueBinder<WriteBuilder> stringFieldBinder = mock(ValueBinder.class);
    when(stringFieldBinder.to(anyString())).thenReturn(null);
    when(writeBuilder.set(eq("custom_col"))).thenReturn(stringFieldBinder);
    ValueBinder<WriteBuilder> booleanFieldBinder = mock(ValueBinder.class);
    when(booleanFieldBinder.to((Boolean) any())).thenReturn(null);
    when(writeBuilder.set(eq("booleanField"))).thenReturn(booleanFieldBinder);
    ValueBinder<WriteBuilder> intFieldBinder = mock(ValueBinder.class);
    when(intFieldBinder.to(anyLong())).thenReturn(null);
    when(writeBuilder.set(eq("intField"))).thenReturn(intFieldBinder);
    ValueBinder<WriteBuilder> intField2Binder = mock(ValueBinder.class);
    when(intField2Binder.to(anyLong())).thenReturn(null);
    when(writeBuilder.set(eq("intField2"))).thenReturn(intField2Binder);
    ValueBinder<WriteBuilder> longFieldBinder = mock(ValueBinder.class);
    when(longFieldBinder.to(anyString())).thenReturn(null);
    when(writeBuilder.set(eq("longField"))).thenReturn(longFieldBinder);
    ValueBinder<WriteBuilder> doubleFieldBinder = mock(ValueBinder.class);
    when(doubleFieldBinder.to(anyDouble())).thenReturn(null);
    when(writeBuilder.set(eq("doubleField"))).thenReturn(doubleFieldBinder);
    ValueBinder<WriteBuilder> doubleArrayFieldBinder = mock(ValueBinder.class);
    when(doubleArrayFieldBinder.toStringArray(any())).thenReturn(null);
    when(writeBuilder.set(eq("doubleArray"))).thenReturn(doubleArrayFieldBinder);
    ValueBinder<WriteBuilder> doubleListFieldBinder = mock(ValueBinder.class);
    when(doubleListFieldBinder.toFloat64Array((Iterable<Double>) any())).thenReturn(null);
    when(writeBuilder.set(eq("doubleList"))).thenReturn(doubleListFieldBinder);
    ValueBinder<WriteBuilder> stringListFieldBinder = mock(ValueBinder.class);
    when(stringListFieldBinder.toStringArray(any())).thenReturn(null);
    when(writeBuilder.set(eq("stringList"))).thenReturn(stringListFieldBinder);
    ValueBinder<WriteBuilder> booleanListFieldBinder = mock(ValueBinder.class);
    when(booleanListFieldBinder.toBoolArray((Iterable<Boolean>) any())).thenReturn(null);
    when(writeBuilder.set(eq("booleanList"))).thenReturn(booleanListFieldBinder);
    ValueBinder<WriteBuilder> longListFieldBinder = mock(ValueBinder.class);
    when(longListFieldBinder.toStringArray(any())).thenReturn(null);
    when(writeBuilder.set(eq("longList"))).thenReturn(longListFieldBinder);
    ValueBinder<WriteBuilder> timestampListFieldBinder = mock(ValueBinder.class);
    when(timestampListFieldBinder.toTimestampArray(any())).thenReturn(null);
    when(writeBuilder.set(eq("timestampList"))).thenReturn(timestampListFieldBinder);
    ValueBinder<WriteBuilder> dateListFieldBinder = mock(ValueBinder.class);
    when(dateListFieldBinder.toDateArray(any())).thenReturn(null);
    when(writeBuilder.set(eq("dateList"))).thenReturn(dateListFieldBinder);
    ValueBinder<WriteBuilder> instantListFieldBinder = mock(ValueBinder.class);
    when(instantListFieldBinder.toTimestampArray(any())).thenReturn(null);
    when(writeBuilder.set(eq("momentsInTime"))).thenReturn(instantListFieldBinder);
    ValueBinder<WriteBuilder> bytesListFieldBinder = mock(ValueBinder.class);
    when(bytesListFieldBinder.toDateArray(any())).thenReturn(null);
    when(writeBuilder.set(eq("bytesList"))).thenReturn(bytesListFieldBinder);
    ValueBinder<WriteBuilder> dateFieldBinder = mock(ValueBinder.class);
    when(dateFieldBinder.to((Date) any())).thenReturn(null);
    when(writeBuilder.set(eq("dateField"))).thenReturn(dateFieldBinder);
    ValueBinder<WriteBuilder> timestampFieldBinder = mock(ValueBinder.class);
    when(timestampFieldBinder.to((Timestamp) any())).thenReturn(null);
    when(writeBuilder.set(eq("timestampField"))).thenReturn(timestampFieldBinder);
    ValueBinder<WriteBuilder> bytesFieldBinder = mock(ValueBinder.class);
    when(bytesFieldBinder.to((ByteArray) any())).thenReturn(null);
    when(writeBuilder.set(eq("bytes"))).thenReturn(bytesFieldBinder);
    ValueBinder<WriteBuilder> commitTimestampBinder = mock(ValueBinder.class);
    when(commitTimestampBinder.to((Timestamp) any())).thenReturn(null);
    when(writeBuilder.set(eq("commitTimestamp"))).thenReturn(commitTimestampBinder);
    ValueBinder<WriteBuilder> bigDecimalFieldBinder = mock(ValueBinder.class);
    when(bigDecimalFieldBinder.to((BigDecimal) any())).thenReturn(null);
    when(writeBuilder.set(eq("bigDecimalField"))).thenReturn(bigDecimalFieldBinder);
    ValueBinder<WriteBuilder> bigDecimalsBinder = mock(ValueBinder.class);
    when(bigDecimalsBinder.toNumericArray(any())).thenReturn(null);
    when(writeBuilder.set(eq("bigDecimals"))).thenReturn(bigDecimalsBinder);
    this.spannerEntityWriter.write(t, writeBuilder::set);
    verify(idBinder, times(1)).to(eq(t.id));
    verify(id2Binder, times(1)).to(eq(t.testEmbeddedColumns.id2));
    verify(id3Binder, times(1)).to(eq(t.testEmbeddedColumns.id3));
    verify(stringFieldBinder, times(1)).to(eq(t.enumField.toString()));
    verify(booleanFieldBinder, times(1)).to(eq(Boolean.valueOf(t.booleanField)));
    verify(intFieldBinder, times(1)).to(eq(Long.valueOf(t.intField)));
    verify(intField2Binder, times(1)).to(eq(Long.valueOf(t.testEmbeddedColumns.intField2)));
    verify(longFieldBinder, times(1)).to(eq(String.valueOf(t.longField)));
    verify(doubleFieldBinder, times(1)).to(eq(Double.valueOf(t.doubleField)));
    verify(doubleArrayFieldBinder, times(1)).to("3.33,3.33,3.33");
    verify(doubleListFieldBinder, times(1)).toFloat64Array(eq(t.doubleList));
    verify(stringListFieldBinder, times(1)).toStringArray(eq(t.stringList));
    verify(booleanListFieldBinder, times(1)).toBoolArray(eq(t.booleanList));
    verify(longListFieldBinder, times(1)).toStringArray(any());
    verify(timestampListFieldBinder, times(1)).toTimestampArray(eq(t.timestampList));
    verify(dateListFieldBinder, times(1)).toDateArray(eq(t.dateList));
    verify(bytesListFieldBinder, times(1)).toBytesArray(eq(t.bytesList));
    verify(dateFieldBinder, times(1)).to(eq(t.dateField));
    verify(timestampFieldBinder, times(1)).to(eq(t.timestampField));
    verify(bytesFieldBinder, times(1)).to(eq(t.bytes));
    verify(instantListFieldBinder, times(1)).toTimestampArray(eq(timestamps));
    // the positive value set earlier must not be passed to Spanner. it must be replaced by
    // the dummy value.
    verify(commitTimestampBinder, times(1)).to(eq(Value.COMMIT_TIMESTAMP));
    verify(bigDecimalFieldBinder, times(1)).to(eq(t.bigDecimalField));
    verify(bigDecimalsBinder, times(1)).toNumericArray(eq(t.bigDecimals));
}