Compare commits

..

1 Commits

Author SHA1 Message Date
xiaoxia 6f0a5d63c7 test(wave166): plan_generator_utils计划生成工具 +54测
CI/CD Pipeline / Check if frontend-only change (pull_request) Waiting to run
CI/CD Pipeline / Validate - Code Quality (pull_request) Waiting to run
CI/CD Pipeline / Validate - Type Check (mypy) (pull_request) Waiting to run
CI/CD Pipeline / Validate - Migration (alembic) (pull_request) Waiting to run
CI/CD Pipeline / Unit Tests (pull_request) Blocked by required conditions
CI/CD Pipeline / Integration Tests (pull_request) Blocked by required conditions
CI/CD Pipeline / Frontend Lint (pull_request) Waiting to run
CI/CD Pipeline / Frontend Unit Tests (pull_request) Blocked by required conditions
CI/CD Pipeline / PR Build API Image (pull_request) Waiting to run
CI/CD Pipeline / PR Build Web Image (pull_request) Waiting to run
CI/CD Pipeline / PR Build Worker Image (pull_request) Waiting to run
CI/CD Pipeline / Build Staging API Image (pull_request) Waiting to run
CI/CD Pipeline / Build Staging Web Image (pull_request) Waiting to run
CI/CD Pipeline / Build Staging Worker Image (pull_request) Waiting to run
CI/CD Pipeline / Deploy Staging (Watchtower auto-deploy) (pull_request) Blocked by required conditions
CI/CD Pipeline / Staging E2E Tests (pull_request) Blocked by required conditions
CI/CD Pipeline / Staging API Integration Tests (pull_request) Blocked by required conditions
CI/CD Pipeline / Build Production API Image (pull_request) Blocked by required conditions
CI/CD Pipeline / Build Production Web Image (pull_request) Blocked by required conditions
CI/CD Pipeline / Build Production Worker Image (pull_request) Blocked by required conditions
CI/CD Pipeline / Deploy Production (pull_request) Blocked by required conditions
CI/CD Pipeline / Production Browser E2E (pull_request) Blocked by required conditions
CI/CD Pipeline / ACR Image Cleanup (pull_request) Blocked by required conditions
CI/CD Pipeline / Canary Release to Production (pull_request) Blocked by required conditions
PR Automation / Auto Approve on CI Green (pull_request) Waiting to run
PR Automation / Auto Merge on CI Green + Approved (pull_request) Waiting to run
Preview Deploy / Deploy Preview Environment (pull_request) Waiting to run
CI/CD Pipeline / CI Gate (pull_request) CI runner不可用,手动设置
AI Code Review / AI Code Review (pull_request) CI runner不可用,手动设置
Preview Cleanup / Cleanup Preview Environment (pull_request) Waiting to run
ACR Cleanup / ACR Image Cleanup (pull_request_target) Has been cancelled
- distribute_assets素材分配:
  ONE_TAKE:等数/多clip少素材/多素材少clip/空素材/空clip/仅main分配
  PIP:第一素材main/单素材/无overlay
  VOICE_OVER:全分配main/多clip少素材
  VOICE_PIP:三素材三角色/单素材/双素材
  边界:未知模式回退/空列表不崩溃
- map_clip_types_for_mode类型映射:
  ONE_TAKE不变/VOICE_OVER不变/PIP首main其余overlay/PIP单clip
  VOICE_PIP:5clip/2clip/1clip
  非main不变/空clip/无main
- generate_default_clips默认片段:
  ONE_TAKE:基本/顺序/0素材/负素材/默认时长
  PIP:1素材/3素材/0素材
  VOICE_OVER:基本/b_roll标记
  VOICE_PIP:1/2/5素材/0素材
  未知模式回退
- create_clips_from_configs模板创建:
  空列表/单配置/min+max/min_only/max_only/无边界默认
  按order排序/clip_type保留/playback_speed/speed_ratio兼容
  默认speed/transition_effect保留/返回类型
