diff options
Diffstat (limited to 'tests/unit/utils.py')
| -rw-r--r-- | tests/unit/utils.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/unit/utils.py b/tests/unit/utils.py index 88d6d12..0056993 100644 --- a/tests/unit/utils.py +++ b/tests/unit/utils.py @@ -154,7 +154,10 @@ def fake_http_connect(*code_iter, **kwargs): sleep(0.1) return ' ' rv = self.body[:amt] - self.body = self.body[amt:] + if amt is not None: + self.body = self.body[amt:] + else: + self.body = '' return rv def send(self, amt=None): |