from remnawave.models import GetNodesUsageByRangeResponseDto, GetNodesRealtimeUsageResponseDto
from tests.utils import generate_isoformat_range


async def test_bandwidthstats(remnawave):
    start, end = generate_isoformat_range()
    nodes_usage_by_range = await remnawave.bandwidthstats.get_nodes_usage_by_range(
        start=start, end=end
    )
    assert isinstance(nodes_usage_by_range, GetNodesUsageByRangeResponseDto)
    
    # Test realtime usage
    realtime_usage = await remnawave.bandwidthstats.get_nodes_usage_realtime()
    assert isinstance(realtime_usage, GetNodesRealtimeUsageResponseDto)