tests/test_utils.py

from unittest import mock

from aiozipkin import utils


@mock.patch("aiozipkin.utils.random.getrandbits", autospec=True)
def test_generate_random_64bit_string(rand: mock.Mock) -> None:
    rand.return_value = 0x17133D482BA4F605
    random_string = utils.generate_random_64bit_string()
    assert random_string == "17133d482ba4f605"
    # This acts as a contract test of sorts. This should return a str
    # in both py2 and py3. IOW, no unicode objects.
    assert isinstance(random_string, str)


@mock.patch("aiozipkin.utils.time.time", autospec=True)
@mock.patch("aiozipkin.utils.random.getrandbits", autospec=True)
def test_generate_random_128bit_string(rand: mock.Mock, mock_time: mock.Mock) -> None:
    rand.return_value = 0x2BA4F60517133D482BA4F605
    mock_time.return_value = float(0x17133D48)
    random_string = utils.generate_random_128bit_string()
    assert random_string == "17133d482ba4f60517133d482ba4f605"
    rand.assert_called_once_with(96)  # 96 bits
    # This acts as a contract test of sorts. This should return a str
    # in both py2 and py3. IOW, no unicode objects.
    assert isinstance(random_string, str)


def test_unsigned_hex_to_signed_int() -> None:
    assert utils.unsigned_hex_to_signed_int("17133d482ba4f605") == 1662740067609015813
    assert utils.unsigned_hex_to_signed_int("b6dbb1c2b362bf51") == -5270423489115668655


def test_signed_int_to_unsigned_hex() -> None:
    assert utils.signed_int_to_unsigned_hex(1662740067609015813) == "17133d482ba4f605"
    assert utils.signed_int_to_unsigned_hex(-5270423489115668655) == "b6dbb1c2b362bf51"

    with mock.patch("builtins.hex") as mock_hex:
        mock_hex.return_value = "0xb6dbb1c2b362bf51L"
        assert (
            utils.signed_int_to_unsigned_hex(-5270423489115668655) == "b6dbb1c2b362bf51"
        )