"""watermark_config 领域模型单测.""" from __future__ import annotations import pytest from packages.domain.watermark_config import ( DEFAULT_FONT_COLOR, DEFAULT_FONT_SIZE, DEFAULT_MODE, DEFAULT_OPACITY, DEFAULT_POSITION, DEFAULT_SCALE, VALID_POSITIONS, WATERMARK_POSITIONS, WatermarkConfig, build_image_watermark_filter, build_text_watermark_filter, calc_position, calc_scroll_x, get_position_display_name, get_position_names, ) # ── 常量测试 ──────────────────────────────────────────────────────────────── class TestConstants: def test_nine_positions(self): assert len(WATERMARK_POSITIONS) == 9 def test_all_position_keys_valid(self): for key in WATERMARK_POSITIONS: assert key in VALID_POSITIONS def test_valid_positions_match(self): assert set(WATERMARK_POSITIONS.keys()) == VALID_POSITIONS def test_default_values(self): assert DEFAULT_POSITION == "bottom_right" assert DEFAULT_MODE == "text" assert DEFAULT_SCALE == 0.2 assert DEFAULT_OPACITY == 0.8 assert DEFAULT_FONT_SIZE == 24 assert DEFAULT_FONT_COLOR == "white" # ── WatermarkConfig.from_dict 测试 ───────────────────────────────────────── class TestWatermarkConfigFromDict: def test_none_returns_none(self): assert WatermarkConfig.from_dict(None) is None def test_empty_dict_returns_none(self): assert WatermarkConfig.from_dict({}) is None def test_disabled_returns_none(self): assert WatermarkConfig.from_dict({"enabled": False}) is None def test_text_mode_basic(self): cfg = WatermarkConfig.from_dict({"enabled": True, "mode": "text", "text": "hello"}) assert cfg is not None assert cfg.mode == "text" assert cfg.text == "hello" assert cfg.position == DEFAULT_POSITION def test_image_mode_basic(self): cfg = WatermarkConfig.from_dict({"enabled": True, "mode": "image", "image_path": "/tmp/wm.png"}) assert cfg is not None assert cfg.mode == "image" assert cfg.image_path == "/tmp/wm.png" def test_image_mode_accepts_image_key(self): cfg = WatermarkConfig.from_dict({"enabled": True, "mode": "image", "image": "/tmp/wm.png"}) assert cfg is not None assert cfg.image_path == "/tmp/wm.png" def test_image_mode_missing_path_returns_none(self): assert WatermarkConfig.from_dict({"enabled": True, "mode": "image"}) is None def test_text_mode_missing_text_returns_none(self): assert WatermarkConfig.from_dict({"enabled": True, "mode": "text"}) is None def test_text_mode_empty_text_returns_none(self): assert WatermarkConfig.from_dict({"enabled": True, "mode": "text", "text": ""}) is None def test_invalid_position_defaults(self): cfg = WatermarkConfig.from_dict({"enabled": True, "mode": "text", "text": "hi", "position": "invalid"}) assert cfg.position == DEFAULT_POSITION def test_custom_all_params(self): cfg = WatermarkConfig.from_dict( { "enabled": True, "mode": "text", "text": "测试水印", "position": "top_left", "font_size": 32, "font_color": "red", "opacity": 0.5, "margin_x": 30, "margin_y": 40, "scroll": True, "scroll_speed": 100, } ) assert cfg is not None assert cfg.text == "测试水印" assert cfg.position == "top_left" assert cfg.font_size == 32 assert cfg.font_color == "red" assert cfg.opacity == 0.5 assert cfg.margin_x == 30 assert cfg.margin_y == 40 assert cfg.scroll is True assert cfg.scroll_speed == 100 def test_default_mode_is_text(self): cfg = WatermarkConfig.from_dict({"enabled": True, "text": "hi"}) assert cfg is not None assert cfg.mode == "text" # ── WatermarkConfig.validate 测试 ────────────────────────────────────────── class TestWatermarkConfigValidate: def test_valid_text_config(self): cfg = WatermarkConfig(mode="text", text="hello") ok, msg = cfg.validate() assert ok is True assert msg == "" def test_valid_image_config(self): cfg = WatermarkConfig(mode="image", image_path="/tmp/wm.png", scale=0.3) ok, msg = cfg.validate() assert ok is True def test_invalid_position(self): cfg = WatermarkConfig(mode="text", text="hi", position="nowhere") ok, msg = cfg.validate() assert ok is False assert "位置" in msg def test_opacity_negative(self): cfg = WatermarkConfig(mode="text", text="hi", opacity=-0.1) ok, msg = cfg.validate() assert ok is False assert "透明度" in msg def test_opacity_over_one(self): cfg = WatermarkConfig(mode="text", text="hi", opacity=1.5) ok, msg = cfg.validate() assert ok is False def test_opacity_boundary_zero(self): cfg = WatermarkConfig(mode="text", text="hi", opacity=0.0) ok, _ = cfg.validate() assert ok is True def test_opacity_boundary_one(self): cfg = WatermarkConfig(mode="text", text="hi", opacity=1.0) ok, _ = cfg.validate() assert ok is True def test_image_missing_path(self): cfg = WatermarkConfig(mode="image") ok, msg = cfg.validate() assert ok is False assert "图片路径" in msg def test_image_scale_too_small(self): cfg = WatermarkConfig(mode="image", image_path="/a.png", scale=0.001) ok, msg = cfg.validate() assert ok is False assert "缩放比例" in msg def test_image_scale_too_large(self): cfg = WatermarkConfig(mode="image", image_path="/a.png", scale=2.0) ok, msg = cfg.validate() assert ok is False def test_image_scale_boundary_low(self): cfg = WatermarkConfig(mode="image", image_path="/a.png", scale=0.01) ok, _ = cfg.validate() assert ok is True def test_image_scale_boundary_high(self): cfg = WatermarkConfig(mode="image", image_path="/a.png", scale=1.0) ok, _ = cfg.validate() assert ok is True def test_text_missing_content(self): cfg = WatermarkConfig(mode="text", text="") ok, msg = cfg.validate() assert ok is False assert "文字内容" in msg def test_text_font_size_zero(self): cfg = WatermarkConfig(mode="text", text="hi", font_size=0) ok, msg = cfg.validate() assert ok is False assert "字体大小" in msg def test_text_font_size_negative(self): cfg = WatermarkConfig(mode="text", text="hi", font_size=-5) ok, msg = cfg.validate() assert ok is False def test_unknown_mode(self): cfg = WatermarkConfig(mode="video") ok, msg = cfg.validate() assert ok is False assert "模式" in msg # ── has_effect 测试 ──────────────────────────────────────────────────────── class TestHasEffect: def test_text_with_content_has_effect(self): cfg = WatermarkConfig(mode="text", text="hello") assert cfg.has_effect() is True def test_text_empty_no_effect(self): cfg = WatermarkConfig(mode="text", text="") assert cfg.has_effect() is False def test_text_zero_opacity_no_effect(self): cfg = WatermarkConfig(mode="text", text="hello", opacity=0.0) assert cfg.has_effect() is False def test_image_with_path_has_effect(self): cfg = WatermarkConfig(mode="image", image_path="/a.png") assert cfg.has_effect() is True def test_image_no_path_no_effect(self): cfg = WatermarkConfig(mode="image") assert cfg.has_effect() is False def test_unknown_mode_no_effect(self): cfg = WatermarkConfig(mode="unknown") assert cfg.has_effect() is False # ── calc_position 测试 ───────────────────────────────────────────────────── class TestCalcPosition: def test_top_left(self): x, y = calc_position("top_left", 1000, 2000, 100, 50, 10, 20) assert (x, y) == (10, 20) def test_top_center(self): x, y = calc_position("top_center", 1000, 2000, 100, 50, 10, 20) assert (x, y) == (450, 20) def test_top_right(self): x, y = calc_position("top_right", 1000, 2000, 100, 50, 10, 20) assert (x, y) == (890, 20) def test_center_left(self): x, y = calc_position("center_left", 1000, 2000, 100, 50, 10, 20) assert (x, y) == (10, 975) def test_center(self): x, y = calc_position("center", 1000, 2000, 100, 50, 10, 20) assert (x, y) == (450, 975) def test_center_right(self): x, y = calc_position("center_right", 1000, 2000, 100, 50, 10, 20) assert (x, y) == (890, 975) def test_bottom_left(self): x, y = calc_position("bottom_left", 1000, 2000, 100, 50, 10, 20) assert (x, y) == (10, 1930) def test_bottom_center(self): x, y = calc_position("bottom_center", 1000, 2000, 100, 50, 10, 20) assert (x, y) == (450, 1930) def test_bottom_right(self): x, y = calc_position("bottom_right", 1000, 2000, 100, 50, 10, 20) assert (x, y) == (890, 1930) def test_unknown_position_defaults_bottom_right(self): x, y = calc_position("invalid", 1000, 2000, 100, 50, 10, 20) assert (x, y) == (890, 1930) def test_zero_margin(self): x, y = calc_position("top_left", 1000, 2000, 100, 50, 0, 0) assert (x, y) == (0, 0) def test_small_output(self): x, y = calc_position("center", 100, 100, 50, 30, 5, 5) assert (x, y) == (25, 35) # ── calc_scroll_x 测试 ───────────────────────────────────────────────────── class TestCalcScrollX: def test_returns_string_expression(self): result = calc_scroll_x("bottom", 1000, 200, 50) assert isinstance(result, str) def test_contains_mod_function(self): result = calc_scroll_x("bottom", 1000, 200, 50) assert "mod" in result def test_contains_speed_and_width(self): result = calc_scroll_x("bottom", 1080, 300, 60) assert "1080" in result assert "60" in result assert "300" in result # ── build_image_watermark_filter 测试 ────────────────────────────────────── class TestBuildImageWatermarkFilter: def test_returns_tuple(self): cfg = WatermarkConfig(mode="image", image_path="/wm.png") result = build_image_watermark_filter("[in]", "/wm.png", 1080, 1920, "[out]", cfg) assert isinstance(result, tuple) assert len(result) == 2 def test_filter_contains_overlay(self): cfg = WatermarkConfig(mode="image", image_path="/wm.png") filter_str, inputs = build_image_watermark_filter("[in]", "/wm.png", 1080, 1920, "[out]", cfg) assert "overlay" in filter_str def test_filter_contains_scale(self): cfg = WatermarkConfig(mode="image", image_path="/wm.png", scale=0.5) filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1080, 1920, "[out]", cfg) assert "scale=" in filter_str def test_full_opacity_no_alpha_filter(self): cfg = WatermarkConfig(mode="image", image_path="/wm.png", opacity=1.0) filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1080, 1920, "[out]", cfg) assert "colorchannelmixer" not in filter_str def test_partial_opacity_has_alpha_filter(self): cfg = WatermarkConfig(mode="image", image_path="/wm.png", opacity=0.5) filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1080, 1920, "[out]", cfg) assert "colorchannelmixer" in filter_str assert "aa=0.5" in filter_str def test_input_args_contains_image_path(self): cfg = WatermarkConfig(mode="image", image_path="/path/to/wm.png") _, inputs = build_image_watermark_filter("[in]", "/path/to/wm.png", 1080, 1920, "[out]", cfg) assert inputs == ["-i", "/path/to/wm.png"] def test_scroll_mode_has_t_variable(self): cfg = WatermarkConfig(mode="image", image_path="/wm.png", scroll=True, scroll_speed=50) filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1080, 1920, "[out]", cfg) assert "t" in filter_str def test_output_label_appears(self): cfg = WatermarkConfig(mode="image", image_path="/wm.png") filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1080, 1920, "[final]", cfg) assert "[final]" in filter_str # ── build_text_watermark_filter 测试 ─────────────────────────────────────── class TestBuildTextWatermarkFilter: def test_returns_string(self): cfg = WatermarkConfig(mode="text", text="hello") result = build_text_watermark_filter("[in]", "[out]", cfg, 1080, 1920) assert isinstance(result, str) def test_contains_drawtext(self): cfg = WatermarkConfig(mode="text", text="hello") result = build_text_watermark_filter("[in]", "[out]", cfg, 1080, 1920) assert "drawtext=" in result def test_contains_text_content(self): cfg = WatermarkConfig(mode="text", text="watermark_test") result = build_text_watermark_filter("[in]", "[out]", cfg, 1080, 1920) assert "watermark_test" in result def test_contains_font_size(self): cfg = WatermarkConfig(mode="text", text="hi", font_size=48) result = build_text_watermark_filter("[in]", "[out]", cfg, 1080, 1920) assert "fontsize=48" in result def test_contains_font_color(self): cfg = WatermarkConfig(mode="text", text="hi", font_color="red") result = build_text_watermark_filter("[in]", "[out]", cfg, 1080, 1920) assert "fontcolor=red" in result def test_font_path_included_when_set(self): cfg = WatermarkConfig(mode="text", text="hi", font_path="/fonts/a.ttf") result = build_text_watermark_filter("[in]", "[out]", cfg, 1080, 1920) assert "fontfile=" in result assert "a.ttf" in result def test_font_path_not_included_when_empty(self): cfg = WatermarkConfig(mode="text", text="hi") result = build_text_watermark_filter("[in]", "[out]", cfg, 1080, 1920) assert "fontfile=" not in result def test_scroll_mode_has_t_variable(self): cfg = WatermarkConfig(mode="text", text="hello", scroll=True, scroll_speed=30) result = build_text_watermark_filter("[in]", "[out]", cfg, 1080, 1920) assert "t" in result def test_no_scroll_uses_fixed_position(self): cfg = WatermarkConfig(mode="text", text="hello", scroll=False) result = build_text_watermark_filter("[in]", "[out]", cfg, 1080, 1920) assert "x=" in result # 非滚动模式 x= 后面应该是数字，不是表达式 # 找 x= 后的第一个字符 import re match = re.search(r"x=(\d+)", result) assert match is not None def test_output_label_appears(self): cfg = WatermarkConfig(mode="text", text="hi") result = build_text_watermark_filter("[in]", "[text_out]", cfg, 1080, 1920) assert "[text_out]" in result def test_special_chars_escaped(self): cfg = WatermarkConfig(mode="text", text="hello:world") result = build_text_watermark_filter("[in]", "[out]", cfg, 1080, 1920) # 冒号应该被转义 assert "hello\\:world" in result or "hello\\\\\\:world" in result or "hello\\:" in result # ── 工具函数测试 ──────────────────────────────────────────────────────────── class TestUtils: def test_get_position_names_returns_nine(self): names = get_position_names() assert len(names) == 9 def test_get_position_names_all_valid(self): names = get_position_names() for name in names: assert name in VALID_POSITIONS def test_get_position_display_name_valid(self): assert get_position_display_name("top_left") == "左上" assert get_position_display_name("bottom_right") == "右下" def test_get_position_display_name_invalid(self): assert get_position_display_name("invalid") == "invalid"