mirror of
https://github.com/Hommy-master/capcut-mate.git
synced 2026-08-30 17:48:47 +08:00
Hommy
4010029b56
# 火山引擎 TOS AccessKeyId（COS、OSS 均未配置完整时作为兜底） TOS_ACCESS_KEY_ID= # 火山引擎 TOS AccessKeySecret TOS_ACCESS_KEY_SECRET= # TOS 存储桶名称 TOS_BUCKET_NAME= # TOS 地域，例如 cn-beijing、cn-shanghai TOS_REGION= # TOS 访问域名（可选，未设置时按地域自动生成，例如 tos-cn-beijing.volces.com） TOS_ENDPOINT= 优先级排最后，当COS，OSS都没有配置时，才使用TOS
87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
"""Unit tests for src.utils.upload_file routing (COS / OSS / TOS selection)."""
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from exceptions import CustomError, CustomException
|
|
|
|
|
|
@pytest.fixture
|
|
def upload_mocks():
|
|
with (
|
|
patch("src.utils.upload_file.cos_upload_file") as m_cos,
|
|
patch("src.utils.upload_file.oss_upload_file") as m_oss,
|
|
patch("src.utils.upload_file.tos_upload_file") as m_tos,
|
|
patch("src.utils.upload_file._is_cos_configured") as m_ic,
|
|
patch("src.utils.upload_file._is_oss_configured") as m_io,
|
|
patch("src.utils.upload_file._is_tos_configured") as m_it,
|
|
):
|
|
yield m_cos, m_oss, m_tos, m_ic, m_io, m_it
|
|
|
|
|
|
def test_upload_file_prefers_cos_when_configured(upload_mocks) -> None:
|
|
m_cos, m_oss, m_tos, m_ic, m_io, m_it = upload_mocks
|
|
m_ic.return_value = True
|
|
m_io.return_value = True
|
|
m_it.return_value = True
|
|
m_cos.return_value = "https://cos.example/signed"
|
|
from src.utils.upload_file import upload_file
|
|
|
|
assert upload_file("/tmp/a.mp4") == "https://cos.example/signed"
|
|
m_cos.assert_called_once()
|
|
m_oss.assert_not_called()
|
|
m_tos.assert_not_called()
|
|
|
|
|
|
def test_upload_file_fallback_oss_when_cos_not_configured(upload_mocks) -> None:
|
|
m_cos, m_oss, m_tos, m_ic, m_io, m_it = upload_mocks
|
|
m_ic.return_value = False
|
|
m_io.return_value = True
|
|
m_it.return_value = True
|
|
m_oss.return_value = "https://oss.example/signed"
|
|
from src.utils.upload_file import upload_file
|
|
|
|
assert upload_file("/tmp/b.mp4") == "https://oss.example/signed"
|
|
m_oss.assert_called_once()
|
|
m_cos.assert_not_called()
|
|
m_tos.assert_not_called()
|
|
|
|
|
|
def test_upload_file_fallback_tos_when_cos_oss_not_configured(upload_mocks) -> None:
|
|
m_cos, m_oss, m_tos, m_ic, m_io, m_it = upload_mocks
|
|
m_ic.return_value = False
|
|
m_io.return_value = False
|
|
m_it.return_value = True
|
|
m_tos.return_value = "https://tos.example/signed"
|
|
from src.utils.upload_file import upload_file
|
|
|
|
assert upload_file("/tmp/c.mp4") == "https://tos.example/signed"
|
|
m_tos.assert_called_once()
|
|
m_cos.assert_not_called()
|
|
m_oss.assert_not_called()
|
|
|
|
|
|
def test_upload_file_neither_storage_raises(upload_mocks) -> None:
|
|
m_cos, m_oss, m_tos, m_ic, m_io, m_it = upload_mocks
|
|
m_ic.return_value = False
|
|
m_io.return_value = False
|
|
m_it.return_value = False
|
|
from src.utils.upload_file import upload_file
|
|
|
|
with pytest.raises(CustomException) as ei:
|
|
upload_file("/tmp/d.mp4")
|
|
assert ei.value.err == CustomError.INTERNAL_SERVER_ERROR
|
|
m_cos.assert_not_called()
|
|
m_oss.assert_not_called()
|
|
m_tos.assert_not_called()
|
|
|
|
|
|
def test_upload_file_passes_expire_days(upload_mocks) -> None:
|
|
m_cos, m_oss, m_tos, m_ic, m_io, m_it = upload_mocks
|
|
m_ic.return_value = True
|
|
m_cos.return_value = "u"
|
|
from src.utils.upload_file import upload_file
|
|
|
|
upload_file("/x", expire_days=14)
|
|
m_cos.assert_called_once_with(file_path="/x", expire_days=14)
|