Ticket #1456: test-run.sh

#!/bin/sh

#BIN="/home/mesh-node/allmydata-tahoe-1.8.2/bin/tahoe"
BIN="/mnt/dev/tahoe-lafs/bin/tahoe"
#BIN="/mnt/dev/tahoe-lafs.git/bin/tahoe"
UPLOAD="/mnt/tahoe-file2upload"
TAHOE_INTRO_DIR="~/.tahoe-intro"
TAHOE_1_DIR="~/.tahoe-1"
TAHOE_2_DIR="~/.tahoe-2"
TMP_DIR="/tmp"
#TEST_TEXT="test download text"
TEST_TEXT="test download text test download text test download text test download text test download text"
NUM_TEST_ROUNDS="1"

# Requirements for this test configuration:
# - tahoe-lafs
# - trickle
# - vnstat
# - TCP Vegas

runtest(){
        TOTAL=0
        #[ -f $TMP_DIR/tahoe-stats-$1.log ] && rm $TMP_DIR/tahoe-stats-$1.log
        date > $TMP_DIR/tahoe-stats-$1.log

        for i in `seq 1 $NUM_TEST_ROUNDS`; do
                [ -f $TMP_DIR/tahoe-file2download ] && rm $TMP_DIR/tahoe-file2download
                /usr/bin/time -o $TMP_DIR/tahoe-stats-$1.log -a -f "%e" $BIN get -d $TAHOE_2_DIR tahoe:tahoe-file2download $TMP_DIR/tahoe-file2download
                [ -z "`grep "$TEST_TEXT" $TMP_DIR/tahoe-file2download`" ] && echo "tahoe get ERROR!" >> $TMP_DIR/tahoe-stats-$1.log
        done
        cat $TMP_DIR/tahoe-stats-$1.log | awk -v ave=0 -v count=$NUM_TEST_ROUNDS '{ ave+=$1; print $1}; END { print "Average: " ave / count }'
}

prepare() {
        # Use TCP Vegas to minimize network latency
        [ -z "`grep "vegas" /proc/sys/net/ipv4/tcp_available_congestion_control`" ] && \
                sudo modprobe tcp_vegas
        [ -z "`grep "vegas" /proc/sys/net/ipv4/tcp_congestion_control`" ] && \
                sudo su -c "echo vegas > /proc/sys/net/ipv4/tcp_congestion_control"

        # Tahoe-1: Storage node
        # Tahoe-2: Gateway node, no storage
        killall tahoe
        $BIN start -d $TAHOE_INTRO_DIR
        $BIN start -d $TAHOE_1_DIR
        $BIN start -d $TAHOE_2_DIR
        sleep 3

        # Prepare and upload the small file which we want to download later
        echo "$TEST_TEXT" > $TMP_DIR/tahoe-file2download2
        $BIN put -d $TAHOE_2_DIR $TMP_DIR/tahoe-file2download2 tahoe:tahoe-file2download
}

vnstatwait() {
        echo "Starting upload. Press enter after you see the speed reaching about 800kbit/s"
        vnstat -i lo -l &
        read FOO
        killall vnstat
}

### Test 0: Download with no parallel upload or limitations
test0() {
        killall tahoe
        $BIN start -d $TAHOE_INTRO_DIR
        $BIN start -d $TAHOE_1_DIR
        $BIN start -d $TAHOE_2_DIR
        sleep 3

        echo "### Running Test 0: Download with no parallel upload or limitations..."
        runtest 0
}

### Test 1: Parallel 'tahoe put', limited to 800kbit/s ###
test1() {
        killall tahoe
        $BIN start -d $TAHOE_INTRO_DIR
        $BIN start -d $TAHOE_1_DIR
        $BIN start -d $TAHOE_2_DIR
        sleep 3

        trickle -s -u 100 -d 100 $BIN put -d $TAHOE_2_DIR $UPLOAD tahoe:tahoe-file2upload 2> /dev/null &
        vnstatwait

        echo "### Running Test 1: Parallel 'tahoe put', limited to 800kbit/s..."
        runtest 1
}

### Test 2: Parallel 'tahoe put', tahoe-2 (gateway) limited to 800kbit/s ###
test2() {
        killall tahoe
        $BIN start -d $TAHOE_INTRO_DIR
        $BIN start -d $TAHOE_1_DIR
        trickle -s -u 100 -d 100 $BIN run -d $TAHOE_2_DIR &
        sleep 3

        $BIN put -d $TAHOE_2_DIR $UPLOAD tahoe:tahoe-file2upload 2> /dev/null &
        vnstatwait

        echo "### Running Test 2: Parallel 'tahoe put', tahoe-2 (gateway) limited to 800kbit/s..."
        runtest 2
}

### Test 3: Parallel 'tahoe put', tahoe-1 (storage) limited to 800kbit/s ###
test3() {
        killall tahoe
        $BIN start -d $TAHOE_INTRO_DIR
        $BIN start -d $TAHOE_2_DIR
        trickle -s -u 100 -d 100 $BIN run -d $TAHOE_1_DIR &
        sleep 3

        $BIN put -d $TAHOE_2_DIR $UPLOAD tahoe:tahoe-file2upload 2> /dev/null &
        vnstatwait

        echo "### Running Test 3: Parallel 'tahoe put', tahoe-1 (storage) limited to 800kbit/s..."
        runtest 3
}


prepare

test0
test1
test2
test3