"""Test get cloud transfer progress."""

import json
from unittest.mock import MagicMock

import boto3
from pytest_mock import MockerFixture

from video import config
from video.models.s3 import cloud_transfer_progress


def test_get_cloud_transfer_progress(mocker: MockerFixture) -> None:
    """Test get cloud transfer progress."""
    mock_client: MagicMock = mocker.patch.object(boto3, "client", autospec=True)
    (
        mock_client.return_value.get_object.return_value["Body"].read.return_value
    ) = json.dumps({"a": "b"})
    assert {"a": "b"} == cloud_transfer_progress.get_cloud_transfer_progress(
        123456, "test_file"
    )
    mock_client.assert_called_with("s3")
    mock_client.return_value.get_object.assert_called_with(
        Bucket=config.VIDEO_BUCKET_NAME, Key="raw/123456/test_file.progress.json"
    )