from app.libs.aras_response import aras_ok, aras_server_error


def test_aras_ok_default():
    result = aras_ok("test message")
    assert result["status"] == "OK"
    assert result["message"] == "test message"
    assert result["data"] is None
    assert result["extra"] is None
    assert result["code"] == "UNKNOWN_CODE"


def test_aras_ok_with_data():
    result = aras_ok("test", data={"key": "value"})
    assert result["data"] == {"key": "value"}


def test_aras_ok_with_all_params():
    result = aras_ok(
        "test",
        data={"key": "value"},
        extra={"extra_key": "extra_value"},
        code="CUSTOM_CODE",
    )
    assert result["status"] == "OK"
    assert result["message"] == "test"
    assert result["data"] == {"key": "value"}
    assert result["extra"] == {"extra_key": "extra_value"}
    assert result["code"] == "CUSTOM_CODE"


def test_aras_server_error_default():
    result = aras_server_error("error message")
    assert result["status"] == "NO_OK"
    assert result["message"] == "error message"
    assert result["data"] is None
    assert result["extra"] is None
    assert result["code"] == "UNKNOWN_CODE"


def test_aras_server_error_with_data():
    result = aras_server_error("error", data={"error_key": "error_value"})
    assert result["data"] == {"error_key": "error_value"}


def test_aras_server_error_with_all_params():
    result = aras_server_error(
        "error",
        data={"error_key": "error_value"},
        extra={"extra_key": "extra_value"},
        code="ERROR_CODE",
    )
    assert result["status"] == "NO_OK"
    assert result["message"] == "error"
    assert result["data"] == {"error_key": "error_value"}
    assert result["extra"] == {"extra_key": "extra_value"}
    assert result["code"] == "ERROR_CODE"