Source file src/google.golang.org/grpc/shared_buffer_pool_test.go

Documentation: google.golang.org/grpc

     1  /*
     2   *
     3   * Copyright 2023 gRPC authors.
     4   *
     5   * Licensed under the Apache License, Version 2.0 (the "License");
     6   * you may not use this file except in compliance with the License.
     7   * You may obtain a copy of the License at
     8   *
     9   *     http://www.apache.org/licenses/LICENSE-2.0
    10   *
    11   * Unless required by applicable law or agreed to in writing, software
    12   * distributed under the License is distributed on an "AS IS" BASIS,
    13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14   * See the License for the specific language governing permissions and
    15   * limitations under the License.
    16   *
    17   */
    18  
    19  package grpc
    20  
    21  import "testing"
    22  
    23  func (s) TestSharedBufferPool(t *testing.T) {
    24  	pools := []SharedBufferPool{
    25  		nopBufferPool{},
    26  		NewSharedBufferPool(),
    27  	}
    28  
    29  	lengths := []int{
    30  		level4PoolMaxSize + 1,
    31  		level4PoolMaxSize,
    32  		level3PoolMaxSize,
    33  		level2PoolMaxSize,
    34  		level1PoolMaxSize,
    35  		level0PoolMaxSize,
    36  	}
    37  
    38  	for _, p := range pools {
    39  		for _, l := range lengths {
    40  			bs := p.Get(l)
    41  			if len(bs) != l {
    42  				t.Fatalf("Expected buffer of length %d, got %d", l, len(bs))
    43  			}
    44  
    45  			p.Put(&bs)
    46  		}
    47  	}
    48  }
    49  

View as plain text