Add basic unittests for devtools http_server.

mojo/devtools/common/devtoolslib/http_server_unittest.py

Index: mojo/devtools/common/devtoolslib/http_server_unittest.py
diff --git a/mojo/devtools/common/devtoolslib/http_server_unittest.py b/mojo/devtools/common/devtoolslib/http_server_unittest.py
new file mode 100644
index 0000000000000000000000000000000000000000..dec4571c0244fbd7e995ee366720c23fc3891d2b
--- /dev/null
+++ b/mojo/devtools/common/devtoolslib/http_server_unittest.py
@@ -0,0 +1,96 @@
+# Copyright 2015 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import StringIO
+import gzip
+import imp
+import os.path
+import shutil
+import sys
+import tempfile
+import unittest
+import urllib2
+
+try:
+  imp.find_module('devtoolslib')
+except ImportError:
+  sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from devtoolslib import http_server
+
+
+class HttpServerTest(unittest.TestCase):
+  """Tests for the http_server module."""
+
+  def setUp(self):
+    """Creates a tree of temporary directories and files of the form described
+    below.
+
+      parent_dir
+        hello_dir
+          hello_subdir
+            hello_file
+        other_dir
+          other_file
+    """
+    self.parent_dir = tempfile.mkdtemp()
+    self.parent_file = tempfile.NamedTemporaryFile(delete=False,
+                                                   dir=self.parent_dir)
+    self.hello_dir = tempfile.mkdtemp(dir=self.parent_dir)
+    self.hello_subdir = tempfile.mkdtemp(dir=self.hello_dir)
+    self.hello_file = tempfile.NamedTemporaryFile(delete=False,
+                                                  dir=self.hello_subdir)
+    self.hello_file.write('hello')
+    self.hello_file.close()
+
+    self.other_dir = tempfile.mkdtemp(dir=self.parent_dir)
+    self.other_file = tempfile.NamedTemporaryFile(delete=False,
+                                                  dir=self.other_dir)
+
+  def tearDown(self):
+    shutil.rmtree(self.parent_dir)
+
+  def test_mappings(self):
+    """Maps two directories and verifies that the server serves files placed
+    there.
+    """
+    mappings = [
+        ('hello/', self.hello_dir),
+        ('other/', self.other_dir),
+    ]
+    server_address = ('http://%s:%u/' %
+                      http_server.start_http_server(mappings))
+
+    hello_relpath = os.path.relpath(self.hello_file.name, self.hello_dir)
+    hello_response = urllib2.urlopen(server_address + 'hello/' +
+                                     hello_relpath)
+    self.assertEquals(200, hello_response.getcode())
+
+    other_relpath = os.path.relpath(self.other_file.name, self.other_dir)
+    other_response = urllib2.urlopen(server_address + 'other/' +
+                                     other_relpath)
+    self.assertEquals(200, other_response.getcode())
+
+  def test_gzip(self):
+    """Verifies the gzip content encoding of the files being served."""
+    mappings = [
+        ('hello/', self.hello_dir),
+    ]
+    server_address = ('http://%s:%u/' %
+                      http_server.start_http_server(mappings))
+
+    hello_relpath = os.path.relpath(self.hello_file.name, self.hello_dir)
+    hello_response = urllib2.urlopen(server_address + 'hello/' +
+                                     hello_relpath)
+    self.assertEquals(200, hello_response.getcode())
+    self.assertTrue('Content-Encoding' in hello_response.info())
+    self.assertEquals('gzip', hello_response.info().get('Content-Encoding'))
+
+    content = gzip.GzipFile(
+        fileobj=StringIO.StringIO(hello_response.read())).read()
+    self.assertEquals('hello', content)
+
+
+if __name__ == "__main__":
+  unittest.main()