From 589278900576030796facb9903d4154b6aba2f14 Mon Sep 17 00:00:00 2001
From: clowwindy <clowwindy42@gmail.com>
Date: Sat, 1 Nov 2014 23:47:38 +0800
Subject: [PATCH] refine tests: add time limit for curl; output shadowsocks
 logs

---
 .travis.yml   |  1 +
 tests/test.py | 64 ++++++++++++++++++++++++++++++++++++---------------
 2 files changed, 46 insertions(+), 19 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index b1a64df..403c88a 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -17,6 +17,7 @@ script:
   - pep8 .
   - pyflakes .
   - python tests/nose_plugin.py -v
+  - python setup.py sdist
   - python tests/test.py -c tests/aes.json
   - python tests/test.py -c tests/aes-ctr.json
   - python tests/test.py -c tests/aes-cfb1.json
diff --git a/tests/test.py b/tests/test.py
index b04862f..39ec374 100755
--- a/tests/test.py
+++ b/tests/test.py
@@ -47,6 +47,16 @@ p1 = Popen(['python', 'shadowsocks/server.py', '-c', server_config],
 p2 = Popen(['python', 'shadowsocks/local.py', '-c', client_config],
            stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True)
 p3 = None
+p4 = None
+p3_fin = False
+p4_fin = False
+
+# 1 shadowsocks started
+# 2 curl started
+# 3 curl finished
+# 4 dig started
+# 5 dig finished
+stage = 1
 
 try:
     local_ready = False
@@ -59,6 +69,11 @@ try:
 
         for fd in r:
             line = fd.readline()
+            if not line:
+                if stage == 2 and fd == p3.stdout:
+                    stage = 3
+                if stage == 4 and fd == p4.stdout:
+                    stage = 5
             if bytes != str:
                 line = str(line, 'utf8')
             sys.stdout.write(line)
@@ -70,27 +85,38 @@ try:
         if local_ready and server_ready and p3 is None:
             time.sleep(1)
 
-            break
-
-    p3 = Popen(['curl', 'http://www.example.com/', '-v', '-L',
-               '--socks5-hostname', '127.0.0.1:1081'], close_fds=True)
-    if p3 is not None:
-        r = p3.wait()
-        if r != 0:
-            sys.exit(r)
-    else:
-        sys.exit(1)
+            p3 = Popen(['curl', 'http://www.example.com/', '-v', '-L',
+                        '--socks5-hostname', '127.0.0.1:1081',
+                        '-m', '15', '--connect-timeout', '10'],
+                       stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True)
+            if p3 is not None:
+                fdset.append(p3.stdout)
+                fdset.append(p3.stderr)
+                stage = 2
+            else:
+                sys.exit(1)
 
-    p4 = Popen(['socksify', 'dig', '@8.8.8.8', 'www.google.com'],
-               close_fds=True)
-    if p4 is not None:
-        r = p4.wait()
-        if r != 0:
-            sys.exit(r)
-    else:
-        sys.exit(1)
-    print('test passed')
+        if stage == 3 and p3 is not None:
+            fdset.remove(p3.stdout)
+            fdset.remove(p3.stderr)
+            r = p3.wait()
+            if r != 0:
+                sys.exit(1)
+            p4 = Popen(['socksify', 'dig', '@8.8.8.8', 'www.google.com'],
+                       stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True)
+            if p4 is not None:
+                fdset.append(p4.stdout)
+                fdset.append(p4.stderr)
+                stage = 4
+            else:
+                sys.exit(1)
 
+        if stage == 5:
+            r = p4.wait()
+            if r != 0:
+                sys.exit(1)
+            print('test passed')
+            break
 finally:
     for p in [p1, p2]:
         try: