mirror of
https://github.com/remnawave/python-sdk.git
synced 2026-05-13 20:26:50 +00:00
9e2822774a
- Updated NodeUsageDto to include node_uuid and changed date to datetime. - Changed total fields in subscription and subscription request history models from float to int. - Introduced response rules and conditions for subscription settings, including new models for response modifications. - Added external squads management with CRUD operations and associated models. - Implemented passkey management with registration and verification endpoints. - Enhanced snippets management with full CRUD operations and validation. - Added OAuth2 provider enum for better authentication handling.
77 lines
No EOL
2.6 KiB
Python
77 lines
No EOL
2.6 KiB
Python
import pytest
|
|
|
|
from remnawave.models import (
|
|
CreateSnippetRequestDto,
|
|
DeleteSnippetRequestDto,
|
|
GetSnippetsResponseDto,
|
|
UpdateSnippetRequestDto,
|
|
)
|
|
|
|
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_snippets_full_workflow(remnawave):
|
|
# Test getting all snippets
|
|
snippets = await remnawave.snippets.get_snippets()
|
|
assert isinstance(snippets, GetSnippetsResponseDto)
|
|
initial_count = snippets.total
|
|
|
|
# Test creating a snippet with unique name
|
|
rand_name = random_string()
|
|
create_request = CreateSnippetRequestDto(
|
|
name=rand_name,
|
|
snippet=[
|
|
{"type": "vmess", "port": 443},
|
|
{"type": "vless", "encryption": "none"}
|
|
]
|
|
)
|
|
created = await remnawave.snippets.create_snippet(create_request)
|
|
assert created.total == initial_count + 1
|
|
|
|
# Verify snippet was created
|
|
snippets_after_create = await remnawave.snippets.get_snippets()
|
|
snippet_names = [s.name for s in snippets_after_create.snippets]
|
|
assert rand_name in snippet_names
|
|
|
|
# Test updating the snippet
|
|
update_request = UpdateSnippetRequestDto(
|
|
name=rand_name,
|
|
snippet=[
|
|
{"type": "shadowsocks", "method": "aes-256-gcm"},
|
|
{"type": "trojan", "password": "test-password"}
|
|
]
|
|
)
|
|
updated = await remnawave.snippets.update_snippet(update_request)
|
|
assert updated.total == initial_count + 1
|
|
|
|
# Test deleting the snippet
|
|
delete_request = DeleteSnippetRequestDto(name=rand_name)
|
|
deleted = await remnawave.snippets.delete_snippet_by_name(delete_request)
|
|
assert deleted.total == initial_count
|
|
|
|
# Verify snippet was deleted
|
|
snippets_after_delete = await remnawave.snippets.get_snippets()
|
|
snippet_names_after = [s.name for s in snippets_after_delete.snippets]
|
|
assert rand_name not in snippet_names_after
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_snippet_name_validation(remnawave):
|
|
"""Test that snippet names are properly validated"""
|
|
# Valid names
|
|
valid_names = ["Test Snippet", "My_Snippet", "Snippet-123", "A B C"]
|
|
|
|
for name in valid_names:
|
|
request = CreateSnippetRequestDto(name=name, snippet=[{"test": "data"}])
|
|
# Should not raise validation error
|
|
assert request.name == name
|
|
|
|
# Invalid names would be caught by Pydantic validation
|
|
with pytest.raises(ValueError):
|
|
CreateSnippetRequestDto(name="x", snippet=[]) # Too short
|
|
|
|
with pytest.raises(ValueError):
|
|
CreateSnippetRequestDto(name="", snippet=[]) # Empty name |