#!/bin/sh
#
# ngIRCd Test Suite
# Copyright (c)2001-2012 Alexander Barton (alex@barton.de) and Contributors.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
# Please read the file COPYING, README and AUTHORS for more information.
#

# detect source directory
[ -z "$srcdir" ] && srcdir=`dirname $0`

# parse command line
[ "$1" -gt 0 ] 2> /dev/null && CLIENTS="$1" || CLIENTS=5
[ "$2" -gt 0 ] 2> /dev/null && MAX="$2" || MAX=-1

# get our name
name=`basename $0`

# create directories
[ -d logs ] || mkdir logs
[ -d tests ] || mkdir tests

# test for required external tools
type expect > /dev/null 2>&1
if [ $? -ne 0 ]; then
  echo "${name}: \"expect\" not found.";  exit 77
fi
type telnet > /dev/null 2>&1
if [ $? -ne 0 ]; then
  echo "${name}: \"telnet\" not found.";  exit 77
fi

# hello world! :-)
echo "stressing server with $CLIENTS clients (be patient!):"

# read in functions
. ${srcdir}/functions.inc

# create scripts for expect(1)
no=0
while [ ${no} -lt $CLIENTS ]; do
  cat ${srcdir}/stress-A.e > tests/${no}.e
  echo "send \"nick test${no}\\r\"" >> tests/${no}.e
  cat ${srcdir}/stress-B.e >> tests/${no}.e
  no=`expr ${no} + 1`
done

# run first script and check if it succeeds
echo_n "checking stress script ..."
expect tests/0.e > logs/stress-0.log 2> /dev/null
if [ $? -ne 0 ]; then
  echo " failure!"
  exit 1
else
  echo " ok."
fi

no=0
while [ ${no} -lt $CLIENTS ]; do
  expect tests/${no}.e > logs/stress-${no}.log 2> /dev/null &

  no=`expr ${no} + 1`
  echo "started client $no/$CLIENTS."

  [ $MAX -gt 0 ] && $srcdir/wait-tests.sh $MAX
done

echo_n "waiting for clients to complete: ."
touch logs/check-idle.log
while true; do
  expect ${srcdir}/check-idle.e >> logs/check-idle.log; res=$?
  echo "====================" >> logs/check-idle.log
  [ $res -ne 99 ] && break

  # there are still clients connected. Wait ...
  sleep 3
  echo_n "."
done

[ $res -eq 0 ] && echo " ok." || echo " failure!"

exit $res

# -eof-