From 3b1b4a6ca03f8c7201cb05f4dee3ba2870b9419a Mon Sep 17 00:00:00 2001
From: Jim Morrison <jim@twist.com>
Date: Thu, 1 Mar 2012 12:09:31 -0800
Subject: [PATCH] Add a test for spdy version 3.

---
 tests/end_to_end.py | 50 +++++++++++++++++++++++++++++----------------
 1 file changed, 32 insertions(+), 18 deletions(-)

diff --git a/tests/end_to_end.py b/tests/end_to_end.py
index fd99c89..0cba6c4 100755
--- a/tests/end_to_end.py
+++ b/tests/end_to_end.py
@@ -20,15 +20,15 @@ import unittest
 _PORT = 9893
 
 
-def _run_server(port):
+def _run_server(port, args):
   srcdir = os.environ.get('srcdir', '.')
   testdata = '%s/testdata' % srcdir
   top_builddir = os.environ.get('top_builddir', '..')
-  return subprocess.Popen([
-      '%s/examples/spdyd' % top_builddir, str(port),
-      '-d', testdata,
-      '%s/privkey.pem' % testdata,
-      '%s/cacert.pem' % testdata])
+  base_args = ['%s/examples/spdyd' % top_builddir, str(port), '-d', testdata,
+               '%s/privkey.pem' % testdata, '%s/cacert.pem' % testdata]
+  if args:
+    base_args.extend(args)
+  return subprocess.Popen(base_args)
 
 def _check_server_up(port):
   # Check this check for now.
@@ -42,6 +42,15 @@ def _kill_server(server):
 
 
 class EndToEndSpdyTests(unittest.TestCase):
+  @classmethod
+  def setUpClass(cls):
+    cls.server = _run_server(_PORT, None)
+    _check_server_up(_PORT)
+
+  @classmethod
+  def tearDownClass(cls):
+    _kill_server(cls.server)
+
   def setUp(self):
     build_dir = os.environ.get('top_builddir', '..')
     self.client = '%s/examples/spdycat' % build_dir
@@ -51,19 +60,24 @@ class EndToEndSpdyTests(unittest.TestCase):
         0, subprocess.call([self.client, 'http://localhost:%d/' % _PORT]))
 
 
-class TestProgram(unittest.TestProgram):
-  def runTests(self):
-    self.testRunner = unittest.TextTestRunner()
-    result = self.testRunner.run(self.test)
-    self.successful = result.wasSuccessful()
+class EndToEndSpdy3Tests(unittest.TestCase):
+  @classmethod
+  def setUpClass(cls):
+    cls.server = _run_server(_PORT, '-3')
+    _check_server_up(_PORT)
 
+  @classmethod
+  def tearDownClass(cls):
+    _kill_server(cls.server)
+
+  def setUp(self):
+    build_dir = os.environ.get('top_builddir', '..')
+    self.client = '%s/examples/spdycat' % build_dir
+
+  def testSimpleRequest(self):
+    self.assertEquals(
+        0, subprocess.call([self.client, 'http://localhost:%d/' % _PORT]))
 
-def main():
-  server = _run_server(_PORT)
-  _check_server_up(_PORT)
-  result = TestProgram()
-  _kill_server(server)
-  return not result.successful
 
 if __name__ == '__main__':
-  sys.exit(main())
+  unittest.main()