add large file test

clowwindy · clowwindy · commit f1b084be060f · 2014-12-23T13:09:51.000+08:00
diff --git a/.jenkins.sh b/.jenkins.sh
@@ -53,6 +53,8 @@ if [ -f /proc/sys/net/ipv4/tcp_fastopen ] ; then
     fi
 fi
 
+run_test tests/test_large_file.sh
+
 coverage combine && coverage report --include=shadowsocks/*
 rm -rf htmlcov
 rm -rf tmp
diff --git a/.travis.yml b/.travis.yml
@@ -9,7 +9,9 @@ cache:
     - dante-1.4.0
 before_install:
   - sudo apt-get update -qq
-  - sudo apt-get install -qq build-essential libssl-dev swig python-m2crypto python-numpy dnsutils
+  - sudo apt-get install -qq build-essential libssl-dev swig python-m2crypto python-numpy dnsutils iproute nginx
+  - sudo dd if=/dev/urandom of=/usr/share/nginx/www/file bs=1M count=10
+  - sudo service nginx restart
   - pip install m2crypto salsa20 pep8 pyflakes nose coverage
   - sudo tests/socksify/install.sh
   - sudo tests/libsodium/install.sh
diff --git a/tests/test_large_file.sh b/tests/test_large_file.sh
@@ -0,0 +1,38 @@
+#!/bin/bash
+
+DEV=lo
+PORT=8388
+DELAY=100ms
+
+PYTHON="coverage run -p -a"
+URL=http://127.0.0.1/file
+
+mkdir -p tmp
+
+type tc > /dev/null && (
+    tc qdisc add dev $DEV root handle 1: prio
+    tc qdisc add dev $DEV parent 1:3 handle 30: netem delay $DELAY
+    tc filter add dev $DEV parent 1:0 protocol ip u32 match ip dport $PORT 0xffff flowid 1:3
+    tc filter add dev $DEV parent 1:0 protocol ip u32 match ip sport $PORT 0xffff flowid 1:3
+    tc qdisc show dev lo
+)
+
+$PYTHON shadowsocks/local.py -c tests/aes.json &
+LOCAL=$!
+
+$PYTHON shadowsocks/server.py -c tests/aes.json &
+SERVER=$!
+
+sleep 3
+
+curl -o tmp/expected $URL
+curl -o tmp/result --socks5-hostname 127.0.0.1:1081 $URL
+
+kill $LOCAL
+kill $SERVER
+
+type tc > /dev/null && tc qdisc del dev lo root
+
+sleep 2
+
+diff tmp/expected tmp/result || exit 1
