ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2024-12-30 01:38:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

handle more inputs in test body-producer

Browse Source
This commit is contained in:
meejah 2020-06-11 19:57:11 -06:00 committed by Sajith Sasidharan
parent 1f2ad6e634
commit c0ce716430
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/allmydata/testing/web.py
View File

@ -16,6 +16,7 @@ from twisted.internet.defer import (
from treq.client import (
HTTPClient,
FileBodyProducer,
)
from treq.testing import (
RequestTraversalAgent,
@ -192,9 +193,12 @@ class _SynchronousProducer(object):
"""
Create a synchronous producer with some bytes.
"""
if isinstance(body, FileBodyProducer):
body = body._inputFile.read()
if not isinstance(body, bytes):
raise ValueError(
"'body' must be bytes"
"'body' must be bytes not '{}'".format(type(body))
)
self.body = body
self.length = len(body)