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"