33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
|
"""cos_url_keys：仅当 host 匹配配置时才解析 key。"""
|
|||
|
|
|
|||
|
|
import unittest
|
|||
|
|
from unittest.mock import patch
|
|||
|
|
|
|||
|
|
from app.core.cos_url_keys import extract_cos_object_key_if_owned
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TestExtractCosObjectKeyIfOwned(unittest.TestCase):
|
|||
|
|
def test_non_http_returns_none(self):
|
|||
|
|
self.assertIsNone(extract_cos_object_key_if_owned("audio-segment:x:0"))
|
|||
|
|
self.assertIsNone(extract_cos_object_key_if_owned(None))
|
|||
|
|
|
|||
|
|
@patch("app.core.cos_url_keys.settings")
|
|||
|
|
def test_matching_host_returns_key(self, mock_settings):
|
|||
|
|
mock_settings.tencent_cos_bucket = "mybucket"
|
|||
|
|
mock_settings.tencent_cos_region = "ap-shanghai"
|
|||
|
|
mock_settings.tencent_cos_base_url = ""
|
|||
|
|
url = "https://mybucket.cos.ap-shanghai.myqcloud.com/chapters/u1/c1/a.png"
|
|||
|
|
self.assertEqual(extract_cos_object_key_if_owned(url), "chapters/u1/c1/a.png")
|
|||
|
|
|
|||
|
|
@patch("app.core.cos_url_keys.settings")
|
|||
|
|
def test_foreign_host_returns_none(self, mock_settings):
|
|||
|
|
mock_settings.tencent_cos_bucket = "mybucket"
|
|||
|
|
mock_settings.tencent_cos_region = "ap-shanghai"
|
|||
|
|
mock_settings.tencent_cos_base_url = ""
|
|||
|
|
url = "https://evil.com/chapters/u1/c1/a.png"
|
|||
|
|
self.assertIsNone(extract_cos_object_key_if_owned(url))
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
unittest.main()
|