"""Tests for ClientMessage and SendFlags."""

import hashlib

from i2p_data.client_message import ClientMessage, SendFlags


def test_client_message_creation():
    msg = ClientMessage(
        from_dest=b"\x01" * 387,
        to_dest=b"\x02" * 387,
        payload=b"hello",
        message_id=42,
        message_nonce=7,
    )
    assert msg.payload == b"hello"
    assert msg.message_id == 42
    assert msg.payload_size == 5


def test_client_message_hashes():
    from_dest = b"\x01" * 387
    to_dest = b"\x02" * 387
    msg = ClientMessage(
        from_dest=from_dest, to_dest=to_dest,
        payload=b"x", message_id=1, message_nonce=1,
    )
    assert msg.from_hash == hashlib.sha256(from_dest).digest()
    assert msg.to_hash == hashlib.sha256(to_dest).digest()


def test_send_flags():
    assert SendFlags.NONE == 0
    assert SendFlags.TAG_THRESHOLD_LOW == 1
    assert SendFlags.TAG_THRESHOLD_HIGH == 2