strukto-ai--mirage/python/tests/core/nextcloud/test_du.py

import pytest

from mirage.core.nextcloud.du import du, du_all
from mirage.types import PathSpec


@pytest.mark.asyncio
async def test_du_sums_file_sizes_recursive(make_acc):
    acc = make_acc({
        "data/a.json": b"12345",
        "data/sub/b.json": b"67",
        "other.txt": b"x",
    })
    total = await du(acc, PathSpec.from_str_path("/data"))
    assert total == 7


@pytest.mark.asyncio
async def test_du_all_returns_per_file_with_total(make_acc):
    acc = make_acc({
        "data/a.json": b"12345",
        "data/b.json": b"67",
    })
    out = await du_all(acc, PathSpec.from_str_path("/data"))
    assert out[-1] == ("/data", 7)
    files = sorted(out[:-1])
    assert files == [("/data/a.json", 5), ("/data/b.json", 2)]


@pytest.mark.asyncio
async def test_du_missing_returns_zero(make_acc):
    acc = make_acc({})
    assert await du(acc, PathSpec.from_str_path("/nope")) == 0


@pytest.mark.asyncio
async def test_du_of_file_returns_its_own_size(make_acc):
    acc = make_acc({"data/a.json": b"12345"})
    assert await du(acc, PathSpec.from_str_path("/data/a.json")) == 5


@pytest.mark.asyncio
async def test_du_all_of_file_is_empty(make_acc):
    acc = make_acc({"data/a.json": b"12345"})
    assert await du_all(acc, PathSpec.from_str_path("/data/a.json")) == []