github.com/pingcap/ticdc@v0.0.0-20220526033649-485a10ef2652/scripts/add-leaktest.sh (about)

     1  #!/bin/sh
     2  # Copyright 2019 PingCAP, Inc.
     3  #
     4  # Licensed under the Apache License, Version 2.0 (the "License");
     5  # you may not use this file except in compliance with the License.
     6  # You may obtain a copy of the License at
     7  #
     8  #     http://www.apache.org/licenses/LICENSE-2.0
     9  #
    10  # Unless required by applicable law or agreed to in writing, software
    11  # distributed under the License is distributed on an "AS IS" BASIS,
    12  # See the License for the specific language governing permissions and
    13  # limitations under the License.
    14  #
    15  # Usage: add-leaktest.sh pkg/*_test.go
    16  
    17  set -eu
    18  
    19  sed -i'~' -e '
    20    /^func (.*) Test.*(c \*check.C) {/ {
    21      n
    22      /testleak.AfterTest/! i\
    23  		defer testleak.AfterTest(c)()
    24    }
    25  ' $@
    26  
    27  for i in $@; do
    28    if ! cmp -s $i $i~ ; then
    29      tools/bin/gofumports -s -w $i
    30    fi
    31    rm -f $i~
    32  done
    33  
    34  git --no-pager diff --exit-code