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