"""Unit tests for the docker plugin's shared query helpers (no DB)."""
from plugins.docker import service as svc


def test_escape_like_makes_wildcards_literal():
    # A log search for "100%" or "user_id" must match those characters, not
    # treat them as LIKE wildcards.
    assert svc.escape_like("100%") == "100\\%"
    assert svc.escape_like("user_id") == "user\\_id"
    assert svc.escape_like("C:\\temp") == "C:\\\\temp"
    assert svc.escape_like("plain text") == "plain text"


def test_bucket_keeps_short_series_and_averages_long_ones():
    assert svc.bucket([1.0, 2.0, 3.0]) == [1.0, 2.0, 3.0]
    out = svc.bucket([float(i) for i in range(400)], target=40)
    assert len(out) == 40
    assert out[0] == sum(range(10)) / 10


def test_prune_extra_vars_mapping():
    """The prune buttons drive one playbook via prune_target; only 'images'
    widens to ALL unused images (docker image prune -a), system stays -f."""
    assert svc.prune_extra_vars("containers") == {"prune_target": "containers"}
    assert svc.prune_extra_vars("images") == {
        "prune_target": "images", "prune_all_images": True,
    }
    # System prune stays conservative — no prune_all_images key.
    assert svc.prune_extra_vars("system") == {"prune_target": "system"}