jekyll-to-hugo/app/tests/utils_test.py

import pytest

from app.utils import (
    guard_against_none,
    guard_against_none_or_empty_str,
    key_error_silence,
)


def test_key_error_silence():
    # Test that the context manager silences the exception
    with key_error_silence():
        raise KeyError
    # Test that the context manager does not silence other exceptions
    with pytest.raises(ValueError):
        with key_error_silence():
            raise ValueError


@pytest.mark.parametrize(
    "input_data",
    [
        (""),
        (None),
        (1),
        (True),
        (False),
        ({}),
        ([]),
        ({"a": 1}),
    ],
)
def test_guard_against_none_or_empty_str(input_data):
    with pytest.raises(ValueError):
        guard_against_none_or_empty_str(input_data, "test")


def test_guard_against_none_or_empty_str_happy():
    guard_against_none_or_empty_str("a", "test")


def test_guard_against_none():
    with pytest.raises(ValueError):
        guard_against_none(None, "test")
    guard_against_none(1, "test")