From 157f5b211fed1cfb9104dfd908d29a021705df74 Mon Sep 17 00:00:00 2001
From: Florian Bruhin <git@the-compiler.org>
Date: Tue, 12 Aug 2014 09:03:38 +0200
Subject: [PATCH] tests: Adjust FakeNetworkReply to work with all headers.

---
 qutebrowser/test/stubs.py                     | 54 ++++++++++++-------
 .../utils/http/test_content_disposition.py    | 12 ++---
 2 files changed, 40 insertions(+), 26 deletions(-)

diff --git a/qutebrowser/test/stubs.py b/qutebrowser/test/stubs.py
index 0e7cc1910..01b0ee118 100644
--- a/qutebrowser/test/stubs.py
+++ b/qutebrowser/test/stubs.py
@@ -25,6 +25,7 @@ from unittest.mock import Mock
 
 from PyQt5.QtCore import QPoint, QProcess
 from PyQt5.QtWebKit import QWebElement
+from PyQt5.QtNetwork import QNetworkRequest
 
 
 class ConfigStub:
@@ -205,10 +206,17 @@ class FakeNetworkReply:
 
     """QNetworkReply stub which provides a Content-Disposition header."""
 
-    def __init__(self, content_disposition=None, url=None):
+    KNOWN_HEADERS = {
+        QNetworkRequest.ContentTypeHeader: 'Content-Type',
+    }
+
+    def __init__(self, headers=None, url=None):
         if url is None:
             url = FakeUrl()
-        self._content_disposition = content_disposition
+        if headers is None:
+            self.headers = {}
+        else:
+            self.headers = headers
         self.url = Mock(return_value=url)
 
     def hasRawHeader(self, name):
@@ -219,15 +227,8 @@ class FakeNetworkReply:
 
         Return:
             True if the header is present, False if not.
-
-        Raise:
-            ValueError: If a header other than Content-Disposition is
-                        requested.
         """
-        if name == 'Content-Disposition':
-            return self._content_disposition is not None
-        else:
-            raise ValueError("Invalid header {}".format(name))
+        return name in self.headers
 
     def rawHeader(self, name):
         """Get the raw header data of a header.
@@ -237,17 +238,30 @@ class FakeNetworkReply:
 
         Return:
             The header data, as ISO-8859-1 encoded bytes() object.
-
-        Raise:
-            ValueError: If a header other than Content-Disposition is
-                        requested.
         """
-        if name != 'Content-Disposition':
-            raise ValueError("Invalid header {}".format(name))
-        cd = self._content_disposition
-        if cd is None:
-            raise ValueError("Content-Disposition is None!")
-        return cd.encode('iso-8859-1')
+        return self.headers[name].encode('iso-8859-1')
+
+    def header(self, known_header):
+        """Get a known header.
+
+        Args:
+            known_header: A QNetworkRequest::KnownHeaders member.
+        """
+        key = self.KNOWN_HEADERS[known_header]
+        try:
+            return self.headers[key]
+        except KeyError:
+            return None
+
+    def setHeader(self, known_header, value):
+        """Set a known header.
+
+        Args:
+            known_header: A QNetworkRequest::KnownHeaders member.
+            value: The value to set.
+        """
+        key = self.KNOWN_HEADERS[known_header]
+        self.headers[key] = value
 
 
 class FakeQProcess:
diff --git a/qutebrowser/test/utils/http/test_content_disposition.py b/qutebrowser/test/utils/http/test_content_disposition.py
index 08c705ef9..f49dac9c2 100644
--- a/qutebrowser/test/utils/http/test_content_disposition.py
+++ b/qutebrowser/test/utils/http/test_content_disposition.py
@@ -39,7 +39,7 @@ class AttachmentTestCase(unittest.TestCase):
 
     def _check_filename(self, header, filename):
         """Check if the passed header has the given filename."""
-        reply = FakeNetworkReply(header)
+        reply = FakeNetworkReply(headers={'Content-Disposition': header})
         cd_inline, cd_filename = httputils.parse_content_disposition(reply)
         self.assertIsNotNone(cd_filename)
         self.assertEqual(cd_filename, filename)
@@ -47,14 +47,14 @@ class AttachmentTestCase(unittest.TestCase):
 
     def _check_ignored(self, header):
         """Check if the passed header is ignored."""
-        reply = FakeNetworkReply(header)
+        reply = FakeNetworkReply(headers={'Content-Disposition': header})
         cd_inline, cd_filename = httputils.parse_content_disposition(reply)
         self.assertEqual(cd_filename, DEFAULT_NAME)
         self.assertTrue(cd_inline)
 
     def _check_unnamed(self, header):
         """Check if the passed header results in an unnamed attachment."""
-        reply = FakeNetworkReply(header)
+        reply = FakeNetworkReply(headers={'Content-Disposition': header})
         cd_inline, cd_filename = httputils.parse_content_disposition(reply)
         self.assertEqual(cd_filename, DEFAULT_NAME)
         self.assertFalse(cd_inline)
@@ -69,14 +69,14 @@ class InlineTests(unittest.TestCase):
 
     def _check_filename(self, header, filename):
         """Check if the passed header has the given filename."""
-        reply = FakeNetworkReply(header)
+        reply = FakeNetworkReply(headers={'Content-Disposition': header})
         cd_inline, cd_filename = httputils.parse_content_disposition(reply)
         self.assertEqual(cd_filename, filename)
         self.assertTrue(cd_inline)
 
     def _check_ignored(self, header):
         """Check if the passed header is ignored."""
-        reply = FakeNetworkReply(header)
+        reply = FakeNetworkReply(headers={'Content-Disposition': header})
         cd_inline, cd_filename = httputils.parse_content_disposition(reply)
         self.assertEqual(cd_filename, DEFAULT_NAME)
         self.assertTrue(cd_inline)
@@ -134,7 +134,7 @@ class AttachmentTests(AttachmentTestCase):
 
         UA should offer to download the resource.
         """
-        reply = FakeNetworkReply('attachment')
+        reply = FakeNetworkReply(headers={'Content-Disposition': 'attachment'})
         cd_inline, cd_filename = httputils.parse_content_disposition(reply)
         self.assertFalse(cd_inline)
         self.assertEqual(cd_filename, DEFAULT_NAME)