def test_exception_thrown_if_request_contains_no_file(self):
# GIVEN a request without any files.
builder = EnvironBuilder(method='POST')
request = Request(builder.get_environ())
# EXPECT an exception is thrown when we process the request.
util.get_filestream_from_request(request)
def test_extracting_filestream_from_request(self):
# GIVEN a file to upload.
with tempfile.NamedTemporaryFile('w+b') as f:
# GIVEN the file's contents.
content = b'foo'
f.write(content)
f.seek(0)
# GIVEN a POST request containing the file to upload.
builder = EnvironBuilder(method='POST', data={'file': f})
request = Request(builder.get_environ())
# WHEN we process the request.
filestream = util.get_filestream_from_request(request)
# THEN the file-stream should correspond to the provided file.
self.assertEqual(content, filestream.read())
