Skip to content

Commit

Permalink
Merge pull request #17 from F-Secure/master
Browse files Browse the repository at this point in the history
Always return binary-strings from _get_data()
  • Loading branch information
tarttelin committed Sep 28, 2017
2 parents c050c08 + f0344aa commit f4ac61e
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 3 deletions.
2 changes: 1 addition & 1 deletion stubserver/webserver.py
Original file line number Diff line number Diff line change
Expand Up @@ -181,7 +181,7 @@ def __init__(self, expectations):
def _get_data(self):
max_chunk_size = 10 * 1024 * 1024
if "content-length" not in self.headers:
return ""
return b''
size_remaining = int(self.headers["content-length"])
data = []
while size_remaining:
Expand Down
7 changes: 5 additions & 2 deletions test.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,10 +19,13 @@ def tearDown(self):
self.server.stop()
self.server.verify() # this is redundant because stop includes verify

def _make_request(self, url, method="GET", payload="", headers={}):
def _make_request(self, url, method="GET", payload=None, headers={}):
self.opener = OpenerDirector()
self.opener.add_handler(HTTPHandler())
request = Request(url, headers=headers, data=payload.encode('utf-8'))
if payload is None:
request = Request(url, headers=headers)
else:
request = Request(url, headers=headers, data=payload.encode('utf-8'))
request.get_method = lambda: method
response = self.opener.open(request)
response_code = getattr(response, 'code', -1)
Expand Down

0 comments on commit f4ac61e

Please sign in to comment.