NIFI-5217: Allow non-nullable fields to be added to MockRecordParser

mattyb149 · mattyb149 · commit d79216d6b08a · 2018-05-18T15:59:12.000-04:00
Signed-off-by: Matthew Burgess <mattyb149@apache.org> This closes #2720
diff --git a/nifi-commons/nifi-record/src/main/java/org/apache/nifi/serialization/record/RecordField.java b/nifi-commons/nifi-record/src/main/java/org/apache/nifi/serialization/record/RecordField.java
@@ -24,7 +24,7 @@
 import java.util.Set;
 
 public class RecordField {
-    private static final boolean DEFAULT_NULLABLE = true;
+    public static final boolean DEFAULT_NULLABLE = true;
 
     private final String fieldName;
     private final DataType dataType;
diff --git a/nifi-nar-bundles/nifi-extension-utils/nifi-record-utils/nifi-mock-record-utils/src/main/java/org/apache/nifi/serialization/record/MockRecordParser.java b/nifi-nar-bundles/nifi-extension-utils/nifi-record-utils/nifi-mock-record-utils/src/main/java/org/apache/nifi/serialization/record/MockRecordParser.java
@@ -51,7 +51,11 @@ public void failAfter(final int failAfterN) {
     }
 
     public void addSchemaField(final String fieldName, final RecordFieldType type) {
-        fields.add(new RecordField(fieldName, type.getDataType()));
+        addSchemaField(fieldName, type, RecordField.DEFAULT_NULLABLE);
+    }
+
+    public void addSchemaField(final String fieldName, final RecordFieldType type, boolean isNullable) {
+        fields.add(new RecordField(fieldName, type.getDataType(), isNullable));
     }
 
     public void addRecord(Object... values) {

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,11 @@ public void failAfter(final int failAfterN) {`
`51`	`51`	`}`
`52`	`52`
`53`	`53`	`public void addSchemaField(final String fieldName, final RecordFieldType type) {`
`54`		`- fields.add(new RecordField(fieldName, type.getDataType()));`
	`54`	`+ addSchemaField(fieldName, type, RecordField.DEFAULT_NULLABLE);`
	`55`	`+ }`
	`56`	`+`
	`57`	`+ public void addSchemaField(final String fieldName, final RecordFieldType type, boolean isNullable) {`
	`58`	`+ fields.add(new RecordField(fieldName, type.getDataType(), isNullable));`
`55`	`59`	`}`
`56`	`60`
`57`	`61`	`public void addRecord(Object... values) {`