github.com/hxx258456/ccgo@v0.0.5-0.20230213014102-48b35f46f66f/grpc/internal/grpctest/grpctest_test.go (about)

     1  /*
     2   *
     3   * Copyright 2018 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 grpctest
    20  
    21  import (
    22  	"reflect"
    23  	"testing"
    24  )
    25  
    26  type tRunST struct {
    27  	setup, test, teardown bool
    28  }
    29  
    30  func (t *tRunST) Setup(*testing.T) {
    31  	t.setup = true
    32  }
    33  func (t *tRunST) TestSubTest(*testing.T) {
    34  	t.test = true
    35  }
    36  func (t *tRunST) Teardown(*testing.T) {
    37  	t.teardown = true
    38  }
    39  
    40  func TestRunSubTests(t *testing.T) {
    41  	x := &tRunST{}
    42  	RunSubTests(t, x)
    43  	if want := (&tRunST{setup: true, test: true, teardown: true}); !reflect.DeepEqual(x, want) {
    44  		t.Fatalf("x = %v; want all fields true", x)
    45  	}
    46  }
    47  
    48  type tNoST struct {
    49  	test bool
    50  }
    51  
    52  func (t *tNoST) TestSubTest(*testing.T) {
    53  	t.test = true
    54  }
    55  
    56  func TestNoSetupOrTeardown(t *testing.T) {
    57  	// Ensures nothing panics or fails if Setup/Teardown are omitted.
    58  	x := &tNoST{}
    59  	RunSubTests(t, x)
    60  	if want := (&tNoST{test: true}); !reflect.DeepEqual(x, want) {
    61  		t.Fatalf("x = %v; want %v", x, want)
    62  	}
    63  }