ShvaykaD
6 years ago
18 changed files with 725 additions and 738 deletions
@ -1,50 +0,0 @@ |
|||
/** |
|||
* Copyright © 2016-2020 The Thingsboard Authors |
|||
* |
|||
* Licensed 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.thingsboard.server.common.data.device.profile; |
|||
|
|||
import com.fasterxml.jackson.core.JsonParser; |
|||
import com.fasterxml.jackson.core.JsonProcessingException; |
|||
import com.fasterxml.jackson.databind.DeserializationContext; |
|||
import com.fasterxml.jackson.databind.JsonNode; |
|||
import com.fasterxml.jackson.databind.deser.std.StdDeserializer; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.thingsboard.server.common.data.TransportPayloadType; |
|||
|
|||
import java.io.IOException; |
|||
|
|||
@Slf4j |
|||
public class MqttTransportConfigurationDeserializer extends StdDeserializer<MqttDeviceProfileTransportConfiguration> { |
|||
|
|||
public MqttTransportConfigurationDeserializer() { |
|||
super(MqttDeviceProfileTransportConfiguration.class); |
|||
} |
|||
|
|||
@Override |
|||
public MqttDeviceProfileTransportConfiguration deserialize(JsonParser jsonParser, DeserializationContext deserializationContext) throws IOException, JsonProcessingException { |
|||
try { |
|||
JsonNode jsonNode = jsonParser.readValueAsTree(); |
|||
if (jsonNode.hasNonNull("transportPayloadType") && jsonNode.get("transportPayloadType").asText().equals(TransportPayloadType.PROTOBUF.name())) { |
|||
return jsonParser.getCodec().treeToValue(jsonNode, MqttProtoDeviceProfileTransportConfiguration.class); |
|||
} else { |
|||
return jsonParser.getCodec().treeToValue(jsonNode, MqttJsonDeviceProfileTransportConfiguration.class); |
|||
} |
|||
} catch (IOException e) { |
|||
log.trace("Failed to deserialize JSON content into equivalent tree model during creating {}!", MqttDeviceProfileTransportConfiguration.class.getSimpleName(), e); |
|||
throw new RuntimeException("Failed to deserialize JSON content into equivalent tree model during creating " + MqttDeviceProfileTransportConfiguration.class.getSimpleName() + "!", e); |
|||
} |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,37 @@ |
|||
/** |
|||
* Copyright © 2016-2020 The Thingsboard Authors |
|||
* |
|||
* Licensed 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.thingsboard.server.common.data.device.profile; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonIgnore; |
|||
import com.fasterxml.jackson.annotation.JsonIgnoreProperties; |
|||
import com.fasterxml.jackson.annotation.JsonSubTypes; |
|||
import com.fasterxml.jackson.annotation.JsonTypeInfo; |
|||
import org.thingsboard.server.common.data.TransportPayloadType; |
|||
|
|||
@JsonIgnoreProperties(ignoreUnknown = true) |
|||
@JsonTypeInfo( |
|||
use = JsonTypeInfo.Id.NAME, |
|||
include = JsonTypeInfo.As.PROPERTY, |
|||
property = "transportPayloadType") |
|||
@JsonSubTypes({ |
|||
@JsonSubTypes.Type(value = JsonTransportPayloadConfiguration.class, name = "JSON"), |
|||
@JsonSubTypes.Type(value = ProtoTransportPayloadConfiguration.class, name = "PROTOBUF")}) |
|||
public interface TransportPayloadTypeConfiguration { |
|||
|
|||
@JsonIgnore |
|||
TransportPayloadType getTransportPayloadType(); |
|||
|
|||
} |
|||
@ -1,406 +0,0 @@ |
|||
/** |
|||
* Copyright © 2016-2020 The Thingsboard Authors |
|||
* |
|||
* Licensed 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.thingsboard.server.transport.mqtt.util; |
|||
|
|||
import com.github.os72.protobuf.dynamic.DynamicSchema; |
|||
import com.google.protobuf.Descriptors; |
|||
import com.google.protobuf.DynamicMessage; |
|||
import com.google.protobuf.InvalidProtocolBufferException; |
|||
import com.google.protobuf.util.JsonFormat; |
|||
import com.squareup.wire.schema.internal.parser.ProtoFileElement; |
|||
import com.squareup.wire.schema.internal.parser.ProtoParser; |
|||
import org.junit.Rule; |
|||
import org.junit.Test; |
|||
import org.junit.rules.ExpectedException; |
|||
import org.junit.runner.RunWith; |
|||
import org.mockito.runners.MockitoJUnitRunner; |
|||
import org.thingsboard.server.common.data.device.profile.MqttProtoDeviceProfileTransportConfiguration; |
|||
|
|||
import java.util.List; |
|||
import java.util.Set; |
|||
|
|||
import static org.junit.Assert.assertEquals; |
|||
import static org.junit.Assert.assertNotNull; |
|||
import static org.junit.Assert.assertTrue; |
|||
import static org.thingsboard.server.common.data.device.profile.MqttProtoDeviceProfileTransportConfiguration.LOCATION; |
|||
|
|||
@RunWith(MockitoJUnitRunner.class) |
|||
public class MqttDynamicProtoSchemaTest { |
|||
|
|||
private static final String PROTO_SCHEMA_WITH_NESTED_MSG_TYPES = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package testnested;\n" + |
|||
"\n" + |
|||
"message Outer {\n" + |
|||
" message MiddleAA {\n" + |
|||
" message Inner {\n" + |
|||
" int64 ival = 1;\n" + |
|||
" bool booly = 2;\n" + |
|||
" }\n" + |
|||
" Inner inner = 1;\n" + |
|||
" }\n" + |
|||
" message MiddleBB {\n" + |
|||
" message Inner {\n" + |
|||
" int32 ival = 1;\n" + |
|||
" bool booly = 2;\n" + |
|||
" }\n" + |
|||
" Inner inner = 1;\n" + |
|||
" }\n" + |
|||
" MiddleAA middleAA = 1;\n" + |
|||
" MiddleBB middleBB = 2;\n" + |
|||
"}\n"; |
|||
|
|||
private static final String PROTO_SCHEMA_WITH_ONE_OFS = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package testoneofs;\n" + |
|||
"\n" + |
|||
"message SubMessage {\n" + |
|||
" repeated string name = 1;\n" + |
|||
"}\n" + |
|||
"\n" + |
|||
"message SampleMessage {\n" + |
|||
" oneof testOneOf {\n" + |
|||
" string name = 4;\n" + |
|||
" SubMessage subMessage = 9;\n" + |
|||
" }\n" + |
|||
"}"; |
|||
|
|||
private static final String IVALID_PROTO_SCHEMA_REQUIRED_FIELD_EXISTS = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"message SchemaValidationTest {\n" + |
|||
" required int32 parameter = 1;\n" + |
|||
"}"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_NOT_VALID_SYNTAX = "syntax = \"proto2\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"message SchemaValidationTest {\n" + |
|||
" required int32 parameter = 1;\n" + |
|||
"}"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_OPTIONS_NOT_SUPPORTED = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"option java_package = \"com.test.schemavalidation\";\n" + |
|||
"option java_multiple_files = true;\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"message SchemaValidationTest {\n" + |
|||
" int32 parameter = 1;\n" + |
|||
"}"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_PUBLIC_IMPORTS_NOT_SUPPORTED = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"import public \"oldschema.proto\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"message SchemaValidationTest {\n" + |
|||
" int32 parameter = 1;\n" + |
|||
"}"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_IMPORTS_NOT_SUPPORTED = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"import \"oldschema.proto\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"message SchemaValidationTest {\n" + |
|||
" int32 parameter = 1;\n" + |
|||
"}"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_EXTEND_DECLARATION_NOT_SUPPORTED = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"extend google.protobuf.MethodOptions {\n" + |
|||
" MyMessage my_method_option = 50007;\n" + |
|||
"}"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_ENUM_OPTIONS_NOT_SUPPORTED = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"enum testEnum {\n" + |
|||
" option allow_alias = true;\n" + |
|||
" DEFAULT = 0;\n" + |
|||
" STARTED = 1;\n" + |
|||
" RUNNING = 2;\n" + |
|||
"}\n" + |
|||
"\n" + |
|||
"message testMessage {\n" + |
|||
" int32 parameter = 1;\n" + |
|||
"}\n"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_NO_MESSAGE_TYPES_EXISTS = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"enum testEnum {\n" + |
|||
" DEFAULT = 0;\n" + |
|||
" STARTED = 1;\n" + |
|||
" RUNNING = 2;\n" + |
|||
"}"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_MESSAGE_OPTIONS_NOT_SUPPORTED = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"message testMessage {\n" + |
|||
" option allow_alias = true;\n" + |
|||
" int32 parameter = 1;\n" + |
|||
"}"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_MESSAGE_EXTENSIONS_NOT_SUPPORTED = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"message TestMessage {\n" + |
|||
" extensions 100 to 199;\n" + |
|||
"}\n"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_MESSAGE_GROUPS_NOT_SUPPORTED = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"message TestMessage {\n" + |
|||
" repeated group Result = 1 {\n" + |
|||
" string url = 2;\n" + |
|||
" string title = 3;\n" + |
|||
" repeated string snippets = 4;\n" + |
|||
" }\n" + |
|||
"}\n"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_MESSAGE_RESERVED_NOT_SUPPORTED = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"message Foo {\n" + |
|||
" reserved 2, 15, 9 to 11;\n" + |
|||
" reserved \"foo\", \"bar\";\n" + |
|||
"}"; |
|||
|
|||
private static final String INVALID_PROTO_SCHEMA_ONE_OFS_GROUPS_NOT_SUPPORTED = "syntax = \"proto3\";\n" + |
|||
"\n" + |
|||
"package schemavalidation;\n" + |
|||
"\n" + |
|||
"message SampleMessage {\n" + |
|||
" oneof test_oneof {\n" + |
|||
" string name = 1;\n" + |
|||
" group Result = 2 {\n" + |
|||
" \tstring url = 3;\n" + |
|||
" \tstring title = 4;\n" + |
|||
" \trepeated string snippets = 5;\n" + |
|||
" }\n" + |
|||
" }\n" + |
|||
"}"; |
|||
|
|||
private static final MqttProtoDeviceProfileTransportConfiguration mqttProtoDeviceProfileTransportConfiguration = new MqttProtoDeviceProfileTransportConfiguration(); |
|||
|
|||
private static void validateTransportProtoSchema(String schema, String schemaName) { |
|||
mqttProtoDeviceProfileTransportConfiguration.validateTransportProtoSchema(schema, schemaName); |
|||
} |
|||
|
|||
private static DynamicSchema getDynamicSchema(String schema, String schemaName) { |
|||
ProtoFileElement protoFileElement = getTransportProtoSchema(schema); |
|||
return mqttProtoDeviceProfileTransportConfiguration.getDynamicSchema(protoFileElement, schemaName); |
|||
} |
|||
|
|||
@Rule |
|||
public ExpectedException exceptionRule = ExpectedException.none(); |
|||
|
|||
@Test |
|||
public void testDynamicSchemaProtoFileValidation() { |
|||
processValidation("[Transport Configuration] failed to parse testParseToProtoFile due to: Syntax error in :6:4: 'required' label forbidden in proto3 field declarations", IVALID_PROTO_SCHEMA_REQUIRED_FIELD_EXISTS, "testParseToProtoFile"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaSyntaxValidation() { |
|||
processValidation("[Transport Configuration] invalid schema syntax: proto2 for: testSyntaxValidation provided! Only proto3 allowed!", INVALID_PROTO_SCHEMA_NOT_VALID_SYNTAX, "testSyntaxValidation"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaOptionsValidation() { |
|||
processValidation("[Transport Configuration] invalid testOptionsValidation schema provided! Schema options don't support!", INVALID_PROTO_SCHEMA_OPTIONS_NOT_SUPPORTED, "testOptionsValidation"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaPublicImportsValidation() { |
|||
processValidation("[Transport Configuration] invalid testPublicImportsValidation schema provided! Schema public imports don't support!", INVALID_PROTO_SCHEMA_PUBLIC_IMPORTS_NOT_SUPPORTED, "testPublicImportsValidation"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaImportsValidation() { |
|||
processValidation("[Transport Configuration] invalid testImportsValidation schema provided! Schema imports don't support!", INVALID_PROTO_SCHEMA_IMPORTS_NOT_SUPPORTED, "testImportsValidation"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaExtendDeclarationsValidation() { |
|||
processValidation("[Transport Configuration] invalid testExtendDeclarationsValidation schema provided! Schema extend declarations don't support!", INVALID_PROTO_SCHEMA_EXTEND_DECLARATION_NOT_SUPPORTED, "testExtendDeclarationsValidation"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaEnumOptionsValidation() { |
|||
processValidation("[Transport Configuration] invalid testEnumOptionsValidation schema provided! Enum definitions options are not supported!", INVALID_PROTO_SCHEMA_ENUM_OPTIONS_NOT_SUPPORTED, "testEnumOptionsValidation"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaNoOneMessageTypeExistsValidation() { |
|||
processValidation("[Transport Configuration] invalid noOneMessageTypeExists schema provided! At least one Message definition should exists!", INVALID_PROTO_SCHEMA_NO_MESSAGE_TYPES_EXISTS, "noOneMessageTypeExists"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaMessageTypeOptionsValidation() { |
|||
processValidation("[Transport Configuration] invalid messageTypeOptions schema provided! Message definition options don't support!", INVALID_PROTO_SCHEMA_MESSAGE_OPTIONS_NOT_SUPPORTED, "messageTypeOptions"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaMessageTypeExtensionsValidation() { |
|||
processValidation("[Transport Configuration] invalid messageTypeExtensions schema provided! Message definition extensions don't support!", INVALID_PROTO_SCHEMA_MESSAGE_EXTENSIONS_NOT_SUPPORTED, "messageTypeExtensions"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaMessageTypeReservedElementsValidation() { |
|||
processValidation("[Transport Configuration] invalid messageTypeReservedElements schema provided! Message definition reserved elements don't support!", INVALID_PROTO_SCHEMA_MESSAGE_RESERVED_NOT_SUPPORTED, "messageTypeReservedElements"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaMessageTypeGroupsElementsValidation() { |
|||
processValidation("[Transport Configuration] invalid messageTypeGroupsElements schema provided! Message definition groups don't support!", INVALID_PROTO_SCHEMA_MESSAGE_GROUPS_NOT_SUPPORTED, "messageTypeGroupsElements"); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaOneOfsTypeGroupsElementsValidation() { |
|||
processValidation("[Transport Configuration] invalid oneOfsTypeGroupsElements schema provided! OneOf definition groups don't support!", INVALID_PROTO_SCHEMA_ONE_OFS_GROUPS_NOT_SUPPORTED, "oneOfsTypeGroupsElements"); |
|||
} |
|||
|
|||
private void processValidation(String expectedMessage, String schema, String schemaName) { |
|||
exceptionRule.expect(IllegalArgumentException.class); |
|||
exceptionRule.expectMessage(expectedMessage); |
|||
validateTransportProtoSchema(schema, schemaName); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaCreationWithMessageNestedTypes() throws Exception { |
|||
String testNestedTypesProtoSchema = "testNestedTypesProtoSchema"; |
|||
validateTransportProtoSchema(PROTO_SCHEMA_WITH_NESTED_MSG_TYPES, testNestedTypesProtoSchema); |
|||
DynamicSchema dynamicSchema = getDynamicSchema(PROTO_SCHEMA_WITH_NESTED_MSG_TYPES, testNestedTypesProtoSchema); |
|||
assertNotNull(dynamicSchema); |
|||
Set<String> messageTypes = dynamicSchema.getMessageTypes(); |
|||
assertEquals(5, messageTypes.size()); |
|||
assertTrue(messageTypes.contains("testnested.Outer")); |
|||
assertTrue(messageTypes.contains("testnested.Outer.MiddleAA")); |
|||
assertTrue(messageTypes.contains("testnested.Outer.MiddleAA.Inner")); |
|||
assertTrue(messageTypes.contains("testnested.Outer.MiddleBB")); |
|||
assertTrue(messageTypes.contains("testnested.Outer.MiddleBB.Inner")); |
|||
|
|||
DynamicMessage.Builder middleAAInnerMsgBuilder = dynamicSchema.newMessageBuilder("testnested.Outer.MiddleAA.Inner"); |
|||
Descriptors.Descriptor middleAAInnerMsgDescriptor = middleAAInnerMsgBuilder.getDescriptorForType(); |
|||
DynamicMessage middleAAInnerMsg = middleAAInnerMsgBuilder |
|||
.setField(middleAAInnerMsgDescriptor.findFieldByName("ival"), 1L) |
|||
.setField(middleAAInnerMsgDescriptor.findFieldByName("booly"), true) |
|||
.build(); |
|||
|
|||
DynamicMessage.Builder middleAAMsgBuilder = dynamicSchema.newMessageBuilder("testnested.Outer.MiddleAA"); |
|||
Descriptors.Descriptor middleAAMsgDescriptor = middleAAMsgBuilder.getDescriptorForType(); |
|||
DynamicMessage middleAAMsg = middleAAMsgBuilder |
|||
.setField(middleAAMsgDescriptor.findFieldByName("inner"), middleAAInnerMsg) |
|||
.build(); |
|||
|
|||
DynamicMessage.Builder middleBBInnerMsgBuilder = dynamicSchema.newMessageBuilder("testnested.Outer.MiddleAA.Inner"); |
|||
Descriptors.Descriptor middleBBInnerMsgDescriptor = middleBBInnerMsgBuilder.getDescriptorForType(); |
|||
DynamicMessage middleBBInnerMsg = middleBBInnerMsgBuilder |
|||
.setField(middleBBInnerMsgDescriptor.findFieldByName("ival"), 0L) |
|||
.setField(middleBBInnerMsgDescriptor.findFieldByName("booly"), false) |
|||
.build(); |
|||
|
|||
DynamicMessage.Builder middleBBMsgBuilder = dynamicSchema.newMessageBuilder("testnested.Outer.MiddleBB"); |
|||
Descriptors.Descriptor middleBBMsgDescriptor = middleBBMsgBuilder.getDescriptorForType(); |
|||
DynamicMessage middleBBMsg = middleBBMsgBuilder |
|||
.setField(middleBBMsgDescriptor.findFieldByName("inner"), middleBBInnerMsg) |
|||
.build(); |
|||
|
|||
|
|||
DynamicMessage.Builder outerMsgBuilder = dynamicSchema.newMessageBuilder("testnested.Outer"); |
|||
Descriptors.Descriptor outerMsgBuilderDescriptor = outerMsgBuilder.getDescriptorForType(); |
|||
DynamicMessage outerMsg = outerMsgBuilder |
|||
.setField(outerMsgBuilderDescriptor.findFieldByName("middleAA"), middleAAMsg) |
|||
.setField(outerMsgBuilderDescriptor.findFieldByName("middleBB"), middleBBMsg) |
|||
.build(); |
|||
|
|||
assertEquals("{\n" + |
|||
" \"middleAA\": {\n" + |
|||
" \"inner\": {\n" + |
|||
" \"ival\": \"1\",\n" + |
|||
" \"booly\": true\n" + |
|||
" }\n" + |
|||
" },\n" + |
|||
" \"middleBB\": {\n" + |
|||
" \"inner\": {\n" + |
|||
" \"ival\": 0,\n" + |
|||
" \"booly\": false\n" + |
|||
" }\n" + |
|||
" }\n" + |
|||
"}", dynamicMsgToJson(outerMsgBuilderDescriptor, outerMsg.toByteArray())); |
|||
} |
|||
|
|||
@Test |
|||
public void testDynamicSchemaCreationWithMessageOneOfs() throws Exception { |
|||
String testOneOfsProtoSchema = "testOneOfsProtoSchema"; |
|||
validateTransportProtoSchema(PROTO_SCHEMA_WITH_ONE_OFS, testOneOfsProtoSchema); |
|||
DynamicSchema dynamicSchema = getDynamicSchema(PROTO_SCHEMA_WITH_ONE_OFS, testOneOfsProtoSchema); |
|||
assertNotNull(dynamicSchema); |
|||
Set<String> messageTypes = dynamicSchema.getMessageTypes(); |
|||
assertEquals(2, messageTypes.size()); |
|||
assertTrue(messageTypes.contains("testoneofs.SubMessage")); |
|||
assertTrue(messageTypes.contains("testoneofs.SampleMessage")); |
|||
|
|||
DynamicMessage.Builder sampleMsgBuilder = dynamicSchema.newMessageBuilder("testoneofs.SampleMessage"); |
|||
Descriptors.Descriptor sampleMsgDescriptor = sampleMsgBuilder.getDescriptorForType(); |
|||
assertNotNull(sampleMsgDescriptor); |
|||
|
|||
List<Descriptors.FieldDescriptor> fields = sampleMsgDescriptor.getFields(); |
|||
assertEquals(2, fields.size()); |
|||
DynamicMessage sampleMsg = sampleMsgBuilder |
|||
.setField(sampleMsgDescriptor.findFieldByName("name"), "Bob") |
|||
.build(); |
|||
assertEquals("{\n" + " \"name\": \"Bob\"\n" + "}", dynamicMsgToJson(sampleMsgDescriptor, sampleMsg.toByteArray())); |
|||
|
|||
DynamicMessage.Builder subMsgBuilder = dynamicSchema.newMessageBuilder("testoneofs.SubMessage"); |
|||
Descriptors.Descriptor subMsgDescriptor = subMsgBuilder.getDescriptorForType(); |
|||
DynamicMessage subMsg = subMsgBuilder |
|||
.addRepeatedField(subMsgDescriptor.findFieldByName("name"), "Alice") |
|||
.addRepeatedField(subMsgDescriptor.findFieldByName("name"), "John") |
|||
.build(); |
|||
|
|||
DynamicMessage sampleMsgWithOneOfSubMessage = sampleMsgBuilder.setField(sampleMsgDescriptor.findFieldByName("subMessage"), subMsg).build(); |
|||
assertEquals("{\n" + " \"subMessage\": {\n" + " \"name\": [\"Alice\", \"John\"]\n" + " }\n" + "}", |
|||
dynamicMsgToJson(sampleMsgDescriptor, sampleMsgWithOneOfSubMessage.toByteArray())); |
|||
|
|||
} |
|||
|
|||
private String dynamicMsgToJson(Descriptors.Descriptor descriptor, byte[] payload) throws InvalidProtocolBufferException { |
|||
DynamicMessage dynamicMessage = DynamicMessage.parseFrom(descriptor, payload); |
|||
return JsonFormat.printer().includingDefaultValueFields().print(dynamicMessage); |
|||
} |
|||
|
|||
private static ProtoFileElement getTransportProtoSchema(String protoSchema) { |
|||
return new ProtoParser(LOCATION, protoSchema.toCharArray()).readProtoFile(); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue