summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBert JW Regeer <bertjw@regeer.org>2016-09-29 00:36:28 -0600
committerBert JW Regeer <bertjw@regeer.org>2016-09-29 19:07:54 -0600
commit4cce6d12ff6e11778613712cb3f9037cfec2b965 (patch)
tree0c2fd959ae8faa99a4170398a6bf794e2e79d9a4
parent64e3407066537e5b8a83208bbaa6d2274507a2f3 (diff)
downloadwebob-4cce6d12ff6e11778613712cb3f9037cfec2b965.tar.gz
Fix some test helpers
-rw-r--r--tests/test_request_nose.py12
1 files changed, 10 insertions, 2 deletions
diff --git a/tests/test_request_nose.py b/tests/test_request_nose.py
index a91f124..d67f96f 100644
--- a/tests/test_request_nose.py
+++ b/tests/test_request_nose.py
@@ -75,11 +75,16 @@ class ReadTracker(object):
def __init__(self, data):
self.data = data
self.was_read = False
+
def read(self, size=-1):
- if size < 0:
+ if size < 0 or size > len(self.data):
size = len(self.data)
- assert size == len(self.data)
+
+ if self.was_read:
+ return b''
+
self.was_read = True
+
return self.data
def test_limited_length_file_repr():
@@ -118,6 +123,9 @@ class _Helper_test_request_wrong_clen(object):
self.file_ended = True
return r
+ def seek(self, pos):
+ pass
+
def test_disconnect_detection_cgi():
data = 'abc'*(1<<20)
req = Request.blank('/', POST={'file':('test-file', data)})