mirror of
https://github.com/remnawave/python-sdk.git
synced 2026-05-13 12:16:42 +00:00
950a7ffab3
- Implemented models for nodes usage history including NodeInfoDto, GetUserAccessibleNodesResponse, and usage statistics. - Created subscription-related models such as UserSubscription, SubscriptionInfoData, and various response DTOs for subscription information. - Developed models for subscription settings, templates, and system statistics, enhancing the API's capability to manage and retrieve subscription configurations. - Introduced user management models including CreateUserRequestDto, UpdateUserRequestDto, and various response DTOs for user actions. - Added bulk actions for user updates and statistics tracking, improving the efficiency of user management operations. - Established a base controller for handling API requests and responses, along with decorators for HTTP methods to streamline API interactions. - Included utility functions for serialization using orjson for better performance with Pydantic models.
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
from uuid import UUID
|
|
import pytest
|
|
|
|
from remnawave.models import (
|
|
AddUsersToInternalSquadRequestDto,
|
|
AddUsersToInternalSquadResponseDto,
|
|
CreateInternalSquadRequestDto,
|
|
CreateInternalSquadResponseDto,
|
|
DeleteInternalSquadResponseDto,
|
|
DeleteUsersFromInternalSquadRequestDto,
|
|
DeleteUsersFromInternalSquadResponseDto,
|
|
GetAllInternalSquadsResponseDto,
|
|
GetInternalSquadByUuidResponseDto,
|
|
UpdateInternalSquadRequestDto,
|
|
UpdateInternalSquadResponseDto,
|
|
)
|
|
from tests.conftest import REMNAWAVE_INBOUND_UUID
|
|
from tests.utils import generate_random_string
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_internal_squads(remnawave) -> None:
|
|
squad_name = f"test_squad_{generate_random_string(length=6)}"
|
|
|
|
# Test create internal squad
|
|
create_squad = await remnawave.internal_squads.create_internal_squad(
|
|
CreateInternalSquadRequestDto(name=squad_name, inbounds=[
|
|
REMNAWAVE_INBOUND_UUID
|
|
])
|
|
)
|
|
|
|
assert isinstance(create_squad, CreateInternalSquadResponseDto)
|
|
assert create_squad.name == squad_name
|
|
|
|
squad_uuid = str(create_squad.uuid)
|
|
|
|
# Test get all internal squads
|
|
all_squads = await remnawave.internal_squads.get_internal_squads()
|
|
assert isinstance(all_squads, GetAllInternalSquadsResponseDto)
|
|
assert len(all_squads.internal_squads) > 0
|
|
|
|
# Test get internal squad by uuid
|
|
squad_by_uuid = await remnawave.internal_squads.get_internal_squad_by_uuid(squad_uuid)
|
|
assert isinstance(squad_by_uuid, GetInternalSquadByUuidResponseDto)
|
|
assert squad_by_uuid.name == squad_name
|
|
|
|
# Test update internal squad
|
|
update_squad = await remnawave.internal_squads.update_internal_squad(
|
|
UpdateInternalSquadRequestDto(
|
|
uuid=create_squad.uuid,
|
|
inbounds=[REMNAWAVE_INBOUND_UUID],
|
|
)
|
|
)
|
|
|
|
assert isinstance(update_squad, UpdateInternalSquadResponseDto)
|
|
assert update_squad.inbounds[0].uuid == UUID(REMNAWAVE_INBOUND_UUID)
|
|
|
|
# Test add users to internal squad (with dummy UUIDs for testing)
|
|
dummy_user_uuids = [] # Empty list for test
|
|
add_users = await remnawave.internal_squads.add_users_to_internal_squad(
|
|
squad_uuid,
|
|
)
|
|
|
|
assert isinstance(add_users, AddUsersToInternalSquadResponseDto)
|
|
|
|
# Test remove users from internal squad
|
|
remove_users = await remnawave.internal_squads.remove_users_from_internal_squad(
|
|
squad_uuid,
|
|
)
|
|
|
|
assert isinstance(remove_users, DeleteUsersFromInternalSquadResponseDto)
|
|
|
|
# Test delete internal squad
|
|
delete_squad = await remnawave.internal_squads.delete_internal_squad(squad_uuid)
|
|
assert isinstance(delete_squad, DeleteInternalSquadResponseDto)
|
|
assert delete_squad.is_deleted is True
|