Spaces:
Running
Running
File size: 5,242 Bytes
4353958 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 | """Tests for CLI utilities."""
from datetime import datetime, timedelta
import click
import pytest
from headroom.cli._utils import (
format_age,
format_bytes,
parse_duration,
truncate,
)
class TestParseDuration:
"""Tests for parse_duration function."""
def test_parse_days(self) -> None:
"""Parse days correctly."""
result = parse_duration("7d")
assert result == timedelta(days=7)
def test_parse_weeks(self) -> None:
"""Parse weeks correctly."""
result = parse_duration("2w")
assert result == timedelta(weeks=2)
def test_parse_months(self) -> None:
"""Parse months as 30 days."""
result = parse_duration("1m")
assert result == timedelta(days=30)
def test_parse_hours(self) -> None:
"""Parse hours correctly."""
result = parse_duration("24h")
assert result == timedelta(hours=24)
def test_parse_large_numbers(self) -> None:
"""Parse large duration values."""
result = parse_duration("365d")
assert result == timedelta(days=365)
def test_invalid_format_raises(self) -> None:
"""Invalid format raises BadParameter."""
with pytest.raises(click.BadParameter):
parse_duration("invalid")
def test_empty_string_raises(self) -> None:
"""Empty string raises BadParameter."""
with pytest.raises(click.BadParameter):
parse_duration("")
def test_no_unit_raises(self) -> None:
"""Number without unit raises BadParameter."""
with pytest.raises(click.BadParameter):
parse_duration("7")
def test_invalid_unit_raises(self) -> None:
"""Invalid unit raises BadParameter."""
with pytest.raises(click.BadParameter):
parse_duration("7x")
def test_negative_not_allowed(self) -> None:
"""Negative values raise BadParameter."""
with pytest.raises(click.BadParameter):
parse_duration("-7d")
class TestFormatAge:
"""Tests for format_age function."""
def test_minutes_ago(self) -> None:
"""Format recent time as minutes."""
dt = datetime.now() - timedelta(minutes=30)
result = format_age(dt)
assert result == "30m"
def test_hours_ago(self) -> None:
"""Format hours ago."""
dt = datetime.now() - timedelta(hours=5)
result = format_age(dt)
assert result == "5h"
def test_days_ago(self) -> None:
"""Format days ago."""
dt = datetime.now() - timedelta(days=3)
result = format_age(dt)
assert result == "3d"
def test_weeks_ago(self) -> None:
"""Format weeks ago for older dates."""
dt = datetime.now() - timedelta(weeks=2)
result = format_age(dt)
assert result == "14d"
def test_just_now(self) -> None:
"""Very recent time shows as 'now'."""
dt = datetime.now() - timedelta(seconds=30)
result = format_age(dt)
assert result == "now"
class TestFormatBytes:
"""Tests for format_bytes function."""
def test_bytes(self) -> None:
"""Format small values as bytes."""
assert format_bytes(500) == "500 B"
def test_kilobytes(self) -> None:
"""Format KB correctly."""
assert format_bytes(1024) == "1.0 KB"
assert format_bytes(2048) == "2.0 KB"
def test_megabytes(self) -> None:
"""Format MB correctly."""
assert format_bytes(1024 * 1024) == "1.0 MB"
assert format_bytes(5 * 1024 * 1024) == "5.0 MB"
def test_gigabytes(self) -> None:
"""Format GB correctly."""
assert format_bytes(1024 * 1024 * 1024) == "1.0 GB"
def test_zero_bytes(self) -> None:
"""Zero bytes handled correctly."""
assert format_bytes(0) == "0 B"
def test_fractional_kb(self) -> None:
"""Fractional KB shows decimal."""
result = format_bytes(1536) # 1.5 KB
assert "1.5 KB" == result
class TestTruncate:
"""Tests for truncate function."""
def test_short_string_unchanged(self) -> None:
"""Short strings are not modified."""
result = truncate("hello", max_len=10)
assert result == "hello"
def test_exact_length_unchanged(self) -> None:
"""String at exact length is not modified."""
result = truncate("hello", max_len=5)
assert result == "hello"
def test_long_string_truncated(self) -> None:
"""Long strings are truncated with ellipsis."""
result = truncate("hello world", max_len=8)
assert result == "hello..."
assert len(result) == 8
def test_empty_string(self) -> None:
"""Empty string returns empty."""
result = truncate("", max_len=10)
assert result == ""
def test_default_length(self) -> None:
"""Default max length is 50."""
long_string = "a" * 100
result = truncate(long_string)
assert len(result) == 50
assert result.endswith("...")
def test_with_newlines(self) -> None:
"""Truncate works with newlines in text."""
result = truncate("hello\nworld", max_len=20)
# Newlines are preserved
assert result == "hello\nworld"
|