devtoolslib/http_server_unittest.py

# 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()