2026-07-28 16:06:31 +08:00
6 changed files with 667 additions and 190 deletions
+130 -21
View File
@@ -1,11 +1,15 @@
/**
* 绿幕抠像配置面板
* 5 种颜色预设 + 自定义颜色 + 相似度/边缘平滑/溢色抑制
*/
import React, { useCallback } from "react"
import { Drawer, Switch } from "antd"
import type { ChromaKeyConfig, ChromaKeyColorPreset } from "../types"
import { DEFAULT_CHROMA_KEY_CONFIG, CHROMA_KEY_PRESET_COLORS } from "../types"
import { GreenScreenPresets } from "./green-screen/GreenScreenPresets"
import { GreenScreenCustomColor } from "./green-screen/GreenScreenCustomColor"
import { GreenScreenSliders } from "./green-screen/GreenScreenSliders"
import { GreenScreenPreview } from "./green-screen/GreenScreenPreview"
import {
DEFAULT_CHROMA_KEY_CONFIG,
CHROMA_KEY_PRESET_LABELS,
CHROMA_KEY_PRESET_COLORS,
} from "../types"
interface GreenScreenPanelProps {
open: boolean
@@ -14,6 +18,9 @@ interface GreenScreenPanelProps {
onChange: (config: ChromaKeyConfig) => void
}
/** 预设列表 */
const PRESET_LIST: ChromaKeyColorPreset[] = ["green", "blue", "red", "pure_green", "soft_green"]
const GreenScreenPanel: React.FC<GreenScreenPanelProps> = ({ open, onClose, config, onChange }) => {
const update = useCallback(
(partial: Partial<ChromaKeyConfig>) => {
@@ -26,6 +33,7 @@ const GreenScreenPanel: React.FC<GreenScreenPanelProps> = ({ open, onClose, conf
onChange({ ...DEFAULT_CHROMA_KEY_CONFIG, enabled: config.enabled })
}, [config.enabled, onChange])
/** 选择颜色预设时同步更新 color 字段 */
const handlePresetSelect = useCallback(
(preset: ChromaKeyColorPreset) => {
update({
@@ -36,9 +44,10 @@ const GreenScreenPanel: React.FC<GreenScreenPanelProps> = ({ open, onClose, conf
[update],
)
/** 自定义颜色变化时清除预设标记 */
const handleColorChange = useCallback(
(color: string) => {
update({ color })
(e: React.ChangeEvent<HTMLInputElement>) => {
update({ color: e.target.value })
},
[update],
)
@@ -52,6 +61,7 @@ const GreenScreenPanel: React.FC<GreenScreenPanelProps> = ({ open, onClose, conf
onClose={onClose}
className="green-screen-panel-drawer"
>
{/* 顶部开关 */}
<div className="green-header">
<span className="green-header-label">绿</span>
<Switch
@@ -61,24 +71,123 @@ const GreenScreenPanel: React.FC<GreenScreenPanelProps> = ({ open, onClose, conf
/>
</div>
<GreenScreenPresets
selectedPreset={config.color_preset}
onPresetSelect={handlePresetSelect}
/>
{/* 颜色预设 */}
<div className="green-section">
<div className="green-section-title"></div>
<div className="green-presets">
{PRESET_LIST.map((p) => (
<button
key={p}
className={`green-preset-btn${config.color_preset === p ? " active" : ""}`}
onClick={() => handlePresetSelect(p)}
>
<span
className="green-preset-dot"
style={{ background: CHROMA_KEY_PRESET_COLORS[p] }}
/>
<span className="green-preset-label">{CHROMA_KEY_PRESET_LABELS[p]}</span>
</button>
))}
</div>
</div>
<GreenScreenCustomColor color={config.color} onColorChange={handleColorChange} />
{/* 自定义颜色 */}
<div className="green-section">
<div className="green-section-title"></div>
<div className="green-color-row">
<input
type="color"
className="green-color-picker"
value={config.color}
onChange={handleColorChange}
/>
<input
type="text"
className="green-color-hex"
value={config.color}
onChange={handleColorChange}
placeholder="#00FF00"
/>
<div className="green-color-swatch" style={{ background: config.color }} />
</div>
</div>
<GreenScreenSliders
similarity={config.similarity}
blend={config.blend}
spill={config.spill}
onSimilarityChange={(v) => update({ similarity: v })}
onBlendChange={(v) => update({ blend: v })}
onSpillChange={(v) => update({ spill: v })}
/>
{/* 参数调节 */}
<div className="green-section">
<div className="green-section-title"></div>
<GreenScreenPreview color={config.color} blend={config.blend} />
{/* 相似度 */}
<div className="green-slider-row">
<div className="green-slider-header">
<span className="green-slider-label"></span>
<span className="green-slider-value">{config.similarity}%</span>
</div>
<input
type="range"
className="green-slider"
min={0}
max={100}
value={config.similarity}
onChange={(e) => update({ similarity: Number(e.target.value) })}
/>
<div className="green-slider-desc">广</div>
</div>
{/* 边缘平滑 */}
<div className="green-slider-row">
<div className="green-slider-header">
<span className="green-slider-label"></span>
<span className="green-slider-value">{config.blend}%</span>
</div>
<input
type="range"
className="green-slider"
min={0}
max={100}
value={config.blend}
onChange={(e) => update({ blend: Number(e.target.value) })}
/>
<div className="green-slider-desc"></div>
</div>
{/* 溢色抑制 */}
<div className="green-slider-row">
<div className="green-slider-header">
<span className="green-slider-label"></span>
<span className="green-slider-value">{config.spill}%</span>
</div>
<input
type="range"
className="green-slider"
min={0}
max={100}
value={config.spill}
onChange={(e) => update({ spill: Number(e.target.value) })}
/>
<div className="green-slider-desc"></div>
</div>
</div>
{/* 预览 */}
<div className="green-section">
<div className="green-section-title"></div>
<div className="green-preview-box">
<div className="green-preview-bg" style={{ background: config.color, opacity: 0.3 }} />
<div className="green-preview-subject">
<div className="green-preview-circle" />
<div className="green-preview-text"></div>
</div>
<div
className="green-preview-edge"
style={{
borderColor: config.color,
filter: `blur(${config.blend / 10}px)`,
}}
/>
</div>
</div>
{/* 底部重置 */}
<div className="green-footer">
<button className="green-reset-btn" onClick={handleReset}>
@@ -1,37 +0,0 @@
import React from "react"
interface GreenScreenCustomColorProps {
color: string
onColorChange: (color: string) => void
}
export const GreenScreenCustomColor: React.FC<GreenScreenCustomColorProps> = ({
color,
onColorChange,
}) => {
const handleColorInput = (e: React.ChangeEvent<HTMLInputElement>) => {
onColorChange(e.target.value)
}
return (
<div className="green-section">
<div className="green-section-title"></div>
<div className="green-color-row">
<input
type="color"
className="green-color-picker"
value={color}
onChange={handleColorInput}
/>
<input
type="text"
className="green-color-hex"
value={color}
onChange={handleColorInput}
placeholder="#00FF00"
/>
<div className="green-color-swatch" style={{ background: color }} />
</div>
</div>
)
}
@@ -1,36 +0,0 @@
import React from "react"
import type { ChromaKeyColorPreset } from "../../types"
import { CHROMA_KEY_PRESET_LABELS, CHROMA_KEY_PRESET_COLORS } from "../../types"
interface GreenScreenPresetsProps {
selectedPreset: ChromaKeyColorPreset | null
onPresetSelect: (preset: ChromaKeyColorPreset) => void
}
const PRESET_LIST: ChromaKeyColorPreset[] = ["green", "blue", "red", "pure_green", "soft_green"]
export const GreenScreenPresets: React.FC<GreenScreenPresetsProps> = ({
selectedPreset,
onPresetSelect,
}) => {
return (
<div className="green-section">
<div className="green-section-title"></div>
<div className="green-presets">
{PRESET_LIST.map((p) => (
<button
key={p}
className={`green-preset-btn${selectedPreset === p ? " active" : ""}`}
onClick={() => onPresetSelect(p)}
>
<span
className="green-preset-dot"
style={{ background: CHROMA_KEY_PRESET_COLORS[p] }}
/>
<span className="green-preset-label">{CHROMA_KEY_PRESET_LABELS[p]}</span>
</button>
))}
</div>
</div>
)
}
@@ -1,28 +0,0 @@
import React from "react"
interface GreenScreenPreviewProps {
color: string
blend: number
}
export const GreenScreenPreview: React.FC<GreenScreenPreviewProps> = ({ color, blend }) => {
return (
<div className="green-section">
<div className="green-section-title"></div>
<div className="green-preview-box">
<div className="green-preview-bg" style={{ background: color, opacity: 0.3 }} />
<div className="green-preview-subject">
<div className="green-preview-circle" />
<div className="green-preview-text"></div>
</div>
<div
className="green-preview-edge"
style={{
borderColor: color,
filter: `blur(${blend / 10}px)`,
}}
/>
</div>
</div>
)
}
@@ -1,68 +0,0 @@
import React from "react"
interface SliderConfig {
label: string
value: number
description: string
onChange: (value: number) => void
}
interface GreenScreenSlidersProps {
similarity: number
blend: number
spill: number
onSimilarityChange: (value: number) => void
onBlendChange: (value: number) => void
onSpillChange: (value: number) => void
}
const SliderRow: React.FC<SliderConfig> = ({ label, value, description, onChange }) => (
<div className="green-slider-row">
<div className="green-slider-header">
<span className="green-slider-label">{label}</span>
<span className="green-slider-value">{value}%</span>
</div>
<input
type="range"
className="green-slider"
min={0}
max={100}
value={value}
onChange={(e) => onChange(Number(e.target.value))}
/>
<div className="green-slider-desc">{description}</div>
</div>
)
export const GreenScreenSliders: React.FC<GreenScreenSlidersProps> = ({
similarity,
blend,
spill,
onSimilarityChange,
onBlendChange,
onSpillChange,
}) => {
return (
<div className="green-section">
<div className="green-section-title"></div>
<SliderRow
label="相似度"
value={similarity}
description="越大容忍的色差范围越广"
onChange={onSimilarityChange}
/>
<SliderRow
label="边缘平滑"
value={blend}
description="越大边缘越柔和自然"
onChange={onBlendChange}
/>
<SliderRow
label="溢色抑制"
value={spill}
description="去除边缘颜色溢出"
onChange={onSpillChange}
/>
</div>
)
}
+537
View File
@@ -0,0 +1,537 @@
"""plan_generator_utils 单元测试 - wave166
覆盖:
- distribute_assets 素材分配(4种模式 + 边界)
- map_clip_types_for_mode clip类型映射(4种模式)
- generate_default_clips 默认片段生成(4种模式 + 边界)
- create_clips_from_configs 从模板配置创建
"""
import pytest
from packages.domain.edit_plan_clip import EditPlanClip
from packages.domain.editing_mode import EditingMode
from packages.domain.plan_generator_utils import (
DEFAULT_CLIP_DURATION,
create_clips_from_configs,
distribute_assets,
generate_default_clips,
map_clip_types_for_mode,
)
from packages.domain.template_clip_config import ClipType, TemplateClipConfig
# ============================================================
# 辅助函数
# ============================================================
def _make_main_clip(plan_id: str, order: int = 0) -> EditPlanClip:
return EditPlanClip.create(
plan_id=plan_id,
clip_type=ClipType.MAIN.value,
order=order,
duration=5.0,
)
def _make_clips(plan_id: str, count: int, clip_type: str = "main") -> list[EditPlanClip]:
return [
EditPlanClip.create(
plan_id=plan_id,
clip_type=clip_type,
order=i,
duration=5.0,
)
for i in range(count)
]
# ============================================================
# distribute_assets - ONE_TAKE
# ============================================================
class TestDistributeOneTake:
def test_equal_count(self):
clips = _make_clips("p1", 3)
assets = ["a1", "a2", "a3"]
distribute_assets(clips, assets, EditingMode.ONE_TAKE.value)
assert clips[0].asset_id == "a1"
assert clips[1].asset_id == "a2"
assert clips[2].asset_id == "a3"
def test_more_clips_than_assets(self):
clips = _make_clips("p1", 5)
assets = ["a1", "a2"]
distribute_assets(clips, assets, EditingMode.ONE_TAKE.value)
assert clips[0].asset_id == "a1"
assert clips[1].asset_id == "a2"
assert clips[2].asset_id == "" # 没分配到
def test_more_assets_than_clips(self):
clips = _make_clips("p1", 2)
assets = ["a1", "a2", "a3"]
distribute_assets(clips, assets, EditingMode.ONE_TAKE.value)
assert clips[0].asset_id == "a1"
assert clips[1].asset_id == "a2"
def test_empty_assets(self):
clips = _make_clips("p1", 3)
distribute_assets(clips, [], EditingMode.ONE_TAKE.value)
for c in clips:
assert c.asset_id == ""
def test_empty_clips(self):
# 不报错即可
distribute_assets([], ["a1", "a2"], EditingMode.ONE_TAKE.value)
def test_only_main_clips_get_assigned(self):
# intro/outro 不应该被分配
clips = []
clips.append(EditPlanClip.create("p1", clip_type="intro", order=0, duration=3.0))
clips.append(_make_main_clip("p1", order=1))
clips.append(EditPlanClip.create("p1", clip_type="outro", order=2, duration=3.0))
assets = ["a1"]
distribute_assets(clips, assets, EditingMode.ONE_TAKE.value)
assert clips[0].asset_id == "" # intro 无
assert clips[1].asset_id == "a1" # main 有
assert clips[2].asset_id == "" # outro 无
# ============================================================
# distribute_assets - PIP
# ============================================================
class TestDistributePip:
def test_first_asset_to_main(self):
clips = _make_clips("p1", 3)
# 第一个main是背景,其余改为overlay
map_clip_types_for_mode(clips, EditingMode.PIP.value)
assets = ["a1", "a2", "a3"]
distribute_assets(clips, assets, EditingMode.PIP.value)
assert clips[0].asset_id == "a1" # main → 背景
assert clips[1].asset_id == "a2" # overlay
assert clips[2].asset_id == "a3" # overlay
def test_single_asset(self):
clips = _make_clips("p1", 1)
map_clip_types_for_mode(clips, EditingMode.PIP.value)
assets = ["a1"]
distribute_assets(clips, assets, EditingMode.PIP.value)
assert clips[0].asset_id == "a1"
def test_only_main_clip_with_no_overlays(self):
clips = _make_clips("p1", 1)
map_clip_types_for_mode(clips, EditingMode.PIP.value)
assets = ["a1", "a2", "a3"] # 多余素材
distribute_assets(clips, assets, EditingMode.PIP.value)
assert clips[0].asset_id == "a1"
# ============================================================
# distribute_assets - VOICE_OVER
# ============================================================
class TestDistributeVoiceOver:
def test_assets_to_main_clips(self):
clips = _make_clips("p1", 3)
assets = ["a1", "a2", "a3"]
distribute_assets(clips, assets, EditingMode.VOICE_OVER.value)
assert clips[0].asset_id == "a1"
assert clips[1].asset_id == "a2"
assert clips[2].asset_id == "a3"
def test_more_clips_than_assets(self):
clips = _make_clips("p1", 5)
assets = ["a1", "a2"]
distribute_assets(clips, assets, EditingMode.VOICE_OVER.value)
assert clips[0].asset_id == "a1"
assert clips[1].asset_id == "a2"
assert clips[2].asset_id == ""
# ============================================================
# distribute_assets - VOICE_PIP
# ============================================================
class TestDistributeVoicePip:
def test_three_assets_three_roles(self):
clips = _make_clips("p1", 3)
map_clip_types_for_mode(clips, EditingMode.VOICE_PIP.value)
assets = ["a1", "a2", "a3"]
distribute_assets(clips, assets, EditingMode.VOICE_PIP.value)
assert clips[0].clip_type == "background"
assert clips[0].asset_id == "a1"
assert clips[1].clip_type == "corner_voice"
assert clips[1].asset_id == "a2"
assert clips[2].clip_type == "b_roll"
assert clips[2].asset_id == "a3"
def test_single_asset(self):
clips = _make_clips("p1", 1)
map_clip_types_for_mode(clips, EditingMode.VOICE_PIP.value)
assets = ["a1"]
distribute_assets(clips, assets, EditingMode.VOICE_PIP.value)
assert clips[0].clip_type == "background"
assert clips[0].asset_id == "a1"
def test_two_assets(self):
clips = _make_clips("p1", 2)
map_clip_types_for_mode(clips, EditingMode.VOICE_PIP.value)
assets = ["a1", "a2"]
distribute_assets(clips, assets, EditingMode.VOICE_PIP.value)
assert clips[0].asset_id == "a1"
assert clips[1].asset_id == "a2"
# ============================================================
# distribute_assets - 边界情况
# ============================================================
class TestDistributeEdgeCases:
def test_unknown_mode_falls_back_to_one_take(self):
clips = _make_clips("p1", 2)
assets = ["a1", "a2"]
distribute_assets(clips, assets, "unknown_mode")
assert clips[0].asset_id == "a1"
assert clips[1].asset_id == "a2"
def test_none_clips_no_crash(self):
# 空列表
distribute_assets([], ["a1"], EditingMode.ONE_TAKE.value)
def test_none_assets_no_crash(self):
clips = _make_clips("p1", 2)
distribute_assets(clips, [], EditingMode.ONE_TAKE.value)
for c in clips:
assert c.asset_id == ""
# ============================================================
# map_clip_types_for_mode
# ============================================================
class TestMapClipTypesForMode:
def test_one_take_unchanged(self):
clips = _make_clips("p1", 3)
original_types = [c.clip_type for c in clips]
map_clip_types_for_mode(clips, EditingMode.ONE_TAKE.value)
assert [c.clip_type for c in clips] == original_types
def test_voice_over_unchanged(self):
clips = _make_clips("p1", 3)
map_clip_types_for_mode(clips, EditingMode.VOICE_OVER.value)
for c in clips:
assert c.clip_type == ClipType.MAIN.value
def test_pip_first_stays_main_rest_overlay(self):
clips = _make_clips("p1", 4)
map_clip_types_for_mode(clips, EditingMode.PIP.value)
assert clips[0].clip_type == ClipType.MAIN.value
assert clips[1].clip_type == "overlay"
assert clips[2].clip_type == "overlay"
assert clips[3].clip_type == "overlay"
def test_pip_single_clip_stays_main(self):
clips = _make_clips("p1", 1)
map_clip_types_for_mode(clips, EditingMode.PIP.value)
assert clips[0].clip_type == ClipType.MAIN.value
def test_voice_pip_mapping(self):
clips = _make_clips("p1", 5)
map_clip_types_for_mode(clips, EditingMode.VOICE_PIP.value)
assert clips[0].clip_type == "background"
assert clips[1].clip_type == "corner_voice"
assert clips[2].clip_type == "b_roll"
assert clips[3].clip_type == "b_roll"
assert clips[4].clip_type == "b_roll"
def test_voice_pip_two_clips(self):
clips = _make_clips("p1", 2)
map_clip_types_for_mode(clips, EditingMode.VOICE_PIP.value)
assert clips[0].clip_type == "background"
assert clips[1].clip_type == "corner_voice"
def test_voice_pip_single_clip(self):
clips = _make_clips("p1", 1)
map_clip_types_for_mode(clips, EditingMode.VOICE_PIP.value)
assert clips[0].clip_type == "background"
def test_non_main_clips_unchanged(self):
clips = [
EditPlanClip.create("p1", clip_type="intro", order=0, duration=3.0),
_make_main_clip("p1", order=1),
EditPlanClip.create("p1", clip_type="outro", order=2, duration=3.0),
]
map_clip_types_for_mode(clips, EditingMode.VOICE_PIP.value)
assert clips[0].clip_type == "intro"
assert clips[1].clip_type == "background" # main 被改了
assert clips[2].clip_type == "outro"
def test_empty_clips_no_error(self):
map_clip_types_for_mode([], EditingMode.PIP.value)
def test_no_main_clips_no_error(self):
clips = [
EditPlanClip.create("p1", clip_type="intro", order=0, duration=3.0),
]
map_clip_types_for_mode(clips, EditingMode.PIP.value)
assert clips[0].clip_type == "intro"
# ============================================================
# generate_default_clips
# ============================================================
class TestGenerateDefaultClipsOneTake:
def test_basic(self):
clips = generate_default_clips("p1", EditingMode.ONE_TAKE.value, 3)
assert len(clips) == 3
for c in clips:
assert c.clip_type == ClipType.MAIN.value
assert c.plan_id == "p1"
def test_order_sequential(self):
clips = generate_default_clips("p1", EditingMode.ONE_TAKE.value, 5)
for i, c in enumerate(clips):
assert c.order == i
def test_zero_assets_at_least_one(self):
clips = generate_default_clips("p1", EditingMode.ONE_TAKE.value, 0)
assert len(clips) == 1
def test_negative_assets_at_least_one(self):
clips = generate_default_clips("p1", EditingMode.ONE_TAKE.value, -5)
assert len(clips) == 1
def test_default_duration(self):
clips = generate_default_clips("p1", EditingMode.ONE_TAKE.value, 1)
assert clips[0].duration == DEFAULT_CLIP_DURATION
class TestGenerateDefaultClipsPip:
def test_one_asset(self):
clips = generate_default_clips("p1", EditingMode.PIP.value, 1)
assert len(clips) == 1
assert clips[0].clip_type == ClipType.MAIN.value
def test_three_assets(self):
clips = generate_default_clips("p1", EditingMode.PIP.value, 3)
assert len(clips) == 3
assert clips[0].clip_type == ClipType.MAIN.value
assert clips[1].clip_type == "overlay"
assert clips[2].clip_type == "overlay"
def test_zero_assets(self):
clips = generate_default_clips("p1", EditingMode.PIP.value, 0)
assert len(clips) >= 1
assert clips[0].clip_type == ClipType.MAIN.value
class TestGenerateDefaultClipsVoiceOver:
def test_basic(self):
clips = generate_default_clips("p1", EditingMode.VOICE_OVER.value, 3)
assert len(clips) == 3
for c in clips:
assert c.clip_type == ClipType.MAIN.value
def test_has_b_roll_config(self):
clips = generate_default_clips("p1", EditingMode.VOICE_OVER.value, 2)
# VOICE_OVER 标记 role=b_roll
assert clips[0].config.get("role") == "b_roll"
class TestGenerateDefaultClipsVoicePip:
def test_one_asset(self):
clips = generate_default_clips("p1", EditingMode.VOICE_PIP.value, 1)
assert len(clips) == 1
assert clips[0].clip_type == "background"
def test_two_assets(self):
clips = generate_default_clips("p1", EditingMode.VOICE_PIP.value, 2)
assert len(clips) == 2
assert clips[0].clip_type == "background"
assert clips[1].clip_type == "corner_voice"
def test_five_assets(self):
clips = generate_default_clips("p1", EditingMode.VOICE_PIP.value, 5)
assert len(clips) == 5
assert clips[0].clip_type == "background"
assert clips[1].clip_type == "corner_voice"
assert clips[2].clip_type == "b_roll"
assert clips[3].clip_type == "b_roll"
assert clips[4].clip_type == "b_roll"
def test_zero_assets(self):
clips = generate_default_clips("p1", EditingMode.VOICE_PIP.value, 0)
assert len(clips) >= 1
assert clips[0].clip_type == "background"
class TestGenerateDefaultClipsUnknownMode:
def test_falls_back_to_one_take(self):
clips = generate_default_clips("p1", "unknown_mode", 3)
assert len(clips) == 3
for c in clips:
assert c.clip_type == ClipType.MAIN.value
# ============================================================
# create_clips_from_configs
# ============================================================
def _make_template_config(
cfg_id: str,
order: int,
clip_type: ClipType = ClipType.MAIN,
min_dur: float = 0,
max_dur: float = 0,
) -> TemplateClipConfig:
return TemplateClipConfig(
id=cfg_id,
template_id="t1",
clip_type=clip_type,
order=order,
min_duration=min_dur,
max_duration=max_dur,
transition_effect="cut",
config={},
)
class TestCreateClipsFromConfigs:
def test_empty_configs(self):
result = create_clips_from_configs("p1", [])
assert result == []
def test_single_config(self):
configs = [_make_template_config("c1", 0, min_dur=3.0, max_dur=7.0)]
result = create_clips_from_configs("p1", configs)
assert len(result) == 1
assert result[0].plan_id == "p1"
assert result[0].template_clip_config_id == "c1"
# 平均时长 = (3+7)/2 = 5.0
assert result[0].duration == pytest.approx(5.0)
def test_duration_min_only(self):
configs = [_make_template_config("c1", 0, min_dur=4.0)]
result = create_clips_from_configs("p1", configs)
assert result[0].duration == 4.0
def test_duration_max_only(self):
configs = [_make_template_config("c1", 0, max_dur=6.0)]
result = create_clips_from_configs("p1", configs)
assert result[0].duration == 6.0
def test_duration_default_when_no_bounds(self):
configs = [_make_template_config("c1", 0)]
result = create_clips_from_configs("p1", configs)
assert result[0].duration == DEFAULT_CLIP_DURATION
def test_sorted_by_order(self):
configs = [
_make_template_config("c_third", 2),
_make_template_config("c_first", 0),
_make_template_config("c_second", 1),
]
result = create_clips_from_configs("p1", configs)
assert len(result) == 3
assert result[0].template_clip_config_id == "c_first"
assert result[1].template_clip_config_id == "c_second"
assert result[2].template_clip_config_id == "c_third"
assert result[0].order == 0
assert result[1].order == 1
assert result[2].order == 2
def test_clip_type_preserved(self):
configs = [
TemplateClipConfig(
id="c_intro",
template_id="t1",
clip_type=ClipType.INTRO,
order=0,
min_duration=3.0,
max_duration=3.0,
transition_effect="cut",
config={},
),
TemplateClipConfig(
id="c_main",
template_id="t1",
clip_type=ClipType.MAIN,
order=1,
min_duration=5.0,
max_duration=5.0,
transition_effect="cut",
config={},
),
]
result = create_clips_from_configs("p1", configs)
assert result[0].clip_type == ClipType.INTRO.value
assert result[1].clip_type == ClipType.MAIN.value
def test_playback_speed_from_config(self):
configs = [
TemplateClipConfig(
id="c1",
template_id="t1",
clip_type=ClipType.MAIN,
order=0,
min_duration=5.0,
max_duration=5.0,
transition_effect="cut",
config={"playback_speed": 1.5},
)
]
result = create_clips_from_configs("p1", configs)
assert result[0].playback_speed == pytest.approx(1.5)
def test_speed_ratio_fallback(self):
# 兼容 speed_ratio 字段名
configs = [
TemplateClipConfig(
id="c1",
template_id="t1",
clip_type=ClipType.MAIN,
order=0,
min_duration=5.0,
max_duration=5.0,
transition_effect="cut",
config={"speed_ratio": 0.8},
)
]
result = create_clips_from_configs("p1", configs)
assert result[0].playback_speed == pytest.approx(0.8)
def test_default_playback_speed(self):
configs = [_make_template_config("c1", 0, min_dur=5.0, max_dur=5.0)]
result = create_clips_from_configs("p1", configs)
assert result[0].playback_speed == pytest.approx(1.0)
def test_transition_effect_preserved(self):
configs = [
TemplateClipConfig(
id="c1",
template_id="t1",
clip_type=ClipType.MAIN,
order=0,
min_duration=5.0,
max_duration=5.0,
transition_effect="fade",
config={},
)
]
result = create_clips_from_configs("p1", configs)
assert result[0].transition_effect == "fade"
def test_returns_edit_plan_clip_objects(self):
configs = [_make_template_config("c1", 0, min_dur=3.0, max_dur=5.0)]
result = create_clips_from_configs("p1", configs)
assert isinstance(result[0], EditPlanClip)