| 
 | 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 encoding  | 
 | 20 | + | 
 | 21 | +import (  | 
 | 22 | +	"testing"  | 
 | 23 | + | 
 | 24 | +	"github.com/google/go-cmp/cmp"  | 
 | 25 | +	"google.golang.org/grpc/internal/grpcutil"  | 
 | 26 | +)  | 
 | 27 | + | 
 | 28 | +type mockNamedCompressor struct {  | 
 | 29 | +	Compressor  | 
 | 30 | +}  | 
 | 31 | + | 
 | 32 | +func (mockNamedCompressor) Name() string {  | 
 | 33 | +	return "mock-compressor"  | 
 | 34 | +}  | 
 | 35 | + | 
 | 36 | +func TestDuplicateCompressorRegister(t *testing.T) {  | 
 | 37 | +	defer func(m map[string]Compressor) { registeredCompressor = m }(registeredCompressor)  | 
 | 38 | +	defer func(c []string) { grpcutil.RegisteredCompressorNames = c }(grpcutil.RegisteredCompressorNames)  | 
 | 39 | +	registeredCompressor = map[string]Compressor{}  | 
 | 40 | +	grpcutil.RegisteredCompressorNames = []string{}  | 
 | 41 | + | 
 | 42 | +	RegisterCompressor(&mockNamedCompressor{})  | 
 | 43 | + | 
 | 44 | +	// Register another instance of the same compressor.  | 
 | 45 | +	mc := &mockNamedCompressor{}  | 
 | 46 | +	RegisterCompressor(mc)  | 
 | 47 | +	if got := registeredCompressor["mock-compressor"]; got != mc {  | 
 | 48 | +		t.Fatalf("Unexpected compressor, got: %+v, want:%+v", got, mc)  | 
 | 49 | +	}  | 
 | 50 | + | 
 | 51 | +	wantNames := []string{"mock-compressor"}  | 
 | 52 | +	if !cmp.Equal(wantNames, grpcutil.RegisteredCompressorNames) {  | 
 | 53 | +		t.Fatalf("Unexpected compressor names, got: %+v, want:%+v", grpcutil.RegisteredCompressorNames, wantNames)  | 
 | 54 | +	}  | 
 | 55 | +}  | 
0 commit comments