lazy_test.go

Documentation: google.golang.org/protobuf/internal/impl

     1  // Copyright 2020 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package impl_test
     6  
     7  import (
     8  	"testing"
     9  
    10  	"google.golang.org/protobuf/internal/flags"
    11  	"google.golang.org/protobuf/internal/impl"
    12  	"google.golang.org/protobuf/internal/protobuild"
    13  	"google.golang.org/protobuf/proto"
    14  
    15  	testpb "google.golang.org/protobuf/internal/testprotos/test"
    16  )
    17  
    18  func TestLazyExtensions(t *testing.T) {
    19  	checkLazy := func(when string, m *testpb.TestAllExtensions, want bool) {
    20  		xd := testpb.E_OptionalNestedMessage.TypeDescriptor()
    21  		if got := impl.IsLazy(m.ProtoReflect(), xd); got != want {
    22  			t.Errorf("%v: m.optional_nested_message lazy=%v, want %v", when, got, want)
    23  		}
    24  		e := proto.GetExtension(m, testpb.E_OptionalNestedMessage).(*testpb.TestAllExtensions_NestedMessage).Corecursive
    25  		if got := impl.IsLazy(e.ProtoReflect(), xd); got != want {
    26  			t.Errorf("%v: m.optional_nested_message.corecursive.optional_nested_message lazy=%v, want %v", when, got, want)
    27  		}
    28  	}
    29  
    30  	m1 := &testpb.TestAllExtensions{}
    31  	protobuild.Message{
    32  		"optional_nested_message": protobuild.Message{
    33  			"a": 1,
    34  			"corecursive": protobuild.Message{
    35  				"optional_nested_message": protobuild.Message{
    36  					"a": 2,
    37  				},
    38  			},
    39  		},
    40  	}.Build(m1.ProtoReflect())
    41  	checkLazy("before unmarshal", m1, false)
    42  
    43  	w, err := proto.Marshal(m1)
    44  	if err != nil {
    45  		t.Fatal(err)
    46  	}
    47  	m := &testpb.TestAllExtensions{}
    48  	if err := proto.Unmarshal(w, m); err != nil {
    49  		t.Fatal(err)
    50  	}
    51  	checkLazy("after unmarshal", m, flags.LazyUnmarshalExtensions)
    52  }
    53
View as plain text