"""Tests for the presence/ring datagram builders."""

import base64

from neotalk.core.entities.constants import DatagramType
from neotalk.core.use_cases.presence.build_presence import (
    build_farewell_use_case,
    build_presence_use_case,
    build_ring_use_case,
)


def test_build_presence_carries_the_identity(identity):
    # When building a presence beacon
    response = build_presence_use_case(identity)

    # Then it advertises our fields with a base64 public key
    datagram = response["data"]["datagram"]
    assert datagram["type"] == DatagramType.HELLO
    assert datagram["peer_id"] == identity.peer_id
    assert datagram["alias"] == identity.alias
    assert datagram["tcp_port"] == identity.tcp_port
    assert base64.b64decode(datagram["pubkey"]) == identity.public_key


def test_build_ring_targets_a_peer(identity):
    # When building a ring aimed at a peer
    response = build_ring_use_case(identity, "bob-id")

    # Then it is a ring datagram carrying our identity and the target
    datagram = response["data"]["datagram"]
    assert datagram["type"] == DatagramType.RING
    assert datagram["peer_id"] == identity.peer_id
    assert datagram["to"] == "bob-id"


def test_build_farewell_is_minimal(identity):
    # When building a farewell
    response = build_farewell_use_case(identity)

    # Then it only carries the type and our id
    datagram = response["data"]["datagram"]
    assert datagram["type"] == DatagramType.BYE
    assert datagram["peer_id"] == identity.peer_id
    assert "pubkey" not in datagram