45 lines
1,018 B
Python
45 lines
1,018 B
Python
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")
|