diff --git a/atst/domain/csp/files.py b/atst/domain/csp/files.py index 905eee30..a74403e5 100644 --- a/atst/domain/csp/files.py +++ b/atst/domain/csp/files.py @@ -57,6 +57,7 @@ class RackspaceFileProvider(FileProviderInterface): object_name = uuid4().hex with NamedTemporaryFile() as tempfile: tempfile.write(fyle.stream.read()) + tempfile.seek(0) self.container.upload_object( file_path=tempfile.name, object_name=object_name, diff --git a/tests/domain/csp/test_files.py b/tests/domain/csp/test_files.py index 0f50cd11..66830a63 100644 --- a/tests/domain/csp/test_files.py +++ b/tests/domain/csp/test_files.py @@ -44,3 +44,14 @@ def test_download(app, uploader, pdf_upload): stream = uploader.download("abc") stream_content = b"".join([b for b in stream]) assert pdf_content == stream_content + + +def test_downloading_uploaded_object(uploader, pdf_upload): + object_name = uploader.upload(pdf_upload) + stream = uploader.download(object_name) + stream_content = b"".join([b for b in stream]) + + pdf_upload.seek(0) + pdf_content = pdf_upload.read() + + assert stream_content == pdf_content