Remnawave_python-sdk/tests/test_hwid.py

import random
import uuid

import pytest

from remnawave_api.models import (
    HWIDDeleteRequest,
    HWIDUserResponseDto,
    HWIDUserResponseDtoList,
    CreateHWIDUser
)
from tests.conftest import REMNAWAVE_USER_UUID

new_hwid = str(uuid.uuid4())

@pytest.mark.asyncio
async def test_get_hwid_user(remnawave):
    hwid = await remnawave.hwid.get_hwid_user(uuid=REMNAWAVE_USER_UUID)
    assert isinstance(hwid, HWIDUserResponseDtoList)
    assert hwid.response is not None


@pytest.mark.asyncio
async def test_add_hwid_to_user(remnawave):
    create_request = CreateHWIDUser(
        hwid=new_hwid,
        userUuid=REMNAWAVE_USER_UUID,
        platform="Windows",
        osVersion="10.0.19042",
        deviceModel="Surface Pro",
        userAgent="Mozilla/5.0"
    )
    response = await remnawave.hwid.add_hwid_to_users(body=create_request)
    assert isinstance(response, HWIDUserResponseDtoList)
    assert any(item.hwid == new_hwid for item in response.response)


@pytest.mark.asyncio
async def test_delete_hwid_user(remnawave):
    delete_request = HWIDDeleteRequest(
        hwid=new_hwid,
        userUuid=REMNAWAVE_USER_UUID
    )
    response = await remnawave.hwid.delete_hwid_to_user(body=delete_request)
    assert isinstance(response, HWIDUserResponseDtoList)
    assert not any(item.hwid == new_hwid for item in response.response)