Home | History | Annotate | Download | only in chrome 
      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 from telemetry.core.backends.chrome import inspector_network
      5 from telemetry.timeline import recording_options
      6 from telemetry.unittest import tab_test_case
      7 
      8 
      9 class InspectorNetworkTabTest(tab_test_case.TabTestCase):
     10   class TestCase(object):
     11     def __init__(self, monitoring = False, responses_count = 0,
     12                  subresources = None):
     13       # Whether to monitor network for this case.
     14       self.monitoring = monitoring
     15       # Number of responses expected for this case.
     16       self.responses_count = responses_count
     17       # List of subresource links for this case.
     18       self.subresources = subresources
     19 
     20   def __init__(self, *args):
     21     super(InspectorNetworkTabTest, self).__init__(*args)
     22 
     23   def _NavigateAndGetHTTPResponseEvents(self, page, record_network=True):
     24     opts = recording_options.TimelineRecordingOptions()
     25     opts.record_network = record_network
     26     self._tab.StartTimelineRecording(opts)
     27     self.Navigate(page)
     28     self._tab.StopTimelineRecording()
     29 
     30     self.assertTrue(self._tab.timeline_model)
     31     return self._tab.timeline_model.GetAllEventsOfName('HTTPResponse')
     32 
     33   def testHTTPResponseTimelineRecorder(self):
     34     tests = {
     35         'blank.html': InspectorNetworkTabTest.TestCase(),
     36         'green_rect.html': InspectorNetworkTabTest.TestCase(
     37             monitoring=True, responses_count=1),
     38         'image_decoding.html': InspectorNetworkTabTest.TestCase(
     39             monitoring=True, responses_count=2, subresources=['image.png']),
     40         }
     41     for page, test in tests.iteritems():
     42       events = self._NavigateAndGetHTTPResponseEvents(page, test.monitoring)
     43       self.assertEqual(test.responses_count, len(events))
     44       if not test.monitoring:
     45         continue
     46 
     47       # Verify required event fields
     48       for event in events:
     49         self.assertEqual('HTTPResponse', event.name)
     50         resp = inspector_network.InspectorNetworkResponseData.FromTimelineEvent(
     51             event)
     52         self.assertLess(0.0, resp.timestamp)
     53 
     54         self.assertTrue(resp.headers)
     55         self.assertTrue(resp.headers['Content-Length'])
     56         body, base64_encoded = resp.GetBody()
     57 
     58         link = resp.url[resp.url.rfind('/') + 1 :]
     59         self.assertTrue(link == page or link in test.subresources)
     60         if link == page:
     61           self.assertEqual(resp.GetHeader('Content-Type'), 'text/html')
     62           self.assertTrue('<!DOCTYPE HTML>' in body)
     63           self.assertFalse(base64_encoded)
     64         else:
     65           # We know this is the only subresource type in our setup.
     66           self.assertEqual(resp.GetHeader('Content-Type'), 'image/png')
     67           self.assertFalse('<!DOCTYPE HTML>' in body)
     68           self.assertTrue(base64_encoded)
     69 
     70   def testCacheableHTTPResponse(self):
     71     # We know this page has one PNG image and its cacheable.
     72     events = self._NavigateAndGetHTTPResponseEvents('image_decoding.html')
     73     images_first = []
     74     for event in events:
     75       resp = inspector_network.InspectorNetworkResponseData.FromTimelineEvent(
     76           event)
     77       if resp.GetHeader('Content-Type') == 'image/png':
     78         images_first.append(resp)
     79 
     80     self.assertEqual(1, len(images_first))
     81     self.assertFalse(images_first[0].served_from_cache)
     82 
     83     events = self._NavigateAndGetHTTPResponseEvents('image_decoding.html')
     84     images_second = []
     85     for event in events:
     86       resp = inspector_network.InspectorNetworkResponseData.FromTimelineEvent(
     87           event)
     88       if resp.GetHeader('Content-Type') == 'image/png':
     89         images_second.append(resp)
     90     self.assertEqual(1, len(images_second))
     91     # On the second fetch, the image is served from cache.
     92     self.assertTrue(images_second[0].served_from_cache)
     93