mirror of
https://github.com/arcan1s/ahriman.git
synced 2025-04-24 07:17:17 +00:00
Evgeniy Alekseev
11732a8609
In case of running command from web interface, it will raise exception because context has been copied with subprocesses
104 lines
2.2 KiB
Python
104 lines
2.2 KiB
Python
import pytest
|
|
|
|
from ahriman.core import _Context
|
|
from ahriman.models.context_key import ContextKey
|
|
|
|
|
|
def test_get_set() -> None:
|
|
"""
|
|
must set and get variable
|
|
"""
|
|
key, value = ContextKey("key", int), 42
|
|
ctx = _Context()
|
|
|
|
ctx.set(key, value)
|
|
assert ctx.get(key) == value
|
|
|
|
|
|
def test_get_key_exception() -> None:
|
|
"""
|
|
must raise KeyError in case if key was not found
|
|
"""
|
|
ctx = _Context()
|
|
with pytest.raises(KeyError):
|
|
ctx.get(ContextKey("key", int))
|
|
|
|
|
|
def test_get_value_exception() -> None:
|
|
"""
|
|
must raise ValueError in case if key type differs from existing value
|
|
"""
|
|
key, value = ContextKey("key", int), 42
|
|
ctx = _Context()
|
|
ctx.set(key, value)
|
|
|
|
with pytest.raises(ValueError):
|
|
ctx.get(ContextKey("key", str))
|
|
|
|
|
|
def test_set_key_exception() -> None:
|
|
"""
|
|
must raise KeyError in case if key already exists
|
|
"""
|
|
key, value = ContextKey("key", int), 42
|
|
ctx = _Context()
|
|
ctx.set(key, value)
|
|
|
|
with pytest.raises(KeyError):
|
|
ctx.set(key, value)
|
|
|
|
|
|
def test_set_value_exception() -> None:
|
|
"""
|
|
must raise ValueError in case if key type differs from new value
|
|
"""
|
|
ctx = _Context()
|
|
with pytest.raises(ValueError):
|
|
ctx.set(ContextKey("key", str), 42)
|
|
|
|
|
|
def test_set_value_exists() -> None:
|
|
"""
|
|
must skip key set in case if key already exists and strict check is disabled
|
|
"""
|
|
key, value = ContextKey("key", int), 42
|
|
ctx = _Context()
|
|
ctx.set(key, value)
|
|
|
|
ctx.set(key, value, strict=False)
|
|
|
|
|
|
def test_contains() -> None:
|
|
"""
|
|
must correctly check if element is in list
|
|
"""
|
|
key, value = ContextKey("key", int), 42
|
|
ctx = _Context()
|
|
ctx.set(key, value)
|
|
|
|
assert key not in ctx
|
|
assert key.key in ctx
|
|
assert "random-key" not in ctx
|
|
|
|
|
|
def test_iter() -> None:
|
|
"""
|
|
must return keys iterator
|
|
"""
|
|
key, value = ContextKey("key", int), 42
|
|
ctx = _Context()
|
|
ctx.set(key, value)
|
|
|
|
assert set(iter(ctx)) == set(iter({key.key: value}))
|
|
|
|
|
|
def test_len() -> None:
|
|
"""
|
|
must correctly define collection length
|
|
"""
|
|
ctx = _Context()
|
|
ctx.set(ContextKey("str", str), "str")
|
|
ctx.set(ContextKey("int", int), 42)
|
|
|
|
assert len(ctx) == 2
|