1# Copyright 2014 The Chromium Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5import os
6
7from telemetry.core import util
8from telemetry.testing import tab_test_case
9
10
11class MemoryCacheHTTPServerTest(tab_test_case.TabTestCase):
12
13  def setUp(self):
14    super(MemoryCacheHTTPServerTest, self).setUp()
15    self._test_filename = 'bear.webm'
16    _test_file = os.path.join(util.GetUnittestDataDir(), 'bear.webm')
17    self._test_file_size = os.stat(_test_file).st_size
18
19  def testBasicHostingAndRangeRequests(self):
20    self.Navigate('blank.html')
21    x = self._tab.EvaluateJavaScript('document.body.innerHTML')
22    x = x.strip()
23
24    # Test basic html hosting.
25    self.assertEquals(x, 'Hello world')
26
27    file_size = self._test_file_size
28    last_byte = file_size - 1
29    # Test byte range request: no end byte.
30    self.CheckContentHeaders('0-', '0-%d' % last_byte, file_size)
31
32    # Test byte range request: greater than zero start byte.
33    self.CheckContentHeaders('100-', '100-%d' % last_byte, file_size - 100)
34
35    # Test byte range request: explicit byte range.
36    self.CheckContentHeaders('2-500', '2-500', '499')
37
38    # Test byte range request: no start byte.
39    self.CheckContentHeaders('-228', '%d-%d' % (file_size - 228, last_byte),
40                             '228')
41
42    # Test byte range request: end byte less than start byte.
43    self.CheckContentHeaders('100-5', '100-%d' % last_byte, file_size - 100)
44
45  def CheckContentHeaders(self, content_range_request, content_range_response,
46                          content_length_response):
47    self._tab.ExecuteJavaScript("""
48        var loaded = false;
49        var xmlhttp = new XMLHttpRequest();
50        xmlhttp.onload = function(e) {
51          loaded = true;
52        };
53        // Avoid cached content by appending unique URL param.
54        xmlhttp.open('GET', {{ url }} + "?t=" + Date.now(), true);
55        xmlhttp.setRequestHeader('Range', {{ range }});
56        xmlhttp.send();
57        """,
58        url=self.UrlOfUnittestFile(self._test_filename),
59        range='bytes=%s' % content_range_request)
60    self._tab.WaitForJavaScriptCondition('loaded', timeout=5)
61    content_range = self._tab.EvaluateJavaScript(
62        'xmlhttp.getResponseHeader("Content-Range");')
63    content_range_response = 'bytes %s/%d' % (content_range_response,
64                                              self._test_file_size)
65    self.assertEquals(content_range, content_range_response)
66    content_length = self._tab.EvaluateJavaScript(
67        'xmlhttp.getResponseHeader("Content-Length");')
68    self.assertEquals(content_length, str(content_length_response))
69
70  def testAbsoluteAndRelativePathsYieldSameURL(self):
71    test_file_rel_path = 'green_rect.html'
72    test_file_abs_path = os.path.abspath(os.path.join(util.GetUnittestDataDir(),
73                                                      test_file_rel_path))
74    # It's necessary to bypass self.UrlOfUnittestFile since that
75    # concatenates the unittest directory on to the incoming path,
76    # causing the same code path to be taken in both cases.
77    self._platform.SetHTTPServerDirectories(util.GetUnittestDataDir())
78    self.assertEquals(
79      self._platform.http_server.UrlOf(test_file_rel_path),
80      self._platform.http_server.UrlOf(test_file_abs_path))
81