Function TestMarshalRace

proto/extensions_test.go:609–645

(t *testing.T)

}

func TestMarshalRace(t *testing.T) {
	ext := &pb2.Ext{}
	m := &pb2.MyMessage{Count: proto.Int32(4)}
	if err := proto.SetExtension(m, pb2.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.

	wg := sync.WaitGroup{}
	errs := make(chan error, 3)
	for n := 3; n > 0; n-- {
		wg.Add(1)
		go func() {
			defer wg.Done()
			_, err := proto.Marshal(m)
			errs <- err
		}()
	}
	wg.Wait()
	close(errs)

	for err = range errs {
		if err != nil {
			t.Fatal(err)
		}
	}
}