Add memory usage tests

2018-01-10 11:43:16 -05:00
parent f7e7834a1c
commit 516976e32f
5 changed files with 147 additions and 0 deletions
@@ -0,0 +1,78 @@
+#!/bin/sh
+currentTest=1
+failedTests=0
+result(){ echo "$1 $currentTest $2"; currentTest=$(( $currentTest + 1 )); }
+ok(){ result 'ok' "- $1"; }
+ko(){ result 'not ok' "- $1"; failedTests=$(( $failedTests + 1 )); }
+
+pgrPort=49421
+
+pgrStopAll(){ pkill -f "$(stack path --local-install-root)/bin/postgrest"; }
+
+pgrStart(){
+  stack build --profile
+  stack exec -- postgrest test/memory-tests/config +RTS -p -h >/dev/null & pgrPID="$!";
+}
+pgrStop(){ kill "$pgrPID" 2>/dev/null; }
+
+setUp(){ pgrStopAll; }
+cleanUp(){ pgrStopAll; }
+
+checkPgrStarted(){
+  while pgrStarted && test $(rootStatus) -ne 200
+  do
+    sleep 1
+  done
+}
+pgrStarted(){ kill -0 "$pgrPID" 2>/dev/null; }
+rootStatus(){
+  curl -s -o /dev/null -I -w '%{http_code}' "http://localhost:$pgrPort/"
+}
+
+memoryTest(){
+  pgrStart
+  checkPgrStarted
+  factor=$(( 3*$(numfmt --from=si $1)/4 )) # 3/4 on $1 is need to maintain the specified size because of base64
+  payload="{\"blob\" : \"$(dd if=/dev/zero bs=$factor count=1 status=none | base64)\"}"
+  httpStatus=$(echo $payload | curl -s -H "Content-Type: application/json" --request $2 -d @- -w '%{http_code}' http://localhost:$pgrPort$3 | tr -d '"')
+  if test "$httpStatus" -ge 200 && test "$httpStatus" -lt 210
+  then
+    pgrStop
+    while [ ! -s postgrest.prof ]
+    do
+      sleep 1
+    done
+    BYTES_FMT=$(cat postgrest.prof | grep -o -P '(?<=alloc =).*(?=bytes)' | tr -d ' ')
+    BYTES=$(echo $BYTES_FMT | tr -d ',')
+    MAX_BYTES=$(numfmt --from=si $4)
+    if test $BYTES -le $MAX_BYTES
+    then
+      ok "$2 $3: with a $1 payload size the memory usage($BYTES_FMT bytes) is less than $4"
+    else
+      ko "$2 $3: with a $1 payload size the memory usage($BYTES_FMT bytes) is more than $4"
+    fi
+  else
+    pgrStop
+    ko "$2 $3: request failed with http $httpStatus"
+  fi
+}
+
+setUp
+
+echo "Running memory usage tests.."
+
+memoryTest "1M" "POST" "/rpc/leak" "15M"
+memoryTest "1M" "POST" "/leak" "15M"
+memoryTest "1M" "PATCH" "/leak?id=eq.1" "15M"
+
+memoryTest "10M" "POST" "/rpc/leak" "105M"
+memoryTest "10M" "POST" "/leak" "105M"
+memoryTest "10M" "PATCH" "/leak?id=eq.1" "105M"
+
+memoryTest "100M" "POST" "/rpc/leak" "895M"
+memoryTest "100M" "POST" "/leak" "895M"
+memoryTest "100M" "PATCH" "/leak?id=eq.1" "895M"
+
+cleanUp
+
+exit $failedTests