Function TestMarshalRace

proto/extensions_test.go:656–691

(t *testing.T)

}

func TestMarshalRace(t *testing.T) {
	ext := &pb.Ext{}
	m := &pb.MyMessage{Count: proto.Int32(4)}
	if err := proto.SetExtension(m, pb.E_Ext_More, ext); err != nil {
		t.Fatalf("proto.SetExtension(m, desc, true): got error %q, want nil", err)
	}

	b, err := proto.Marshal(m)
	if err != nil {
		t.Fatalf("Could not marshal message: %v", err)
	}
	if err := proto.Unmarshal(b, m); err != nil {
		t.Fatalf("Could not unmarshal message: %v", err)
	}
	// after Unmarshal, the extension is in undecoded form.
	// GetExtension will decode it lazily. Make sure this does
	// not race against Marshal.

	errChan := make(chan error, 6)
	for n := 3; n > 0; n-- {
		go func() {
			_, err := proto.Marshal(m)
			errChan <- err
		}()
		go func() {
			_, err := proto.GetExtension(m, pb.E_Ext_More)
			errChan <- err
		}()
	}
	for i := 0; i < 6; i++ {
		err := <-errChan
		if err != nil {
			t.Fatal(err)
		}
	}
}