OpenViking/tests/cli/test_cli_content.py

# Copyright (c) 2026 Beijing Volcano Engine Technology Co., Ltd.
# SPDX-License-Identifier: AGPL-3.0
"""CLI content operation tests (read, abstract, overview, download, write, reindex)."""

import os
import tempfile
import uuid

import pytest
from conftest import ov, ov_add_resource, ov_reindex, ov_rm, ov_write

pytestmark = pytest.mark.cli_remote


class TestContentRead:
    def test_read(self, test_file_uri):
        r = ov(["read", test_file_uri, "-o", "json"])
        assert r["exit_code"] == 0, (
            f"ov read should exit 0, got {r['exit_code']}: {r['stderr'][:300]}"
        )
        assert len(r["stdout"]) > 0, "read output should not be empty"
        assert "CLI Test" in r["stdout"] or "test" in r["stdout"].lower(), (
            f"read output should contain test content, got: {r['stdout'][:200]}"
        )


class TestContentAbstract:
    def test_abstract(self, test_pack_uri):
        r = ov(["abstract", test_pack_uri, "-o", "json"])
        assert r["exit_code"] == 0, (
            f"ov abstract should exit 0, got {r['exit_code']}: {r['stderr'][:300]}"
        )
        assert len(r["stdout"]) > 0, "abstract output should not be empty"


class TestContentOverview:
    def test_overview(self, test_pack_uri):
        r = ov(["overview", test_pack_uri, "-o", "json"])
        assert r["exit_code"] == 0, (
            f"ov overview should exit 0, got {r['exit_code']}: {r['stderr'][:300]}"
        )
        assert len(r["stdout"]) > 0, "overview output should not be empty"


class TestContentDownload:
    def test_get_download(self, test_file_uri, tmp_path):
        local_path = str(tmp_path / "downloaded.txt")
        r = ov(["get", test_file_uri, local_path, "-o", "json"])
        assert r["exit_code"] == 0, (
            f"ov get should exit 0, got {r['exit_code']}: {r['stderr'][:300]}"
        )
        assert os.path.exists(local_path), "downloaded file should exist"
        assert os.path.getsize(local_path) > 0, "downloaded file should not be empty"


class TestContentWrite:
    def test_write_replace(self, test_file_uri):
        r = ov_write(test_file_uri, "Updated via CLI write.", "--mode", "replace")
        assert r["exit_code"] == 0, (
            f"ov write replace should exit 0, got {r['exit_code']}: {r['stderr'][:300]}"
        )
        data = r["json"]
        assert data is not None and data.get("ok") is True, "Expected ok=true"

    def test_write_append(self, test_file_uri):
        r = ov_write(test_file_uri, "\nAppended via CLI.", "--append")
        assert r["exit_code"] == 0, (
            f"ov write append should exit 0, got {r['exit_code']}: {r['stderr'][:300]}"
        )
        data = r["json"]
        assert data is not None and data.get("ok") is True, "Expected ok=true"


class TestContentReindex:
    def test_reindex(self):
        reindex_pack = f"viking://~/resources/reindex_{uuid.uuid4().hex[:6]}"
        with tempfile.NamedTemporaryFile(suffix=".txt", delete=False, mode="w") as f:
            f.write("# Reindex Test\n\nThis is an independent resource for reindex testing.")
            temp_path = f.name
        try:
            try:
                r = ov_add_resource(temp_path, reindex_pack)
                assert r["exit_code"] == 0, f"add-resource for reindex failed: {r['stderr'][:300]}"
            finally:
                os.unlink(temp_path)

            r = ov_reindex(reindex_pack)
            assert r["exit_code"] == 0, (
                f"ov reindex should exit 0, got {r['exit_code']}: {r['stderr'][:300]}"
            )
            data = r["json"]
            assert data is not None and data.get("ok") is True, "Expected ok=true"
        finally:
            ov_rm(reindex_pack)