get full coverage check

Author: joshua-goldstein

test.sh

@@ -4,16 +4,25 @@ set -eo pipefail
 
 go version
 
+# Check if Github Actions is running
+if [ $CI = "true" ]; then
+  # Enable code coverage
+  # export because tests run in a subprocess
+  export covermode="-covermode=atomic"
+  export coverprofile="-coverprofile=cover_tmp.out"
+  echo "mode: atomic" >> cover.out
+fi
+
 # Run `go list` BEFORE setting GOFLAGS so that the output is in the right
 # format for grep.
 # export packages because the test will run in a sub process.
 export packages=$(go list ./... | grep "github.com/dgraph-io/badger/v3/")
 
 tags="-tags=jemalloc"
 
-# Ensure that we can compile the binary.
+# Compile the Badger binary
 pushd badger
-go build -v $tags .
+  go build -v $tags .
 popd
 
 # Run the memory intensive tests first.
@@ -23,18 +32,22 @@ manual() {
   set -e
   for pkg in $packages; do
     echo "===> Testing $pkg"
-    go test $tags -timeout=25m -race $pkg -parallel 16
+    go test $tags -timeout=25m $covermode $coverprofile -race -parallel 16 $pkg
+    write_coverage
   done
   echo "==> DONE package tests"
 
   echo "==> Running manual tests"
   # Run the special Truncate test.
   rm -rf p
   set -e
-  go test $tags $timeout -run='TestTruncateVlogNoClose$' --manual=true
+  go test $tags $timeout $covermode $coverprofile -run='TestTruncateVlogNoClose$' --manual=true
+  write_coverage
   truncate --size=4096 p/000000.vlog
-  go test $tags $timeout -run='TestTruncateVlogNoClose2$' --manual=true
-  go test $tags $timeout -run='TestTruncateVlogNoClose3$' --manual=true
+  go test $tags $timeout $covermode $coverprofile -run='TestTruncateVlogNoClose2$' --manual=true
+  write_coverage
+  go test $tags $timeout $covermode $coverprofile -run='TestTruncateVlogNoClose3$' --manual=true
+  write_coverage
   rm -rf p
 
   # TODO(ibrahim): Let's make these tests have Manual prefix.
@@ -43,14 +56,22 @@ manual() {
   # TestValueGCManaged
   # TestDropPrefix
   # TestDropAllManaged
-  go test $tags $timeout -run='TestBigKeyValuePairs$' --manual=true
-  go test $tags $timeout -run='TestPushValueLogLimit' --manual=true
-  go test $tags $timeout -run='TestKeyCount' --manual=true
-  go test $tags $timeout -run='TestIteratePrefix' --manual=true
-  go test $tags $timeout -run='TestIterateParallel' --manual=true
-  go test $tags $timeout -run='TestBigStream' --manual=true
-  go test $tags $timeout -run='TestGoroutineLeak' --manual=true
-  go test $tags $timeout -run='TestGetMore' --manual=true
+  go test $tags $timeout $covermode $coverprofile -run='TestBigKeyValuePairs$' --manual=true
+  write_coverage
+  go test $tags $timeout $covermode $coverprofile -run='TestPushValueLogLimit' --manual=true
+  write_coverage
+  go test $tags $timeout $covermode $coverprofile -run='TestKeyCount' --manual=true
+  write_coverage
+  go test $tags $timeout $covermode $coverprofile -run='TestIteratePrefix' --manual=true
+  write_coverage
+  go test $tags $timeout $covermode $coverprofile -run='TestIterateParallel' --manual=true
+  write_coverage
+  go test $tags $timeout $covermode $coverprofile -run='TestBigStream' --manual=true
+  write_coverage
+  go test $tags $timeout $covermode $coverprofile -run='TestGoroutineLeak' --manual=true
+  write_coverage
+  go test $tags $timeout $covermode $coverprofile -run='TestGetMore' --manual=true
+  write_coverage
 
   echo "==> DONE manual tests"
 }
@@ -61,39 +82,48 @@ root() {
 
   echo "==> Running root level tests."
   set -e
-  go test $tags -v -race -parallel=16 -timeout=25m -covermode=atomic -coverprofile=cover_tmp.out .
-  cat cover_tmp.out >> cover.out
+  go test $tags -v -race -parallel=16 -timeout=25m $covermode $coverprofile .
+  write_coverage
   echo "==> DONE root level tests"
 }
 
 stream() {
   set -eo pipefail
   pushd badger
-  baseDir=$(mktemp -d -p .)
-  ./badger benchmark write -s --dir=$baseDir/test | tee $baseDir/log.txt
-  ./badger benchmark read --dir=$baseDir/test --full-scan | tee --append $baseDir/log.txt
-  ./badger benchmark read --dir=$baseDir/test -d=30s | tee --append $baseDir/log.txt
-  ./badger stream --dir=$baseDir/test -o "$baseDir/test2" | tee --append $baseDir/log.txt
-  count=$(cat "$baseDir/log.txt" | grep "at program end: 0 B" | wc -l)
-  rm -rf $baseDir
-  if [ $count -ne 4 ]; then
-    echo "LEAK detected in Badger stream."
-    return 1
-  fi
-  echo "==> DONE stream test"
+    baseDir=$(mktemp -d -p .)
+    ./badger benchmark write -s --dir=$baseDir/test | tee $baseDir/log.txt
+    ./badger benchmark read --dir=$baseDir/test --full-scan | tee --append $baseDir/log.txt
+    ./badger benchmark read --dir=$baseDir/test -d=30s | tee --append $baseDir/log.txt
+    ./badger stream --dir=$baseDir/test -o "$baseDir/test2" | tee --append $baseDir/log.txt
+    count=$(cat "$baseDir/log.txt" | grep "at program end: 0 B" | wc -l)
+    rm -rf $baseDir
+    if [ $count -ne 4 ]; then
+      echo "LEAK detected in Badger stream."
+      return 1
+    fi
+    echo "==> DONE stream test"
   popd
   return 0
 }
 
+write_coverage() {
+  if [ $CI = "true" ]; then
+    if [ -f cover_tmp.out ]; then
+      sed -i '1d' cover_tmp.out
+      cat cover_tmp.out >> cover.out && rm cover_tmp.out
+    fi
+  fi
+  
+}
+
 export -f stream
 export -f manual
 export -f root
+export -f write_coverage
 
 # parallel tests currently not working
 # parallel --halt now,fail=1 --progress --line-buffer ::: stream manual root
 
-touch cover.out
-
 # run tests in sequence
 root
 stream