github.com/whtcorpsinc/milevadb-prod@v0.0.0-20211104133533-f57f4be3b597/soliton/testleak/add-leaktest.sh (about)

     1  #!/bin/sh
     2  # Copyright 2020 WHTCORPS INC, 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 (s \*test.*Suite) Test.*(c \*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      goimports -w $i
    30    fi
    31  echo $i
    32    rm -f $i~
    33  done