6f0e82f351
- 平台治理: 租户用户权限层次、资源ACL、审批中心与审批模板、访问申请 - 存储: MinIO 存储进度迁移、对象存储安全加固与测试 - 计算: GPU 资源预留、compute 轮询与同步增强 - 权限: permission v2 迁移、权限安全验收测试 - 日志: 后端运行日志中文说明、操作日志整合 - 数据处理/评测: 数据转换与模型评测优化 Co-Authored-By: Claude <noreply@anthropic.com>
27 lines
792 B
Python
27 lines
792 B
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from app.api.v1.endpoints.platform import _validate_storage_key
|
|
|
|
|
|
def test_storage_key_is_scoped_to_resource_version() -> None:
|
|
assert (
|
|
_validate_storage_key("dataset", "ds_123", "v1", None, "train.jsonl")
|
|
== "datasets/ds_123/versions/v1/train.jsonl"
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"object_key",
|
|
[
|
|
"models/other/versions/v1/model.bin",
|
|
"datasets/ds_123/versions/v1/../secret.bin",
|
|
"datasets/ds_123/versions/v1/../../secret.bin",
|
|
"/datasets/ds_123/versions/v1/model.bin",
|
|
],
|
|
)
|
|
def test_storage_key_rejects_escape_or_cross_resource_paths(object_key: str) -> None:
|
|
with pytest.raises(ValueError):
|
|
_validate_storage_key("dataset", "ds_123", "v1", object_key, None)
|