1 // Copyright (c) 2017 Ernest Micklei 2 // 3 // MIT License 4 // 5 // Permission is hereby granted, free of charge, to any person obtaining 6 // a copy of this software and associated documentation files (the 7 // "Software"), to deal in the Software without restriction, including 8 // without limitation the rights to use, copy, modify, merge, publish, 9 // distribute, sublicense, and/or sell copies of the Software, and to 10 // permit persons to whom the Software is furnished to do so, subject to 11 // the following conditions: 12 // 13 // The above copyright notice and this permission notice shall be 14 // included in all copies or substantial portions of the Software. 15 // 16 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 17 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 18 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 19 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE 20 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION 21 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION 22 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 23 24 package proto 25 26 import "testing" 27 28 func TestExtensions(t *testing.T) { 29 proto := `message M { 30 // extensions 31 extensions 4, 20 to max; // max 32 }` 33 p := newParserOn(proto) 34 p.next() // consume message 35 m := new(Message) 36 err := m.parse(p) 37 if err != nil { 38 t.Fatal(err) 39 } 40 if len(m.Elements) != 1 { 41 t.Fatal("1 elements expected, got", len(m.Elements), m.Elements) 42 } 43 f := m.Elements[0].(*Extensions) 44 if got, want := len(f.Ranges), 2; got != want { 45 t.Fatalf("got [%d] want [%d]", got, want) 46 } 47 if got, want := f.Position.Line, 3; got != want { 48 t.Fatalf("got [%d] want [%d]", got, want) 49 } 50 if got, want := f.Ranges[1].SourceRepresentation(), "20 to max"; got != want { 51 t.Errorf("got [%s] want [%s]", got, want) 52 } 53 if f.Comment == nil { 54 t.Fatal("comment expected") 55 } 56 if got, want := f.InlineComment.Message(), " max"; got != want { 57 t.Errorf("got [%s] want [%s]", got, want) 58 } 59 } 60