tools/android/loading Add 'redirect' and 'ping' content types.

tools/android/loading/request_track_unittest.py

 # Copyright 2016 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 copy
 import json
 import unittest
 
 from request_track import (TimeBetween, Request, RequestTrack, TimingFromDict)
 
@@ skipping 29 matching lines @@
     r.response_headers = {}
     self.assertEquals(None, r.GetContentType())
     r.response_headers = {'Content-Type': 'application/javascript'}
     self.assertEquals('application/javascript', r.GetContentType())
     # Case-insensitive match.
     r.response_headers = {'content-type': 'application/javascript'}
     self.assertEquals('application/javascript', r.GetContentType())
     # Parameters are filtered out.
     r.response_headers = {'Content-Type': 'application/javascript;bla'}
     self.assertEquals('application/javascript', r.GetContentType())
-    # MIME type takes precedence over headers.
+    # MIME type takes precedence over 'Content-Type' header.
     r.mime_type = 'image/webp'
     self.assertEquals('image/webp', r.GetContentType())
     r.mime_type = None
+    # Test for 'ping' type.
+    r.status = 204
+    self.assertEquals('ping', r.GetContentType())
+    r.status = None
+    r.response_headers = {'Content-Type': 'application/javascript',
+                          'content-length': '0'}
+    self.assertEquals('ping', r.GetContentType())
+    # Test for 'redirect' type.
+    r.response_headers = {'Content-Type': 'application/javascript',
+                          'location': 'http://foo',
+                          'content-length': '0'}
+    self.assertEquals('redirect', r.GetContentType())
+
+  def testGetHTTPResponseHeader(self):
+    r = Request()
+    r.response_headers = {}
+    self.assertEquals(None, r.GetHTTPResponseHeader('Foo'))
+    r.response_headers = {'Foo': 'Bar', 'Baz': 'Foo'}
+    self.assertEquals('Bar', r.GetHTTPResponseHeader('Foo'))
+    r.response_headers = {'foo': 'Bar', 'Baz': 'Foo'}
+    self.assertEquals('Bar', r.GetHTTPResponseHeader('Foo'))
 
 
 class RequestTrackTestCase(unittest.TestCase):
   _REQUEST_WILL_BE_SENT = {
       'method': 'Network.requestWillBeSent',
       'params': {
           'documentURL': 'http://example.com/',
           'frameId': '32493.1',
           'initiator': {
               'type': 'other'
@@ skipping 318 matching lines @@
   def _ValidSequence(cls, request_track):
     request_track.Handle(
         'Network.requestWillBeSent', cls._REQUEST_WILL_BE_SENT)
     request_track.Handle('Network.responseReceived', cls._RESPONSE_RECEIVED)
     request_track.Handle('Network.dataReceived', cls._DATA_RECEIVED_1)
     request_track.Handle('Network.dataReceived', cls._DATA_RECEIVED_2)
     request_track.Handle('Network.loadingFinished', cls._LOADING_FINISHED)
 
 if __name__ == '__main__':
   unittest.main()