Apache
/
nifi
mirror of https://github.com/apache/nifi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

NIFI-12042 Refactored Groovy test to Java in nifi-mock-record-utils 

This closes #7681

Signed-off-by: David Handermann <exceptionfactory@apache.org>
This commit is contained in:
dan-s1 2023-09-11 21:56:57 +00:00 committed by exceptionfactory
parent f0b7775203
commit 09c1c8b2b3
No known key found for this signature in database
GPG Key ID: 29B6A52D2AAE8DBA
4 changed files with 60 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nifi-nar-bundles/nifi-extension-utils/nifi-record-utils/nifi-mock-record-utils/pom.xml
View File

@ -52,12 +52,6 @@
<version>2.0.0-SNAPSHOT</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.codehaus.groovy</groupId>
<artifactId>groovy-json</artifactId>
<version>${nifi.groovy.version}</version>
<scope>test</scope>
</dependency>
</dependencies>
<build>

2
nifi-nar-bundles/nifi-extension-utils/nifi-record-utils/nifi-mock-record-utils/src/main/java/org/apache/nifi/serialization/record/MockSchemaRegistry.java
View File

@ -40,7 +40,7 @@ public class MockSchemaRegistry extends AbstractControllerService implements Sch
schemaNameMap.put(name, schema);
}
private RecordSchema retrieveSchemaByName(final SchemaIdentifier schemaIdentifier) throws IOException, SchemaNotFoundException {
RecordSchema retrieveSchemaByName(final SchemaIdentifier schemaIdentifier) throws IOException, SchemaNotFoundException {
final Optional<String> schemaName = schemaIdentifier.getName();
if (!schemaName.isPresent()) {
throw new org.apache.nifi.schema.access.SchemaNotFoundException("Cannot retrieve schema because Schema Name is not present");

48
nifi-nar-bundles/nifi-extension-utils/nifi-record-utils/nifi-mock-record-utils/src/test/groovy/org/apache/nifi/serialization/record/TestMockSchemaRegistry.groovy
View File

@ -1,48 +0,0 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.nifi.serialization.record
import org.apache.avro.Schema
import org.apache.nifi.avro.AvroTypeUtil
import org.junit.jupiter.api.Test
import static groovy.json.JsonOutput.*
import static org.junit.jupiter.api.Assertions.assertEquals
import static org.junit.jupiter.api.Assertions.assertNotNull
class TestMockSchemaRegistry {
@Test
void testGetSchemaByName() {
def registry = new MockSchemaRegistry()
def schema = prettyPrint(toJson([
name: "TestSchema",
type: "record",
fields: [
[ name: "msg", type: "string" ]
]
]))
def recordSchema = AvroTypeUtil.createSchema(new Schema.Parser().parse(schema))
registry.addSchema("simple", recordSchema)
def identifier = SchemaIdentifier.builder().name("simple").build()
def result = registry.retrieveSchemaByName(identifier)
assertNotNull(result, "Failed to load schema.")
assertEquals(result.fieldNames, recordSchema.fieldNames)
}
}

59
nifi-nar-bundles/nifi-extension-utils/nifi-record-utils/nifi-mock-record-utils/src/test/java/org/apache/nifi/serialization/record/TestMockSchemaRegistry.java Normal file
View File

@ -0,0 +1,59 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.nifi.serialization.record;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializationFeature;
import org.apache.avro.Schema;
import org.apache.nifi.avro.AvroTypeUtil;
import org.apache.nifi.schema.access.SchemaNotFoundException;
import org.junit.jupiter.api.Test;
import java.io.IOException;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.Map;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull;
class TestMockSchemaRegistry {
@Test
void testGetSchemaByName() throws IOException, SchemaNotFoundException {
final ObjectMapper mapper = new ObjectMapper().enable(SerializationFeature.INDENT_OUTPUT);
final MockSchemaRegistry registry = new MockSchemaRegistry();
final Map<String, Serializable> map = new LinkedHashMap<>(3);
map.put("name", "TestSchema");
map.put("type", "record");
final Map<String, String> map1 = new LinkedHashMap<>(2);
map1.put("name", "msg");
map1.put("type", "string");
map.put("fields", new ArrayList<>(Collections.singletonList(map1)));
final String schema = mapper.writeValueAsString(map);
final RecordSchema recordSchema = AvroTypeUtil.createSchema(new Schema.Parser().parse(schema));
registry.addSchema("simple", recordSchema);
final SchemaIdentifier identifier = SchemaIdentifier.builder().name("simple").build();
final RecordSchema result = registry.retrieveSchemaByName(identifier);
assertNotNull(result, "Failed to load schema.");
assertEquals(result.getFieldNames(), recordSchema.getFieldNames());
}
}