Add basic unittests for devtools http_server.
R=qsr@chromium.org
Review URL: https://codereview.chromium.org/1260183003 .
Cr-Mirrored-From: https://github.com/domokit/mojo
Cr-Mirrored-Commit: 5aee8c2d8d5c705b20a7f27b86817cb10c1b799e
diff --git a/devtoolslib/http_server_unittest.py b/devtoolslib/http_server_unittest.py
new file mode 100644
index 0000000..dec4571
--- /dev/null
+++ b/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()
