test: skip watermark integration when darktable export fails
Add darktable style probe and pytest integration/slow markers so CI and machines without a working darktable-cli setup skip the full workflow test. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -0,0 +1,53 @@
|
|||||||
|
"""Shared helpers for integration tests."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import shutil
|
||||||
|
import subprocess
|
||||||
|
import tempfile
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from tests.conftest import make_png
|
||||||
|
|
||||||
|
|
||||||
|
def has_darktable_style(style_name: str, config_dir: Path | None = None) -> bool:
|
||||||
|
"""Return True if darktable-cli can apply ``style_name`` to a tiny PNG."""
|
||||||
|
if not shutil.which("darktable-cli"):
|
||||||
|
return False
|
||||||
|
|
||||||
|
config = config_dir or Path.home() / ".config" / "darktable"
|
||||||
|
styles_dir = config / "styles"
|
||||||
|
if not styles_dir.is_dir():
|
||||||
|
return False
|
||||||
|
if not any(p.stem == style_name for p in styles_dir.glob("*.dtstyle")):
|
||||||
|
return False
|
||||||
|
|
||||||
|
with tempfile.TemporaryDirectory() as tmp:
|
||||||
|
tmp_path = Path(tmp)
|
||||||
|
src = tmp_path / "probe.png"
|
||||||
|
out_dir = tmp_path / "out"
|
||||||
|
out_dir.mkdir()
|
||||||
|
make_png(src)
|
||||||
|
cmd = [
|
||||||
|
"darktable-cli",
|
||||||
|
str(src),
|
||||||
|
str(out_dir),
|
||||||
|
"--style",
|
||||||
|
style_name,
|
||||||
|
"--out-ext",
|
||||||
|
"png",
|
||||||
|
"--core",
|
||||||
|
"--configdir",
|
||||||
|
str(config),
|
||||||
|
"--style-overwrite",
|
||||||
|
]
|
||||||
|
result = subprocess.run(cmd, capture_output=True, text=True)
|
||||||
|
if result.returncode != 0:
|
||||||
|
return False
|
||||||
|
return any(out_dir.iterdir())
|
||||||
|
|
||||||
|
|
||||||
|
def has_workflow_tools() -> bool:
|
||||||
|
"""True when rembg, gmic, ImageMagick, and darktable-cli are on PATH."""
|
||||||
|
has_magick = bool(shutil.which("magick") or shutil.which("convert"))
|
||||||
|
return all(shutil.which(name) for name in ("rembg", "gmic", "darktable-cli")) and has_magick
|
||||||
@@ -5,7 +5,6 @@ from pathlib import Path
|
|||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from imagepipeline.core.params import validate_params
|
|
||||||
from imagepipeline.modules.color_to_alpha import (
|
from imagepipeline.modules.color_to_alpha import (
|
||||||
ColorToAlphaModule,
|
ColorToAlphaModule,
|
||||||
build_color_to_alpha_args,
|
build_color_to_alpha_args,
|
||||||
@@ -13,18 +12,19 @@ from imagepipeline.modules.color_to_alpha import (
|
|||||||
from imagepipeline.modules.composite import CompositeModule
|
from imagepipeline.modules.composite import CompositeModule
|
||||||
from imagepipeline.modules.crop_square import CropSquareModule
|
from imagepipeline.modules.crop_square import CropSquareModule
|
||||||
from imagepipeline.modules.darktable_style import DarktableStyleModule
|
from imagepipeline.modules.darktable_style import DarktableStyleModule
|
||||||
|
from imagepipeline.modules.gmic_grayscale import GmicGrayscale
|
||||||
from imagepipeline.modules.imagemagick_fill import (
|
from imagepipeline.modules.imagemagick_fill import (
|
||||||
ImageMagickFillModule,
|
ImageMagickFillModule,
|
||||||
build_fill_arguments,
|
build_fill_arguments,
|
||||||
)
|
)
|
||||||
from imagepipeline.modules.imagemagick_grayscale import ImageMagickGrayscale
|
|
||||||
from imagepipeline.modules.imagemagick_resize import (
|
from imagepipeline.modules.imagemagick_resize import (
|
||||||
ImageMagickResizeModule,
|
ImageMagickResizeModule,
|
||||||
build_resize_arguments,
|
build_resize_arguments,
|
||||||
)
|
)
|
||||||
from imagepipeline.modules.gmic_grayscale import GmicGrayscale
|
|
||||||
from imagepipeline.modules.registry import get_module, list_modules
|
from imagepipeline.modules.registry import get_module, list_modules
|
||||||
from imagepipeline.modules.rembg import RembgModule
|
from imagepipeline.modules.rembg import RembgModule
|
||||||
|
from tests.integration_helpers import has_darktable_style
|
||||||
|
from tests.integration_helpers import has_workflow_tools as check_workflow_tools
|
||||||
|
|
||||||
has_magick = bool(shutil.which("magick") or shutil.which("convert"))
|
has_magick = bool(shutil.which("magick") or shutil.which("convert"))
|
||||||
|
|
||||||
@@ -184,7 +184,6 @@ class TestImageMagickFill:
|
|||||||
def test_output_matches_input_size(self, tmp_path: Path) -> None:
|
def test_output_matches_input_size(self, tmp_path: Path) -> None:
|
||||||
from imagepipeline.core.context import ModuleContext
|
from imagepipeline.core.context import ModuleContext
|
||||||
from imagepipeline.utils.subprocess import run_command
|
from imagepipeline.utils.subprocess import run_command
|
||||||
|
|
||||||
from tests.conftest import make_png
|
from tests.conftest import make_png
|
||||||
|
|
||||||
src = tmp_path / "ref.png"
|
src = tmp_path / "ref.png"
|
||||||
@@ -221,7 +220,6 @@ class TestCropSquare:
|
|||||||
def test_center_crops_to_square(self, tmp_path: Path) -> None:
|
def test_center_crops_to_square(self, tmp_path: Path) -> None:
|
||||||
from imagepipeline.core.context import ModuleContext
|
from imagepipeline.core.context import ModuleContext
|
||||||
from imagepipeline.utils.subprocess import run_command
|
from imagepipeline.utils.subprocess import run_command
|
||||||
|
|
||||||
from tests.conftest import make_png
|
from tests.conftest import make_png
|
||||||
|
|
||||||
src = tmp_path / "wide.png"
|
src = tmp_path / "wide.png"
|
||||||
@@ -255,9 +253,7 @@ class TestModuleParameters:
|
|||||||
DarktableStyleModule.validate_module_params({})
|
DarktableStyleModule.validate_module_params({})
|
||||||
|
|
||||||
def test_darktable_style_accepts_style(self) -> None:
|
def test_darktable_style_accepts_style(self) -> None:
|
||||||
params = DarktableStyleModule.validate_module_params(
|
params = DarktableStyleModule.validate_module_params({"style": "Watermark F12.rocks"})
|
||||||
{"style": "Watermark F12.rocks"}
|
|
||||||
)
|
|
||||||
assert params["style"] == "Watermark F12.rocks"
|
assert params["style"] == "Watermark F12.rocks"
|
||||||
assert params["style_overwrite"] is True
|
assert params["style_overwrite"] is True
|
||||||
|
|
||||||
@@ -273,9 +269,7 @@ class TestModuleParameters:
|
|||||||
def test_darktable_export_conf_jpeg(self) -> None:
|
def test_darktable_export_conf_jpeg(self) -> None:
|
||||||
from imagepipeline.modules.darktable_style import export_conf_options
|
from imagepipeline.modules.darktable_style import export_conf_options
|
||||||
|
|
||||||
assert export_conf_options("jpeg") == [
|
assert export_conf_options("jpeg") == ["plugins/imageio/format/jpeg/quality=90"]
|
||||||
"plugins/imageio/format/jpeg/quality=90"
|
|
||||||
]
|
|
||||||
|
|
||||||
def test_darktable_export_conf_png(self) -> None:
|
def test_darktable_export_conf_png(self) -> None:
|
||||||
from imagepipeline.modules.darktable_style import export_conf_options
|
from imagepipeline.modules.darktable_style import export_conf_options
|
||||||
@@ -308,7 +302,6 @@ class TestCompositeColor:
|
|||||||
def test_preserves_color_over_grayscale_background(self, tmp_path: Path) -> None:
|
def test_preserves_color_over_grayscale_background(self, tmp_path: Path) -> None:
|
||||||
from imagepipeline.core.context import ModuleContext
|
from imagepipeline.core.context import ModuleContext
|
||||||
from imagepipeline.utils.subprocess import run_command
|
from imagepipeline.utils.subprocess import run_command
|
||||||
|
|
||||||
from tests.conftest import make_png
|
from tests.conftest import make_png
|
||||||
|
|
||||||
src = tmp_path / "src.png"
|
src = tmp_path / "src.png"
|
||||||
@@ -347,19 +340,20 @@ class TestCompositeColor:
|
|||||||
|
|
||||||
output = output_dir / "fg.png"
|
output = output_dir / "fg.png"
|
||||||
assert output.is_file()
|
assert output.is_file()
|
||||||
result = run_command(
|
result = run_command([magick, "identify", "-format", "%[type]", str(output)])
|
||||||
[magick, "identify", "-format", "%[type]", str(output)]
|
|
||||||
)
|
|
||||||
assert result.stdout.strip() != "Grayscale"
|
assert result.stdout.strip() != "Grayscale"
|
||||||
|
|
||||||
|
|
||||||
has_workflow_tools = all(
|
has_workflow_tools = check_workflow_tools()
|
||||||
shutil.which(name)
|
|
||||||
for name in ("rembg", "gmic", "magick", "darktable-cli")
|
|
||||||
) or all(shutil.which(name) for name in ("rembg", "gmic", "convert", "darktable-cli"))
|
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.integration
|
||||||
|
@pytest.mark.slow
|
||||||
@pytest.mark.skipif(not has_workflow_tools, reason="Workflow CLI tools not installed")
|
@pytest.mark.skipif(not has_workflow_tools, reason="Workflow CLI tools not installed")
|
||||||
|
@pytest.mark.skipif(
|
||||||
|
not has_darktable_style("Watermark F12.rocks"),
|
||||||
|
reason="darktable style 'Watermark F12.rocks' not available or darktable-cli export failed",
|
||||||
|
)
|
||||||
class TestWorkflowIntegration:
|
class TestWorkflowIntegration:
|
||||||
def test_watermark_pipeline(self, input_dir, output_base) -> None:
|
def test_watermark_pipeline(self, input_dir, output_base) -> None:
|
||||||
from imagepipeline import Pipeline
|
from imagepipeline import Pipeline
|
||||||
|
|||||||
Reference in New Issue
Block a user