Skip to content

Commit 9f7779f

Browse files
committed
fix(QTDI-1559): Add record with sub-record to junit test.
1 parent a5768b4 commit 9f7779f

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

component-studio/component-runtime-di/src/test/java/org/talend/sdk/component/runtime/di/record/DiRowStructVisitorTest.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,7 @@ void visit() {
126126
createMetadata(dynamic, "DATES", StudioTypes.LIST, DATES);
127127
createMetadata(dynamic, "RECORDS", StudioTypes.LIST, RECORDS);
128128
createMetadata(dynamic, "RECORD", StudioTypes.OBJECT, RECORD);
129+
createMetadata(dynamic, "RECORD_SUB", StudioTypes.OBJECT, RECORD_WITH_SUB);
129130
createMetadata(dynamic, "BIG_DECIMALS", StudioTypes.LIST, BIG_DECIMALS);
130131
createMetadata(dynamic, "dynDate", StudioTypes.DATE, DATE);
131132
createMetadata(dynamic, "dynStringDate", StudioTypes.STRING,
@@ -142,7 +143,7 @@ void visit() {
142143
final Record record = visitor.get(rowStruct, factory);
143144
final Schema schema = record.getSchema();
144145
// should have 3 excluded fields
145-
assertEquals(59, schema.getEntries().size());
146+
assertEquals(60, schema.getEntries().size());
146147

147148
// schema metadata
148149
assertFalse(schema.getEntry("id").isNullable());
@@ -221,6 +222,7 @@ void visit() {
221222
assertEquals(dynObject, record.get(Object.class, "dynObject"));
222223
assertEquals(STRINGS, record.getArray(String.class, "STRINGS"));
223224
assertEquals(RECORD, record.get(Record.class, "RECORD"));
225+
assertEquals(RECORD_WITH_SUB, record.get(Record.class, "RECORD_SUB"));
224226
assertEquals(LONGS, record.getArray(Long.class, "LONGS"));
225227
assertEquals(FLOATS, record.getArray(Float.class, "FLOATS"));
226228
assertEquals(DOUBLES, record.getArray(Double.class, "DOUBLES"));

component-studio/component-runtime-di/src/test/java/org/talend/sdk/component/runtime/di/record/VisitorsTest.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,9 @@ public class VisitorsTest {
7171
protected static final Record RECORD =
7272
factory.newRecordBuilder().withInt("ntgr", 1).withString("str", "one").build();
7373

74+
protected static final Record RECORD_WITH_SUB =
75+
factory.newRecordBuilder().withInt("ntgr", 1).withRecord("record1", RECORD).build();
76+
7477
protected static final Object OBJECT = new Object();
7578

7679
protected static final Document DOCUMENT = new Document();

0 commit comments

Comments
 (0)