mirror of
https://github.com/FuzzingLabs/fuzzforge_ai.git
synced 2026-08-11 11:30:18 +02:00
rename: FuzzForge → SecPipe
Rename the entire project from FuzzForge to SecPipe: - Python packages: fuzzforge_cli → secpipe_cli, fuzzforge_common → secpipe_common, fuzzforge_mcp → secpipe_mcp, fuzzforge_tests → secpipe_tests - Directories: fuzzforge-cli → secpipe-cli, fuzzforge-common → secpipe-common, fuzzforge-mcp → secpipe-mcp, fuzzforge-tests → secpipe-tests - Environment variables: FUZZFORGE_* → SECPIPE_* - MCP server name: SecPipe MCP Server - CI workflows, Makefile, Dockerfile, hub-config, NOTICE updated - Fix mcp-server.yml to use uvicorn secpipe_mcp.application:app
This commit is contained in:
@@ -0,0 +1 @@
|
||||
pytest_plugins = ["secpipe_tests.fixtures"]
|
||||
@@ -0,0 +1,3 @@
|
||||
"""Conftest for engine tests."""
|
||||
|
||||
# No special fixtures needed - PodmanCLI tests use tmp_path directly
|
||||
@@ -0,0 +1,81 @@
|
||||
"""Tests for the DockerCLI engine."""
|
||||
|
||||
from unittest import mock
|
||||
|
||||
import pytest
|
||||
|
||||
from secpipe_common.sandboxes.engines.docker.cli import DockerCLI
|
||||
|
||||
|
||||
def test_docker_cli_base_cmd() -> None:
|
||||
"""Test that base command is just 'docker'."""
|
||||
engine = DockerCLI()
|
||||
base_cmd = engine._base_cmd()
|
||||
|
||||
assert base_cmd == ["docker"]
|
||||
|
||||
|
||||
def test_docker_cli_list_images_returns_list() -> None:
|
||||
"""Test that list_images returns a list (mocked)."""
|
||||
engine = DockerCLI()
|
||||
|
||||
# Mock the _run method to return empty JSON
|
||||
with mock.patch.object(engine, "_run") as mock_run:
|
||||
mock_run.return_value = mock.Mock(stdout="", returncode=0)
|
||||
images = engine.list_images()
|
||||
|
||||
assert isinstance(images, list)
|
||||
assert len(images) == 0
|
||||
|
||||
|
||||
def test_docker_cli_list_images_parses_output() -> None:
|
||||
"""Test that list_images correctly parses Docker JSON output."""
|
||||
engine = DockerCLI()
|
||||
|
||||
# Docker outputs one JSON object per line
|
||||
docker_output = '{"Repository":"alpine","Tag":"latest","ID":"abc123","Size":"5MB"}\n{"Repository":"ubuntu","Tag":"22.04","ID":"def456","Size":"77MB"}'
|
||||
|
||||
with mock.patch.object(engine, "_run") as mock_run:
|
||||
mock_run.return_value = mock.Mock(stdout=docker_output, returncode=0)
|
||||
images = engine.list_images()
|
||||
|
||||
assert len(images) == 2
|
||||
assert images[0].repository == "alpine"
|
||||
assert images[0].tag == "latest"
|
||||
assert images[1].repository == "ubuntu"
|
||||
assert images[1].tag == "22.04"
|
||||
|
||||
|
||||
def test_docker_cli_image_exists_mocked() -> None:
|
||||
"""Test image_exists with mocked response."""
|
||||
engine = DockerCLI()
|
||||
|
||||
with mock.patch.object(engine, "_run") as mock_run:
|
||||
# Image exists
|
||||
mock_run.return_value = mock.Mock(returncode=0)
|
||||
assert engine.image_exists("alpine:latest") is True
|
||||
|
||||
# Image doesn't exist
|
||||
mock_run.return_value = mock.Mock(returncode=1)
|
||||
assert engine.image_exists("nonexistent:image") is False
|
||||
|
||||
|
||||
def test_docker_cli_create_container_with_volumes() -> None:
|
||||
"""Test create_container generates correct command with volumes."""
|
||||
engine = DockerCLI()
|
||||
|
||||
with mock.patch.object(engine, "_run") as mock_run:
|
||||
mock_run.return_value = mock.Mock(stdout="container123\n", returncode=0)
|
||||
|
||||
container_id = engine.create_container(
|
||||
"alpine:latest",
|
||||
volumes={"/host/path": "/container/path"}
|
||||
)
|
||||
|
||||
# Check the command was called with volume flag
|
||||
call_args = mock_run.call_args[0][0]
|
||||
assert "create" in call_args
|
||||
assert "-v" in call_args
|
||||
assert "/host/path:/container/path:ro" in call_args
|
||||
assert "alpine:latest" in call_args
|
||||
assert container_id == "container123"
|
||||
@@ -0,0 +1,149 @@
|
||||
"""Tests for the PodmanCLI engine (OSS container engine)."""
|
||||
|
||||
import os
|
||||
import shutil
|
||||
import sys
|
||||
import uuid
|
||||
from pathlib import Path
|
||||
from unittest import mock
|
||||
|
||||
import pytest
|
||||
|
||||
from secpipe_common.exceptions import SecPipeError
|
||||
from secpipe_common.sandboxes.engines.podman.cli import PodmanCLI, _is_running_under_snap
|
||||
|
||||
|
||||
# Helper to mock Linux platform for testing (since Podman is Linux-only)
|
||||
def _mock_linux_platform() -> mock._patch[str]:
|
||||
"""Context manager to mock sys.platform as 'linux'."""
|
||||
return mock.patch.object(sys, "platform", "linux")
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def podman_cli_engine() -> PodmanCLI:
|
||||
"""Create a PodmanCLI engine with temporary storage.
|
||||
|
||||
Uses short paths in /tmp to avoid podman's 50-char runroot limit.
|
||||
Simulates Snap environment to test custom storage paths.
|
||||
Mocks Linux platform since Podman is Linux-only.
|
||||
"""
|
||||
short_id = str(uuid.uuid4())[:8]
|
||||
graphroot = Path(f"/tmp/ff-{short_id}/storage")
|
||||
runroot = Path(f"/tmp/ff-{short_id}/run")
|
||||
|
||||
# Simulate Snap environment for testing on Linux
|
||||
with _mock_linux_platform(), mock.patch.dict(os.environ, {"SNAP": "/snap/code/123"}):
|
||||
engine = PodmanCLI(graphroot=graphroot, runroot=runroot)
|
||||
|
||||
yield engine
|
||||
|
||||
# Cleanup
|
||||
parent = graphroot.parent
|
||||
if parent.exists():
|
||||
shutil.rmtree(parent, ignore_errors=True)
|
||||
|
||||
|
||||
def test_snap_detection_when_snap_set() -> None:
|
||||
"""Test that SNAP environment is detected."""
|
||||
with mock.patch.dict(os.environ, {"SNAP": "/snap/code/123"}):
|
||||
assert _is_running_under_snap() is True
|
||||
|
||||
|
||||
def test_snap_detection_when_snap_not_set() -> None:
|
||||
"""Test that non-Snap environment is detected."""
|
||||
env = os.environ.copy()
|
||||
env.pop("SNAP", None)
|
||||
with mock.patch.dict(os.environ, env, clear=True):
|
||||
assert _is_running_under_snap() is False
|
||||
|
||||
|
||||
def test_podman_cli_blocks_macos() -> None:
|
||||
"""Test that PodmanCLI raises error on macOS."""
|
||||
with mock.patch.object(sys, "platform", "darwin"):
|
||||
with pytest.raises(SecPipeError) as exc_info:
|
||||
PodmanCLI()
|
||||
assert "Podman is not supported on macOS" in str(exc_info.value)
|
||||
assert "Docker" in str(exc_info.value)
|
||||
|
||||
|
||||
def test_podman_cli_creates_storage_directories_under_snap() -> None:
|
||||
"""Test that PodmanCLI creates storage directories when under Snap."""
|
||||
short_id = str(uuid.uuid4())[:8]
|
||||
graphroot = Path(f"/tmp/ff-{short_id}/storage")
|
||||
runroot = Path(f"/tmp/ff-{short_id}/run")
|
||||
|
||||
assert not graphroot.exists()
|
||||
assert not runroot.exists()
|
||||
|
||||
with _mock_linux_platform(), mock.patch.dict(os.environ, {"SNAP": "/snap/code/123"}):
|
||||
PodmanCLI(graphroot=graphroot, runroot=runroot)
|
||||
|
||||
assert graphroot.exists()
|
||||
assert runroot.exists()
|
||||
|
||||
# Cleanup
|
||||
shutil.rmtree(graphroot.parent, ignore_errors=True)
|
||||
|
||||
|
||||
def test_podman_cli_base_cmd_under_snap() -> None:
|
||||
"""Test that base command includes --root/--runroot under Snap."""
|
||||
short_id = str(uuid.uuid4())[:8]
|
||||
graphroot = Path(f"/tmp/ff-{short_id}/storage")
|
||||
runroot = Path(f"/tmp/ff-{short_id}/run")
|
||||
|
||||
with _mock_linux_platform(), mock.patch.dict(os.environ, {"SNAP": "/snap/code/123"}):
|
||||
engine = PodmanCLI(graphroot=graphroot, runroot=runroot)
|
||||
base_cmd = engine._base_cmd()
|
||||
|
||||
assert "podman" in base_cmd
|
||||
assert "--root" in base_cmd
|
||||
assert "--runroot" in base_cmd
|
||||
|
||||
# Cleanup
|
||||
shutil.rmtree(graphroot.parent, ignore_errors=True)
|
||||
|
||||
|
||||
def test_podman_cli_base_cmd_without_snap() -> None:
|
||||
"""Test that base command is plain 'podman' when not under Snap."""
|
||||
short_id = str(uuid.uuid4())[:8]
|
||||
graphroot = Path(f"/tmp/ff-{short_id}/storage")
|
||||
runroot = Path(f"/tmp/ff-{short_id}/run")
|
||||
|
||||
env = os.environ.copy()
|
||||
env.pop("SNAP", None)
|
||||
with _mock_linux_platform(), mock.patch.dict(os.environ, env, clear=True):
|
||||
engine = PodmanCLI(graphroot=graphroot, runroot=runroot)
|
||||
base_cmd = engine._base_cmd()
|
||||
|
||||
assert base_cmd == ["podman"]
|
||||
assert "--root" not in base_cmd
|
||||
|
||||
# Directories should NOT be created when not under Snap
|
||||
assert not graphroot.exists()
|
||||
|
||||
|
||||
def test_podman_cli_default_mode() -> None:
|
||||
"""Test PodmanCLI without custom storage paths."""
|
||||
with _mock_linux_platform():
|
||||
engine = PodmanCLI() # No paths provided
|
||||
base_cmd = engine._base_cmd()
|
||||
|
||||
assert base_cmd == ["podman"]
|
||||
assert "--root" not in base_cmd
|
||||
|
||||
|
||||
def test_podman_cli_list_images_returns_list(podman_cli_engine: PodmanCLI) -> None:
|
||||
"""Test that list_images returns a list (even if empty)."""
|
||||
images = podman_cli_engine.list_images()
|
||||
|
||||
assert isinstance(images, list)
|
||||
|
||||
|
||||
@pytest.mark.skip(reason="Requires pulling images, slow integration test")
|
||||
def test_podman_cli_can_pull_and_list_image(podman_cli_engine: PodmanCLI) -> None:
|
||||
"""Test pulling an image and listing it."""
|
||||
# Pull a small image
|
||||
podman_cli_engine._run(["pull", "docker.io/library/alpine:latest"])
|
||||
|
||||
images = podman_cli_engine.list_images()
|
||||
assert any("alpine" in img.identifier for img in images)
|
||||
Reference in New Issue
Block a user