#!/bin/sh
# Copyright 2019 PingCAP, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# Usage: check-leaktest.sh
# It needs to run under the github.com/pingcap/tidb directory.

set -e

pkgs=$(git grep 'Suite' |grep -vE "Godeps|tags" |awk -F: '{print $1}' | xargs -n1 dirname | sort |uniq)
echo $pkgs
for pkg in ${pkgs}; do
  if [ -z "$(ls ${pkg}/*_test.go 2>/dev/null)" ]; then
    continue
  fi
  awk -F'[(]' '
/func \(s .*Suite\) Test.*C\) {/ {
  test = $1"("$2
  next
}

/defer testleak.AfterTest/ {
  test = 0
  next
}

{
    if (test && (FILENAME != "./tidb_test.go")) {
    	printf "%s: %s: missing defer testleak.AfterTest\n", FILENAME, test
    	test = 0
    	code = 1
    }
}

END {
  exit code
}

' ${pkg}/*_test.go
done
