#!/bin/sh

GREEN="\\033[1;32m"
NORMAL="\\033[0;39m"
YELLOW="\\033[1;33m"

bench_dd() {
   TEST_NAME=$1
   DUMP_FILE=$2

   DUMP_FILE_SIZE=4700

   echo "$YELLOW"
   echo "-------------------------"
   echo "TESTS $TEST_NAME"
   echo "-------------------------" "$NORMAL"

   for n in 1 2 3; do
        echo "$GREEN" "TEST ECRITURE $TEST_NAME #$n" "$NORMAL"
        sync && echo 3 > /proc/sys/vm/drop_caches && dd if=/dev/zero of=$DUMP_FILE bs=1M count=$DUMP_FILE_SIZE
   done;

   for n in 1 2 3; do
        echo "$GREEN" "TEST LECTURE $TEST_NAME #$n" "$NORMAL"
        sync && echo 3 > /proc/sys/vm/drop_caches && dd if=$DUMP_FILE of=/dev/null | grep copied
   done;

   rm -f $DUMP_FILE
}

# Tests HDD
DUMP_DIR="/media/test-hdd"
DUMP_FILE="$DUMP_DIR/dump.dd"
ln -s /media/da4c0b13-0020-4b43-9b4a-79f5e12b9593 $DUMP_DIR
bench_dd "HDD" $DUMP_FILE
unlink $DUMP_DIR

# Tests NFS (localhost)
DUMP_DIR="/media/test-nfs"
DUMP_FILE="$DUMP_DIR/dump.dd"
mkdir -p $DUMP_DIR
mount -t nfs 127.0.0.1:/export/media $DUMP_DIR
bench_dd "NFS (localhost)" $DUMP_FILE
umount $DUMP_DIR
rmdir $DUMP_DIR

# Tests CIFS (localhost)
DUMP_DIR="/media/test-cifs"
DUMP_FILE="$DUMP_DIR/dump.dd"
mkdir -p $DUMP_DIR
mount -t cifs \\\\127.0.0.1\\media $DUMP_DIR -o guest
bench_dd "CIFS (localhost)" $DUMP_FILE
umount $DUMP_DIR
rmdir $DUMP_DIR