Fix header encoding in FakeNetworkReply test stub.

This was broken since ef9ddb2d5f but we never
noticed it, as an __init__.py was missing in the test folder since
06ec1a3885.
This commit is contained in:
Florian Bruhin 2015-02-05 07:12:36 +01:00
parent 0957d5df8e
commit 370c182f48

View File

@ -136,22 +136,23 @@ class FakeNetworkReply:
"""Check if the reply has a certain header. """Check if the reply has a certain header.
Args: Args:
name: The name of the header. name: The name of the header as ISO-8859-1 encoded bytes object.
Return: Return:
True if the header is present, False if not. True if the header is present, False if not.
""" """
return name in self.headers return name.decode('iso-8859-1') in self.headers
def rawHeader(self, name): def rawHeader(self, name):
"""Get the raw header data of a header. """Get the raw header data of a header.
Args: Args:
name: The name of the header. name: The name of the header as ISO-8859-1 encoded bytes object.
Return: Return:
The header data, as ISO-8859-1 encoded bytes() object. The header data, as ISO-8859-1 encoded bytes() object.
""" """
name = name.decode('iso-8859-1')
return self.headers[name].encode('iso-8859-1') return self.headers[name].encode('iso-8859-1')
def header(self, known_header): def header(self, known_header):