|
- /*
- *
- * Copyright 2018 gRPC authors.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- *
- */
-
- package proto
-
- import (
- "bytes"
- "sync"
- "testing"
-
- "google.golang.org/grpc/encoding"
- "google.golang.org/grpc/test/codec_perf"
- )
-
- func marshalAndUnmarshal(t *testing.T, codec encoding.Codec, expectedBody []byte) {
- p := &codec_perf.Buffer{}
- p.Body = expectedBody
-
- marshalledBytes, err := codec.Marshal(p)
- if err != nil {
- t.Errorf("codec.Marshal(_) returned an error")
- }
-
- if err := codec.Unmarshal(marshalledBytes, p); err != nil {
- t.Errorf("codec.Unmarshal(_) returned an error")
- }
-
- if !bytes.Equal(p.GetBody(), expectedBody) {
- t.Errorf("Unexpected body; got %v; want %v", p.GetBody(), expectedBody)
- }
- }
-
- func TestBasicProtoCodecMarshalAndUnmarshal(t *testing.T) {
- marshalAndUnmarshal(t, codec{}, []byte{1, 2, 3})
- }
-
- // Try to catch possible race conditions around use of pools
- func TestConcurrentUsage(t *testing.T) {
- const (
- numGoRoutines = 100
- numMarshUnmarsh = 1000
- )
-
- // small, arbitrary byte slices
- protoBodies := [][]byte{
- []byte("one"),
- []byte("two"),
- []byte("three"),
- []byte("four"),
- []byte("five"),
- }
-
- var wg sync.WaitGroup
- codec := codec{}
-
- for i := 0; i < numGoRoutines; i++ {
- wg.Add(1)
- go func() {
- defer wg.Done()
- for k := 0; k < numMarshUnmarsh; k++ {
- marshalAndUnmarshal(t, codec, protoBodies[k%len(protoBodies)])
- }
- }()
- }
-
- wg.Wait()
- }
-
- // TestStaggeredMarshalAndUnmarshalUsingSamePool tries to catch potential errors in which slices get
- // stomped on during reuse of a proto.Buffer.
- func TestStaggeredMarshalAndUnmarshalUsingSamePool(t *testing.T) {
- codec1 := codec{}
- codec2 := codec{}
-
- expectedBody1 := []byte{1, 2, 3}
- expectedBody2 := []byte{4, 5, 6}
-
- proto1 := codec_perf.Buffer{Body: expectedBody1}
- proto2 := codec_perf.Buffer{Body: expectedBody2}
-
- var m1, m2 []byte
- var err error
-
- if m1, err = codec1.Marshal(&proto1); err != nil {
- t.Errorf("codec.Marshal(%v) failed", proto1)
- }
-
- if m2, err = codec2.Marshal(&proto2); err != nil {
- t.Errorf("codec.Marshal(%v) failed", proto2)
- }
-
- if err = codec1.Unmarshal(m1, &proto1); err != nil {
- t.Errorf("codec.Unmarshal(%v) failed", m1)
- }
-
- if err = codec2.Unmarshal(m2, &proto2); err != nil {
- t.Errorf("codec.Unmarshal(%v) failed", m2)
- }
-
- b1 := proto1.GetBody()
- b2 := proto2.GetBody()
-
- for i, v := range b1 {
- if expectedBody1[i] != v {
- t.Errorf("expected %v at index %v but got %v", i, expectedBody1[i], v)
- }
- }
-
- for i, v := range b2 {
- if expectedBody2[i] != v {
- t.Errorf("expected %v at index %v but got %v", i, expectedBody2[i], v)
- }
- }
- }
|