gitee.com/ks-custle/core-gm@v0.0.0-20230922171213-b83bdd97b62c/grpc/internal/leakcheck/leakcheck_test.go (about)

     1  /*
     2   *
     3   * Copyright 2017 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 leakcheck
    20  
    21  import (
    22  	"fmt"
    23  	"strings"
    24  	"testing"
    25  	"time"
    26  )
    27  
    28  type testErrorfer struct {
    29  	errorCount int
    30  	errors     []string
    31  }
    32  
    33  func (e *testErrorfer) Errorf(format string, args ...interface{}) {
    34  	e.errors = append(e.errors, fmt.Sprintf(format, args...))
    35  	e.errorCount++
    36  }
    37  
    38  func TestCheck(t *testing.T) {
    39  	const leakCount = 3
    40  	for i := 0; i < leakCount; i++ {
    41  		go func() { time.Sleep(2 * time.Second) }()
    42  	}
    43  	if ig := interestingGoroutines(); len(ig) == 0 {
    44  		t.Error("blah")
    45  	}
    46  	e := &testErrorfer{}
    47  	check(e, time.Second)
    48  	if e.errorCount != leakCount {
    49  		t.Errorf("check found %v leaks, want %v leaks", e.errorCount, leakCount)
    50  		t.Logf("leaked goroutines:\n%v", strings.Join(e.errors, "\n"))
    51  	}
    52  	check(t, 3*time.Second)
    53  }
    54  
    55  func ignoredTestingLeak(d time.Duration) {
    56  	time.Sleep(d)
    57  }
    58  
    59  func TestCheckRegisterIgnore(t *testing.T) {
    60  	RegisterIgnoreGoroutine("ignoredTestingLeak")
    61  	const leakCount = 3
    62  	for i := 0; i < leakCount; i++ {
    63  		go func() { time.Sleep(2 * time.Second) }()
    64  	}
    65  	go func() { ignoredTestingLeak(3 * time.Second) }()
    66  	if ig := interestingGoroutines(); len(ig) == 0 {
    67  		t.Error("blah")
    68  	}
    69  	e := &testErrorfer{}
    70  	check(e, time.Second)
    71  	if e.errorCount != leakCount {
    72  		t.Errorf("check found %v leaks, want %v leaks", e.errorCount, leakCount)
    73  		t.Logf("leaked goroutines:\n%v", strings.Join(e.errors, "\n"))
    74  	}
    75  	check(t, 3*time.Second)
    76  }