6f0df5a774
Stop tracking personal workflow artifacts at repo root, add CI and MIT license, align README and agent skills with artifacts/ defaults, and enable Ruff in dev/CI so releases are verifiable without local-only runs. Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com> Co-authored-by: Cursor <cursoragent@cursor.com>
116 lines
4.0 KiB
Python
116 lines
4.0 KiB
Python
"""Tests for utility functions."""
|
|
|
|
from pathlib import Path
|
|
|
|
from vlm.utils import canonical_path, canonical_path_str, format_size, utc_now
|
|
|
|
|
|
class TestCanonicalPath:
|
|
"""Tests for path canonicalization."""
|
|
|
|
def test_canonical_path_resolves_absolute(self, tmp_path):
|
|
"""Test that canonical_path returns absolute path."""
|
|
rel_path = Path("relative/path/file.txt")
|
|
canonical = canonical_path(rel_path)
|
|
assert canonical.is_absolute()
|
|
|
|
def test_canonical_path_resolves_symlinks(self, tmp_path):
|
|
"""Test that canonical_path resolves symlinks."""
|
|
# Create a real file
|
|
real_file = tmp_path / "real_file.txt"
|
|
real_file.write_text("content")
|
|
|
|
# Create a symlink
|
|
symlink = tmp_path / "symlink.txt"
|
|
symlink.symlink_to(real_file)
|
|
|
|
# Canonicalize both paths
|
|
canonical_real = canonical_path(real_file)
|
|
canonical_symlink = canonical_path(symlink)
|
|
|
|
# Both should resolve to the same canonical path
|
|
assert canonical_real == canonical_symlink
|
|
assert canonical_real == real_file.resolve()
|
|
|
|
def test_canonical_path_with_dot_segments(self, tmp_path):
|
|
"""Test that canonical_path removes . and .. segments."""
|
|
# Create a file
|
|
test_file = tmp_path / "test.txt"
|
|
test_file.write_text("content")
|
|
|
|
# Create path with . and ..
|
|
weird_path = tmp_path / "." / "subdir" / ".." / "test.txt"
|
|
|
|
canonical = canonical_path(weird_path)
|
|
assert canonical == test_file.resolve()
|
|
assert ".." not in str(canonical)
|
|
assert "/." not in str(canonical)
|
|
|
|
def test_canonical_path_str(self, tmp_path):
|
|
"""Test that canonical_path_str returns string."""
|
|
test_file = tmp_path / "test.txt"
|
|
test_file.write_text("content")
|
|
|
|
canonical_str = canonical_path_str(test_file)
|
|
assert isinstance(canonical_str, str)
|
|
assert canonical_str == str(canonical_path(test_file))
|
|
|
|
def test_multiple_symlinks_resolve_to_same_path(self, tmp_path):
|
|
"""Test that multiple symlinks to the same file resolve to the same canonical path."""
|
|
# Create real file
|
|
real_file = tmp_path / "real.txt"
|
|
real_file.write_text("content")
|
|
|
|
# Create multiple symlinks
|
|
symlink1 = tmp_path / "link1.txt"
|
|
symlink2 = tmp_path / "link2.txt"
|
|
symlink3 = tmp_path / "subdir" / "link3.txt"
|
|
symlink3.parent.mkdir(exist_ok=True)
|
|
|
|
symlink1.symlink_to(real_file)
|
|
symlink2.symlink_to(real_file)
|
|
symlink3.symlink_to(real_file)
|
|
|
|
# All should resolve to same canonical path
|
|
canonical_real = canonical_path(real_file)
|
|
canonical_link1 = canonical_path(symlink1)
|
|
canonical_link2 = canonical_path(symlink2)
|
|
canonical_link3 = canonical_path(symlink3)
|
|
|
|
assert canonical_real == canonical_link1 == canonical_link2 == canonical_link3
|
|
|
|
|
|
class TestFormatSize:
|
|
"""Tests for format_size function."""
|
|
|
|
def test_format_size_bytes(self):
|
|
"""Test formatting bytes."""
|
|
assert format_size(0) == "0.0 B"
|
|
assert format_size(512) == "512.0 B"
|
|
assert format_size(1023) == "1023.0 B"
|
|
|
|
def test_format_size_kilobytes(self):
|
|
"""Test formatting kilobytes."""
|
|
assert format_size(1024) == "1.0 KB"
|
|
assert format_size(1536) == "1.5 KB"
|
|
|
|
def test_format_size_megabytes(self):
|
|
"""Test formatting megabytes."""
|
|
assert format_size(1024 * 1024) == "1.0 MB"
|
|
assert format_size(int(1.5 * 1024 * 1024)) == "1.5 MB"
|
|
|
|
def test_format_size_gigabytes(self):
|
|
"""Test formatting gigabytes."""
|
|
assert format_size(1024 * 1024 * 1024) == "1.0 GB"
|
|
assert format_size(int(2.5 * 1024 * 1024 * 1024)) == "2.5 GB"
|
|
|
|
|
|
class TestUtcNow:
|
|
"""Tests for utc_now function."""
|
|
|
|
def test_utc_now_returns_aware_datetime(self):
|
|
"""Test that utc_now returns timezone-aware datetime."""
|
|
now = utc_now()
|
|
assert now.tzinfo is not None
|
|
assert now.tzinfo.tzname(None) == "UTC"
|