Search in sources :

Example 1 with FieldElement

use of com.squareup.wire.schema.internal.parser.FieldElement in project wire by square.

the class ProtoFileTest method roundTripToElement.

@Test
public void roundTripToElement() {
    TypeElement element1 = MessageElement.builder(location.at(11, 1)).name("Message1").documentation("Some comments about Message1").build();
    TypeElement element2 = MessageElement.builder(location.at(12, 1)).name("Message2").fields(ImmutableList.<FieldElement>of(FieldElement.builder(location.at(13, 3)).type("string").name("field").tag(1).build())).build();
    ExtendElement extend1 = ExtendElement.builder(location.at(16, 1)).name("Extend1").build();
    ExtendElement extend2 = ExtendElement.builder(location.at(17, 1)).name("Extend2").build();
    OptionElement option1 = OptionElement.create("kit", OptionElement.Kind.STRING, "kat");
    OptionElement option2 = OptionElement.create("foo", OptionElement.Kind.STRING, "bar");
    ServiceElement service1 = ServiceElement.builder(location.at(19, 1)).name("Service1").rpcs(ImmutableList.<RpcElement>of(RpcElement.builder(location.at(20, 3)).name("MethodA").requestType("Message2").responseType("Message1").options(ImmutableList.<OptionElement>of(OptionElement.create("methodoption", OptionElement.Kind.NUMBER, 1))).build())).build();
    ServiceElement service2 = ServiceElement.builder(location.at(24, 1)).name("Service2").build();
    ProtoFileElement fileElement = ProtoFileElement.builder(location).packageName("example.simple").imports(ImmutableList.of("example.thing")).publicImports(ImmutableList.of("example.other")).types(ImmutableList.of(element1, element2)).services(ImmutableList.of(service1, service2)).extendDeclarations(ImmutableList.of(extend1, extend2)).options(ImmutableList.of(option1, option2)).build();
    ProtoFile file = ProtoFile.get(fileElement);
    String expected = "" + "// file.proto\n" + "package example.simple;\n" + "\n" + "import \"example.thing\";\n" + "import public \"example.other\";\n" + "\n" + "option kit = \"kat\";\n" + "option foo = \"bar\";\n" + "\n" + "// Some comments about Message1\n" + "message Message1 {}\n" + "message Message2 {\n" + "  string field = 1;\n" + "}\n" + "\n" + "extend Extend1 {}\n" + "extend Extend2 {}\n" + "\n" + "service Service1 {\n" + "  rpc MethodA (Message2) returns (Message1) {\n" + "    option methodoption = 1;\n" + "  };\n" + "}\n" + "service Service2 {}\n";
    assertThat(file.toElement().toSchema()).isEqualTo(expected);
    assertThat(file.toElement()).isEqualToComparingFieldByField(fileElement);
}
Also used : RpcElement(com.squareup.wire.schema.internal.parser.RpcElement) TypeElement(com.squareup.wire.schema.internal.parser.TypeElement) OptionElement(com.squareup.wire.schema.internal.parser.OptionElement) FieldElement(com.squareup.wire.schema.internal.parser.FieldElement) ExtendElement(com.squareup.wire.schema.internal.parser.ExtendElement) ProtoFileElement(com.squareup.wire.schema.internal.parser.ProtoFileElement) ServiceElement(com.squareup.wire.schema.internal.parser.ServiceElement) Test(org.junit.Test)

Aggregations

ExtendElement (com.squareup.wire.schema.internal.parser.ExtendElement)1 FieldElement (com.squareup.wire.schema.internal.parser.FieldElement)1 OptionElement (com.squareup.wire.schema.internal.parser.OptionElement)1 ProtoFileElement (com.squareup.wire.schema.internal.parser.ProtoFileElement)1 RpcElement (com.squareup.wire.schema.internal.parser.RpcElement)1 ServiceElement (com.squareup.wire.schema.internal.parser.ServiceElement)1 TypeElement (com.squareup.wire.schema.internal.parser.TypeElement)1 Test (org.junit.Test)1