e32e211966
Renames Python classes that refer to Upload to something to FileService. We added this change because these classes now handle downloading as well as uploading.
23 lines
751 B
Python
23 lines
751 B
Python
from atst.domain.csp.files import AzureFileService
|
|
from azure.storage.blob.models import Blob
|
|
|
|
|
|
class MockBlockBlobService(object):
|
|
def __init__(self, exception=None, **kwargs):
|
|
self.exception = exception
|
|
|
|
def get_blob_to_bytes(self, blob_name="test.pdf", **kwargs):
|
|
if self.exception:
|
|
raise self.exception
|
|
else:
|
|
return Blob(name=blob_name, content=b"mock content")
|
|
|
|
|
|
def test_download_task_order_success(app, monkeypatch):
|
|
file_service = AzureFileService(config=app.config)
|
|
file_service.BlockBlobService = MockBlockBlobService
|
|
|
|
task_order = file_service.download_task_order("test.pdf")
|
|
assert task_order["name"] == "test.pdf"
|
|
assert task_order["content"] == b"mock content"
|