2026-07-28 16:18:39 +08:00
1 changed files with 602 additions and 0 deletions
@@ -0,0 +1,602 @@
+"""watermark_config 单元测试 - wave163
+
+覆盖：
+- WatermarkConfig 数据类 / from_dict / validate / has_effect
+- calc_position 9宫格位置计算
+- calc_scroll_x 滚动计算
+- build_image_watermark_filter 图片水印滤镜
+- build_text_watermark_filter 文字水印滤镜
+- get_position_names / get_position_display_name
+- 常量验证
+"""
+
+import pytest
+
+from packages.domain.watermark_config import (
+    DEFAULT_FONT_COLOR,
+    DEFAULT_FONT_SIZE,
+    DEFAULT_MARGIN_X,
+    DEFAULT_MARGIN_Y,
+    DEFAULT_MODE,
+    DEFAULT_OPACITY,
+    DEFAULT_POSITION,
+    DEFAULT_SCALE,
+    DEFAULT_SCROLL_SPEED,
+    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(VALID_POSITIONS) == 9
+
+    def test_watermark_positions_contains_all(self):
+        assert set(WATERMARK_POSITIONS.keys()) == VALID_POSITIONS
+
+    def test_default_position_valid(self):
+        assert DEFAULT_POSITION in VALID_POSITIONS
+
+    def test_default_values(self):
+        assert DEFAULT_MODE in ("text", "image")
+        assert 0 < DEFAULT_SCALE <= 1.0
+        assert 0 <= DEFAULT_OPACITY <= 1.0
+        assert DEFAULT_FONT_SIZE > 0
+        assert DEFAULT_MARGIN_X >= 0
+        assert DEFAULT_MARGIN_Y >= 0
+        assert DEFAULT_SCROLL_SPEED > 0
+        assert isinstance(DEFAULT_FONT_COLOR, str)
+
+
+# ============================================================
+# WatermarkConfig 默认值
+# ============================================================
+
+
+class TestWatermarkConfigDefaults:
+    def test_default_constructor(self):
+        config = WatermarkConfig()
+        assert config.mode == DEFAULT_MODE
+        assert config.position == DEFAULT_POSITION
+        assert config.scale == DEFAULT_SCALE
+        assert config.opacity == DEFAULT_OPACITY
+        assert config.font_size == DEFAULT_FONT_SIZE
+        assert config.font_color == DEFAULT_FONT_COLOR
+        assert config.margin_x == DEFAULT_MARGIN_X
+        assert config.margin_y == DEFAULT_MARGIN_Y
+        assert config.scroll is False
+        assert config.scroll_speed == DEFAULT_SCROLL_SPEED
+
+    def test_text_mode_default(self):
+        config = WatermarkConfig()
+        assert config.mode == "text"
+        assert config.text == ""
+        assert config.font_path == ""
+
+    def test_image_mode_default(self):
+        config = WatermarkConfig(mode="image")
+        assert config.image_path == ""
+
+
+# ============================================================
+# from_dict
+# ============================================================
+
+
+class TestFromDict:
+    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_image_mode_missing_path_returns_none(self):
+        result = WatermarkConfig.from_dict({"enabled": True, "mode": "image"})
+        assert result is None
+
+    def test_image_mode_with_image_field(self):
+        result = WatermarkConfig.from_dict({"enabled": True, "mode": "image", "image": "/path/to/wm.png"})
+        assert result is not None
+        assert result.mode == "image"
+        assert result.image_path == "/path/to/wm.png"
+
+    def test_image_mode_with_image_path_field(self):
+        result = WatermarkConfig.from_dict({"enabled": True, "mode": "image", "image_path": "/path/wm.png"})
+        assert result is not None
+        assert result.image_path == "/path/wm.png"
+
+    def test_text_mode_missing_text_returns_none(self):
+        result = WatermarkConfig.from_dict({"enabled": True, "mode": "text"})
+        assert result is None
+
+    def test_text_mode_with_text(self):
+        result = WatermarkConfig.from_dict({"enabled": True, "mode": "text", "text": "hello"})
+        assert result is not None
+        assert result.mode == "text"
+        assert result.text == "hello"
+
+    def test_invalid_position_falls_back_default(self):
+        result = WatermarkConfig.from_dict(
+            {
+                "enabled": True,
+                "mode": "text",
+                "text": "hello",
+                "position": "invalid_pos",
+            }
+        )
+        assert result is not None
+        assert result.position == DEFAULT_POSITION
+
+    def test_custom_values_propagated(self):
+        result = WatermarkConfig.from_dict(
+            {
+                "enabled": True,
+                "mode": "text",
+                "text": "测试",
+                "position": "top_left",
+                "opacity": 0.5,
+                "font_size": 32,
+                "font_color": "red",
+                "margin_x": 30,
+                "margin_y": 40,
+                "scroll": True,
+                "scroll_speed": 100,
+            }
+        )
+        assert result is not None
+        assert result.position == "top_left"
+        assert result.opacity == 0.5
+        assert result.font_size == 32
+        assert result.font_color == "red"
+        assert result.margin_x == 30
+        assert result.margin_y == 40
+        assert result.scroll is True
+        assert result.scroll_speed == 100
+
+    def test_image_mode_custom_values(self):
+        result = WatermarkConfig.from_dict(
+            {
+                "enabled": True,
+                "mode": "image",
+                "image_path": "/wm.png",
+                "scale": 0.3,
+                "opacity": 0.6,
+            }
+        )
+        assert result is not None
+        assert result.scale == 0.3
+        assert result.opacity == 0.6
+
+    def test_default_mode_when_unspecified(self):
+        # 只给enabled和text，mode默认text
+        result = WatermarkConfig.from_dict({"enabled": True, "text": "hello"})
+        assert result is not None
+        assert result.mode == DEFAULT_MODE
+
+    def test_text_empty_string_returns_none(self):
+        result = WatermarkConfig.from_dict({"enabled": True, "mode": "text", "text": ""})
+        assert result is None
+
+
+# ============================================================
+# validate
+# ============================================================
+
+
+class TestValidate:
+    def test_valid_text_config(self):
+        config = WatermarkConfig(mode="text", text="hello")
+        ok, msg = config.validate()
+        assert ok is True
+        assert msg == ""
+
+    def test_valid_image_config(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png")
+        ok, msg = config.validate()
+        assert ok is True
+        assert msg == ""
+
+    def test_invalid_position(self):
+        config = WatermarkConfig(mode="text", text="hi", position="nowhere")
+        ok, msg = config.validate()
+        assert ok is False
+        assert "位置" in msg
+
+    def test_opacity_too_high(self):
+        config = WatermarkConfig(mode="text", text="hi", opacity=1.5)
+        ok, msg = config.validate()
+        assert ok is False
+        assert "透明度" in msg
+
+    def test_opacity_negative(self):
+        config = WatermarkConfig(mode="text", text="hi", opacity=-0.1)
+        ok, msg = config.validate()
+        assert ok is False
+        assert "透明度" in msg
+
+    def test_opacity_boundary_zero(self):
+        config = WatermarkConfig(mode="text", text="hi", opacity=0.0)
+        ok, _ = config.validate()
+        assert ok is True
+
+    def test_opacity_boundary_one(self):
+        config = WatermarkConfig(mode="text", text="hi", opacity=1.0)
+        ok, _ = config.validate()
+        assert ok is True
+
+    def test_image_missing_path(self):
+        config = WatermarkConfig(mode="image")
+        ok, msg = config.validate()
+        assert ok is False
+        assert "图片路径" in msg
+
+    def test_image_scale_too_small(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", scale=0.001)
+        ok, msg = config.validate()
+        assert ok is False
+        assert "缩放" in msg
+
+    def test_image_scale_too_large(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", scale=1.5)
+        ok, msg = config.validate()
+        assert ok is False
+        assert "缩放" in msg
+
+    def test_image_scale_boundary_low(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", scale=0.01)
+        ok, _ = config.validate()
+        assert ok is True
+
+    def test_image_scale_boundary_high(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", scale=1.0)
+        ok, _ = config.validate()
+        assert ok is True
+
+    def test_text_empty(self):
+        config = WatermarkConfig(mode="text", text="")
+        ok, msg = config.validate()
+        assert ok is False
+        assert "文字内容" in msg
+
+    def test_font_size_zero(self):
+        config = WatermarkConfig(mode="text", text="hi", font_size=0)
+        ok, msg = config.validate()
+        assert ok is False
+        assert "字体大小" in msg
+
+    def test_font_size_negative(self):
+        config = WatermarkConfig(mode="text", text="hi", font_size=-5)
+        ok, msg = config.validate()
+        assert ok is False
+        assert "字体大小" in msg
+
+    def test_unknown_mode(self):
+        config = WatermarkConfig(mode="video", text="hi")
+        ok, msg = config.validate()
+        assert ok is False
+        assert "模式" in msg
+
+
+# ============================================================
+# has_effect
+# ============================================================
+
+
+class TestHasEffect:
+    def test_text_has_effect(self):
+        config = WatermarkConfig(mode="text", text="hello", opacity=0.5, font_size=24)
+        assert config.has_effect() is True
+
+    def test_text_empty_no_effect(self):
+        config = WatermarkConfig(mode="text", text="")
+        assert config.has_effect() is False
+
+    def test_text_zero_opacity_no_effect(self):
+        config = WatermarkConfig(mode="text", text="hi", opacity=0.0)
+        assert config.has_effect() is False
+
+    def test_text_zero_font_size_no_effect(self):
+        config = WatermarkConfig(mode="text", text="hi", font_size=0)
+        assert config.has_effect() is False
+
+    def test_image_has_effect(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", opacity=0.5)
+        assert config.has_effect() is True
+
+    def test_image_no_path_no_effect(self):
+        config = WatermarkConfig(mode="image")
+        assert config.has_effect() is False
+
+    def test_image_zero_opacity_no_effect(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", opacity=0.0)
+        assert config.has_effect() is False
+
+    def test_unknown_mode_no_effect(self):
+        config = WatermarkConfig(mode="invalid")
+        assert config.has_effect() is False
+
+
+# ============================================================
+# calc_position
+# ============================================================
+
+
+class TestCalcPosition:
+    OUT_W = 1920
+    OUT_H = 1080
+    WM_W = 200
+    WM_H = 50
+    MX = 20
+    MY = 20
+
+    def test_top_left(self):
+        x, y = calc_position("top_left", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+        assert x == 20
+        assert y == 20
+
+    def test_top_center(self):
+        x, y = calc_position("top_center", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+        assert x == (1920 - 200) // 2
+        assert y == 20
+
+    def test_top_right(self):
+        x, y = calc_position("top_right", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+        assert x == 1920 - 200 - 20
+        assert y == 20
+
+    def test_center_left(self):
+        x, y = calc_position("center_left", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+        assert x == 20
+        assert y == (1080 - 50) // 2
+
+    def test_center(self):
+        x, y = calc_position("center", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+        assert x == (1920 - 200) // 2
+        assert y == (1080 - 50) // 2
+
+    def test_center_right(self):
+        x, y = calc_position("center_right", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+        assert x == 1920 - 200 - 20
+        assert y == (1080 - 50) // 2
+
+    def test_bottom_left(self):
+        x, y = calc_position("bottom_left", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+        assert x == 20
+        assert y == 1080 - 50 - 20
+
+    def test_bottom_center(self):
+        x, y = calc_position("bottom_center", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+        assert x == (1920 - 200) // 2
+        assert y == 1080 - 50 - 20
+
+    def test_bottom_right(self):
+        x, y = calc_position("bottom_right", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+        assert x == 1920 - 200 - 20
+        assert y == 1080 - 50 - 20
+
+    def test_invalid_position_defaults_bottom_right(self):
+        x, y = calc_position("invalid", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+        assert x == 1920 - 200 - 20
+        assert y == 1080 - 50 - 20
+
+    def test_zero_margins(self):
+        x, y = calc_position("top_left", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, 0, 0)
+        assert x == 0
+        assert y == 0
+
+    def test_all_nine_positions_are_unique(self):
+        positions = set()
+        for pos in VALID_POSITIONS:
+            pos_xy = calc_position(pos, self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
+            positions.add(pos_xy)
+        assert len(positions) == 9
+
+
+# ============================================================
+# calc_scroll_x
+# ============================================================
+
+
+class TestCalcScrollX:
+    def test_returns_string(self):
+        result = calc_scroll_x("bottom_left", 1920, 200, 50)
+        assert isinstance(result, str)
+
+    def test_contains_output_width(self):
+        result = calc_scroll_x("bottom_left", 1920, 200, 50)
+        assert "1920" in result
+
+    def test_contains_wm_width(self):
+        result = calc_scroll_x("bottom_left", 1920, 200, 50)
+        assert "200" in result
+
+    def test_contains_speed(self):
+        result = calc_scroll_x("bottom_left", 1920, 200, 50)
+        assert "50" in result
+
+    def test_contains_mod_keyword(self):
+        result = calc_scroll_x("bottom_left", 1920, 200, 50)
+        assert "mod" in result
+
+
+# ============================================================
+# build_image_watermark_filter
+# ============================================================
+
+
+class TestBuildImageWatermarkFilter:
+    def test_returns_tuple(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", position="top_left")
+        result = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
+        assert isinstance(result, tuple)
+        assert len(result) == 2
+
+    def test_filter_contains_overlay(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", position="bottom_right")
+        filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
+        assert "overlay" in filter_str
+
+    def test_filter_contains_scale(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", scale=0.3)
+        filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
+        assert "scale=" in filter_str
+
+    def test_opacity_applied_when_below_one(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", opacity=0.5)
+        filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
+        assert "colorchannelmixer" in filter_str
+        assert "aa=0.5" in filter_str
+
+    def test_opacity_one_no_mixer(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", opacity=1.0)
+        filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
+        assert "colorchannelmixer" not in filter_str
+
+    def test_input_args_contain_image_path(self):
+        config = WatermarkConfig(mode="image", image_path="/path/to/wm.png")
+        _, input_args = build_image_watermark_filter("[in]", "/path/to/wm.png", 1920, 1080, "[out]", config)
+        assert input_args == ["-i", "/path/to/wm.png"]
+
+    def test_output_label_present(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png")
+        filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[final]", config)
+        assert "[final]" in filter_str
+
+    def test_scroll_mode_contains_t_variable(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", scroll=True, scroll_speed=60)
+        filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
+        assert "*t" in filter_str or "t*" in filter_str
+
+    def test_static_mode_no_t_variable_in_overlay_x(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png", scroll=False)
+        filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
+        # 静态模式 x 是具体数值，不包含 t 变量
+        overlay_part = filter_str.split("overlay=")[1]
+        assert "t" not in overlay_part.split(":")[0] or "wm" in overlay_part.split(":")[0]
+
+    def test_semicolon_separated_filters(self):
+        config = WatermarkConfig(mode="image", image_path="/wm.png")
+        filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
+        assert ";" in filter_str
+
+
+# ============================================================
+# build_text_watermark_filter
+# ============================================================
+
+
+class TestBuildTextWatermarkFilter:
+    def test_returns_string(self):
+        config = WatermarkConfig(mode="text", text="hello")
+        result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
+        assert isinstance(result, str)
+
+    def test_contains_drawtext(self):
+        config = WatermarkConfig(mode="text", text="hello")
+        result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
+        assert "drawtext=" in result
+
+    def test_contains_text(self):
+        config = WatermarkConfig(mode="text", text="测试水印")
+        result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
+        assert "测试水印" in result
+
+    def test_fontsize_in_filter(self):
+        config = WatermarkConfig(mode="text", text="hi", font_size=36)
+        result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
+        assert "fontsize=36" in result
+
+    def test_fontcolor_with_opacity(self):
+        config = WatermarkConfig(mode="text", text="hi", font_color="red", opacity=0.5)
+        result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
+        assert "fontcolor=red@0.5" in result
+
+    def test_font_path_included(self):
+        config = WatermarkConfig(mode="text", text="hi", font_path="/fonts/simsun.ttf")
+        result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
+        assert "fontfile=" in result
+        assert "simsun.ttf" in result
+
+    def test_no_font_path_when_empty(self):
+        config = WatermarkConfig(mode="text", text="hi", font_path="")
+        result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
+        assert "fontfile=" not in result
+
+    def test_text_colon_escaped(self):
+        config = WatermarkConfig(mode="text", text="time: 00:00")
+        result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
+        # 冒号应该被转义
+        assert "time\\:" in result or "time\\\\:" in result
+
+    def test_scroll_mode_contains_mod(self):
+        config = WatermarkConfig(mode="text", text="scrolling", scroll=True, scroll_speed=50)
+        result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
+        assert "mod(" in result
+
+    def test_static_mode_has_numeric_x_y(self):
+        config = WatermarkConfig(mode="text", text="hi", position="top_left")
+        result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
+        assert "x=" in result
+        assert "y=" in result
+
+    def test_output_label_present(self):
+        config = WatermarkConfig(mode="text", text="hi")
+        result = build_text_watermark_filter("[in]", "[final_out]", config, 1920, 1080)
+        assert "[final_out]" in result
+
+    def test_input_label_present(self):
+        config = WatermarkConfig(mode="text", text="hi")
+        result = build_text_watermark_filter("[video_in]", "[out]", config, 1920, 1080)
+        assert result.startswith("[video_in]")
+
+
+# ============================================================
+# 工具函数
+# ============================================================
+
+
+class TestGetPositionNames:
+    def test_returns_nine_names(self):
+        names = get_position_names()
+        assert len(names) == 9
+
+    def test_order_is_correct(self):
+        names = get_position_names()
+        # 按从上到下、从左到右
+        assert names[0] == "top_left"
+        assert names[1] == "top_center"
+        assert names[2] == "top_right"
+        assert names[3] == "center_left"
+        assert names[4] == "center"
+        assert names[5] == "center_right"
+        assert names[6] == "bottom_left"
+        assert names[7] == "bottom_center"
+        assert names[8] == "bottom_right"
+
+    def test_all_valid(self):
+        names = get_position_names()
+        assert set(names) == VALID_POSITIONS
+
+
+class TestGetPositionDisplayName:
+    def test_known_position(self):
+        assert get_position_display_name("top_left") == "左上"
+        assert get_position_display_name("center") == "中心"
+        assert get_position_display_name("bottom_right") == "右下"
+
+    def test_unknown_position_returns_original(self):
+        assert get_position_display_name("invalid") == "invalid"