mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 10:05:28 +03:00
109 lines
3.6 KiB
Python
109 lines
3.6 KiB
Python
from datetime import date, datetime
|
|
|
|
import pytest
|
|
|
|
from argenta.data_bridge import DataBridge
|
|
from argenta.command.flag.models import InputFlag
|
|
from argenta.command.flag.flags.models import InputFlags
|
|
from argenta.response.entity import EMPTY_INPUT_FLAGS, Response
|
|
from argenta.response.status import ResponseStatus
|
|
|
|
|
|
@pytest.fixture
|
|
def data_bridge():
|
|
"""Create a new DataBridge instance for each test"""
|
|
return DataBridge()
|
|
|
|
|
|
def test_update_data_basic(data_bridge: DataBridge):
|
|
"""Test basic data update functionality"""
|
|
test_data = {"key1": "value1", "key2": "value2"}
|
|
data_bridge.update(test_data)
|
|
assert data_bridge.get_all() == test_data
|
|
|
|
|
|
def test_update_data_with_datetime(data_bridge: DataBridge):
|
|
"""Test updating data with datetime objects"""
|
|
test_datetime = datetime(2024, 1, 15, 10, 30, 45)
|
|
test_data = {"created_at": test_datetime, "name": "test"}
|
|
data_bridge.update(test_data)
|
|
|
|
result = data_bridge.get_all()
|
|
assert result["created_at"] == test_datetime
|
|
assert result["name"] == "test"
|
|
|
|
|
|
def test_update_data_multiple_calls(data_bridge: DataBridge):
|
|
"""Test multiple update calls merge data"""
|
|
first_data = {"key1": "value1"}
|
|
second_data = {"key2": "value2"}
|
|
data_bridge.update(first_data)
|
|
data_bridge.update(second_data)
|
|
assert len(data_bridge.get_all()) == 2
|
|
|
|
|
|
def test_get_data_empty(data_bridge: DataBridge):
|
|
"""Test get_all returns empty dict when no data"""
|
|
assert data_bridge.get_all() == {}
|
|
|
|
|
|
def test_clear_data(data_bridge: DataBridge):
|
|
"""Test clear_all removes all data"""
|
|
data_bridge.update({"key": "value"})
|
|
assert data_bridge.get_all() != {}
|
|
data_bridge.clear_all()
|
|
assert data_bridge.get_all() == {}
|
|
|
|
|
|
def test_delete_from_data(data_bridge: DataBridge):
|
|
"""Test delete_by_key removes specific key"""
|
|
test_data = {"key1": "value1", "key2": "value2"}
|
|
data_bridge.update(test_data)
|
|
data_bridge.delete_by_key("key1")
|
|
result = data_bridge.get_all()
|
|
assert "key1" not in result
|
|
assert "key2" in result
|
|
|
|
|
|
def test_delete_from_data_nonexistent_key(data_bridge: DataBridge):
|
|
"""Test delete_by_key with nonexistent key raises KeyError"""
|
|
with pytest.raises(KeyError):
|
|
data_bridge.delete_by_key("nonexistent_key")
|
|
|
|
|
|
def test_get_by_key(data_bridge: DataBridge):
|
|
"""Test get_by_key retrieves correct value"""
|
|
test_data = {"key1": "value1", "key2": date(2024, 1, 1)}
|
|
data_bridge.update(test_data)
|
|
assert data_bridge.get_by_key("key1") == "value1"
|
|
assert data_bridge.get_by_key("key2") == date(2024, 1, 1)
|
|
assert data_bridge.get_by_key("nonexistent") is None
|
|
|
|
|
|
def test_response_initialization_basic():
|
|
"""Test basic Response initialization"""
|
|
response = Response(ResponseStatus.ALL_FLAGS_VALID)
|
|
assert response.status == ResponseStatus.ALL_FLAGS_VALID
|
|
assert response.input_flags == EMPTY_INPUT_FLAGS
|
|
|
|
|
|
def test_response_initialization_with_flags():
|
|
"""Test Response initialization with input flags"""
|
|
input_flags = InputFlags([InputFlag('test', input_value='value', status=None)])
|
|
response = Response(ResponseStatus.INVALID_VALUE_FLAGS, input_flags)
|
|
assert response.status == ResponseStatus.INVALID_VALUE_FLAGS
|
|
assert response.input_flags == input_flags
|
|
|
|
|
|
def test_response_status_types():
|
|
"""Test Response with different status types"""
|
|
statuses = [
|
|
ResponseStatus.ALL_FLAGS_VALID,
|
|
ResponseStatus.UNDEFINED_FLAGS,
|
|
ResponseStatus.INVALID_VALUE_FLAGS,
|
|
ResponseStatus.UNDEFINED_AND_INVALID_FLAGS
|
|
]
|
|
for status in statuses:
|
|
response = Response(status)
|
|
assert response.status == status
|