Remnawave_python-sdk/tests/test_subscriptions_template.py

import pytest
from remnawave.enums import TemplateType
from remnawave.models import (
    CreateSubscriptionTemplateRequestDto,
    CreateSubscriptionTemplateResponseDto,
    DeleteSubscriptionTemplateResponseDto,
    GetTemplateResponseDto,
    GetTemplatesResponseDto,
    ReorderTemplateItem,
    ReorderSubscriptionTemplatesRequestDto,
    ReorderSubscriptionTemplatesResponseDto,
    UpdateTemplateRequestDto,
    UpdateTemplateResponseDto,
)

def random_string(length=10):
    import random
    import string
    return ''.join(random.choices(string.ascii_letters + string.digits, k=length))

@pytest.mark.asyncio
async def test_get_all_templates(remnawave):
    """Проверка получения всех шаблонов"""
    templates = await remnawave.subscriptions_template.get_all_templates()
    assert isinstance(templates, GetTemplatesResponseDto)


@pytest.mark.asyncio
async def test_create_template(remnawave):
    """Проверка создания шаблона"""
    rand_name = random_string()
    create_request = CreateSubscriptionTemplateRequestDto(
        name=rand_name,
        template_type=TemplateType.SINGBOX,
    )
    created_template = await remnawave.subscriptions_template.create_template(create_request)
    assert isinstance(created_template, CreateSubscriptionTemplateResponseDto)
    assert created_template.name == rand_name
    assert created_template.template_type == TemplateType.SINGBOX

    # Удаляем после проверки
    await remnawave.subscriptions_template.delete_template(str(created_template.uuid))


@pytest.fixture
async def created_template(remnawave):
    """Фикстура: создать временный шаблон и удалить после теста"""
    create_request = CreateSubscriptionTemplateRequestDto(
        name="Temp Template",
        template_type=TemplateType.SINGBOX,
    )
    template = await remnawave.subscriptions_template.create_template(create_request)
    yield template
    await remnawave.subscriptions_template.delete_template(str(template.uuid))


@pytest.mark.asyncio
async def test_get_template_by_uuid(remnawave, created_template):
    """Проверка получения шаблона по UUID"""
    template = await remnawave.subscriptions_template.get_template_by_uuid(
        str(created_template.uuid)
    )
    assert isinstance(template, GetTemplateResponseDto)
    assert template.uuid == created_template.uuid


@pytest.mark.asyncio
async def test_update_template(remnawave, created_template):
    """Проверка обновления шаблона"""
    update_request = UpdateTemplateRequestDto(
        uuid=created_template.uuid,
        name="Updated Template Name",
    )
    updated_template = await remnawave.subscriptions_template.update_template(update_request)
    assert isinstance(updated_template, UpdateTemplateResponseDto)
    assert updated_template.name == "Updated Template Name"


@pytest.mark.asyncio
async def test_delete_template(remnawave):
    """Проверка удаления шаблона"""
    # Сначала создаем
    create_request = CreateSubscriptionTemplateRequestDto(
        name="Temp Delete Template",
        template_type=TemplateType.SINGBOX,
    )
    created = await remnawave.subscriptions_template.create_template(create_request)

    # Теперь удаляем
    delete_response = await remnawave.subscriptions_template.delete_template(
        str(created.uuid)
    )
    assert isinstance(delete_response, DeleteSubscriptionTemplateResponseDto)
    assert delete_response.is_deleted is True


@pytest.mark.asyncio
async def test_reorder_templates(remnawave):
    """Проверка изменения порядка шаблонов"""
    templates = await remnawave.subscriptions_template.get_all_templates()
    assert isinstance(templates, GetTemplatesResponseDto)

    if len(templates.templates) >= 2:
        items = [
            ReorderTemplateItem(
                uuid=tmpl.uuid,
                view_position=idx
            )
            for idx, tmpl in enumerate(templates.templates)
        ]
        reorder_result = await remnawave.subscriptions_template.reorder_templates(
            ReorderSubscriptionTemplatesRequestDto(items=items)
        )
        assert isinstance(reorder_result, ReorderSubscriptionTemplatesResponseDto)