free-disk/tests/test_cleanup.py

import collections
import logging
import os
import pathlib
import unittest.mock

import _pytest.logging
import free_disk

# pylint: disable=protected-access

_DiskUsage = collections.namedtuple("_DiskUsage", ("free",))


def _folder_content_size_bytes(path: pathlib.Path) -> int:
    return sum(p.stat().st_size for p in path.rglob("*") if p.is_file())


def test__main_remove_some(
    caplog: _pytest.logging.LogCaptureFixture, tmp_path: pathlib.Path
) -> None:
    tmp_path.joinpath("a").write_bytes(b"a" * 4)
    tmp_path.joinpath("b").write_bytes(b"b" * 3)
    tmp_path.joinpath("c").write_bytes(b"c" * 5)
    tmp_path.joinpath("d").write_bytes(b"d" * 2)
    tmp_path.joinpath("e").write_bytes(b"d" * 7)
    with unittest.mock.patch(
        "shutil.disk_usage",
        lambda p: _DiskUsage(free=42 - _folder_content_size_bytes(tmp_path)),
    ), unittest.mock.patch(
        "sys.argv", ["", "--free-bytes", "30B", str(tmp_path)]
    ), caplog.at_level(
        logging.DEBUG
    ):
        free_disk._main()
    assert {p.name for p in tmp_path.rglob("*")} == {"e", "d"}
    assert caplog.record_tuples[:-1] == [
        ("root", logging.DEBUG, m)
        for m in [
            "Required free bytes: 30",
            "_DiskUsage(free=21)",
            f"Removed file {tmp_path.joinpath('a')}",
            f"Removed file {tmp_path.joinpath('b')}",
            f"Removed file {tmp_path.joinpath('c')}",
        ]
    ]
    assert caplog.records[-1].levelno == logging.INFO
    assert caplog.records[-1].message.startswith(
        "Removed 3 file(s) with modification date <= 20"
    )


def test__main_remove_from_subfolder(
    caplog: _pytest.logging.LogCaptureFixture, tmp_path: pathlib.Path
) -> None:
    tmp_path.joinpath("a").mkdir()
    tmp_path.joinpath("a", "aa").write_bytes(b"a" * 4)
    tmp_path.joinpath("b").write_bytes(b"b" * 3)
    tmp_path.joinpath("c").write_bytes(b"c" * 5)
    with unittest.mock.patch(
        "shutil.disk_usage",
        lambda p: _DiskUsage(free=42 - _folder_content_size_bytes(tmp_path)),
    ), unittest.mock.patch(
        "sys.argv", ["", "--free-bytes", "35B", str(tmp_path)]
    ), caplog.at_level(
        logging.DEBUG
    ):
        free_disk._main()
    assert {p.name for p in tmp_path.rglob("*")} == {"a", "c"}
    assert caplog.record_tuples[:-1] == [
        ("root", logging.DEBUG, m)
        for m in [
            "Required free bytes: 35",
            "_DiskUsage(free=30)",
            f"Removed file {tmp_path.joinpath('a', 'aa')}",
            f"Removed file {tmp_path.joinpath('b')}",
        ]
    ]
    assert caplog.records[-1].levelno == logging.INFO
    assert caplog.records[-1].message.startswith(
        "Removed 2 file(s) with modification date <= 20"
    )


def test__main_sufficient_space(
    caplog: _pytest.logging.LogCaptureFixture, tmp_path: pathlib.Path
) -> None:
    tmp_path.joinpath("a").write_bytes(b"a" * 4)
    tmp_path.joinpath("b").write_bytes(b"b" * 3)
    tmp_path.joinpath("c").write_bytes(b"c" * 5)
    with unittest.mock.patch(
        "shutil.disk_usage",
        lambda p: _DiskUsage(free=42 - _folder_content_size_bytes(tmp_path)),
    ), unittest.mock.patch(
        "sys.argv", ["", "--free-bytes", "30B", str(tmp_path)]
    ), caplog.at_level(
        logging.DEBUG
    ):
        free_disk._main()
    assert {p.name for p in tmp_path.rglob("*")} == {"a", "b", "c"}
    assert caplog.record_tuples == [
        ("root", logging.DEBUG, "Required free bytes: 30"),
        ("root", logging.DEBUG, "_DiskUsage(free=30)"),
        ("root", logging.DEBUG, "Requirement already fulfilled"),
    ]


def test__main_no_files(
    caplog: _pytest.logging.LogCaptureFixture, tmp_path: pathlib.Path
) -> None:
    with unittest.mock.patch(
        "shutil.disk_usage", return_value=_DiskUsage(free=21)
    ), unittest.mock.patch(
        "sys.argv", ["", "--free-bytes", "30B", str(tmp_path)]
    ), caplog.at_level(
        logging.DEBUG
    ):
        free_disk._main()
    assert caplog.record_tuples == [
        ("root", logging.DEBUG, "Required free bytes: 30"),
        ("root", logging.DEBUG, "_DiskUsage(free=21)"),
        ("root", logging.WARNING, "No files to remove"),
    ]


def test__main_path_regex_absolute(
    caplog: _pytest.logging.LogCaptureFixture, tmp_path: pathlib.Path
) -> None:
    tmp_path.joinpath("a").mkdir()
    tmp_path.joinpath("a", "aa").write_bytes(b"aa")
    tmp_path.joinpath("a", "a~").write_bytes(b"a~")
    tmp_path.joinpath("b").write_bytes(b"b")
    tmp_path.joinpath("c").write_bytes(b"c")
    tmp_path.joinpath("d").write_bytes(b"d")
    with unittest.mock.patch(
        "shutil.disk_usage",
        lambda p: _DiskUsage(free=42 - _folder_content_size_bytes(tmp_path)),
    ), unittest.mock.patch(
        "sys.argv",
        [
            "",
            "--free-bytes",
            "42B",
            str(tmp_path),
            "--delete-path-regex",
            r"a/a|^b|c$|^.*/d$",
        ],
    ), caplog.at_level(
        logging.INFO
    ):
        free_disk._main()
    assert {p.name for p in tmp_path.rglob("*")} == {"a", "b"}
    assert caplog.records[-1].message.startswith(
        "Removed 4 file(s) with modification date <= 20"
    )


def test__main_path_regex_relative(
    caplog: _pytest.logging.LogCaptureFixture, tmp_path: pathlib.Path
) -> None:
    tmp_path.joinpath("a").mkdir()
    tmp_path.joinpath("a", "aa").write_bytes(b"aa")
    tmp_path.joinpath("a", "aaa").write_bytes(b"aaa")
    tmp_path.joinpath("a", "A").write_bytes(b"A")
    tmp_path.joinpath("b").write_bytes(b"b")
    tmp_path.joinpath("b2").write_bytes(b"b2")
    tmp_path.joinpath("c").write_bytes(b"c")
    with unittest.mock.patch(
        "shutil.disk_usage",
        lambda p: _DiskUsage(free=42 - _folder_content_size_bytes(tmp_path)),
    ), unittest.mock.patch(
        "sys.argv",
        ["", "--free-bytes", "123B", ".", "--delete-path-regex", r"/aa|^b|\d$|^\./c$"],
    ), caplog.at_level(
        logging.INFO
    ):
        old_working_dir = os.getcwd()
        try:
            os.chdir(tmp_path)
            free_disk._main()
        finally:
            os.chdir(old_working_dir)
    assert {p.name for p in tmp_path.rglob("*")} == {"a", "A", "b"}
    assert caplog.records[-1].message.startswith(
        "Removed 4 file(s) with modification date <= 20"
    )