mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 04:04:14 +00:00
efbe5fd477
- add the `llama stack version` command that accepts modifiers to get additional informations - without any argument it displays the build version as the llama --version command - -o table,json allows to change output format - --dependencies display the dependency list used to build llamastack - --build-info displays the build information - --all display both and build version Signed-off-by: Akram Ben Aissi <akram.benaissi@gmail.com>
89 lines
3.5 KiB
Python
89 lines
3.5 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
import argparse
|
|
import json
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from llama_stack.cli.version import VersionCommand
|
|
|
|
|
|
class TestVersionCommand:
|
|
"""Test suite for the VersionCommand class"""
|
|
|
|
@pytest.fixture
|
|
def version_command(self):
|
|
"""Create a VersionCommand instance for testing"""
|
|
parser = argparse.ArgumentParser()
|
|
subparsers = parser.add_subparsers()
|
|
return VersionCommand(subparsers)
|
|
|
|
def test_version_command_basic_functionality(self, version_command):
|
|
"""Test basic version command functionality"""
|
|
# Test package version retrieval
|
|
with patch("llama_stack.cli.version.version") as mock_version:
|
|
mock_version.return_value = "0.2.12"
|
|
assert version_command._get_package_version("llama-stack") == "0.2.12"
|
|
|
|
# Test missing package
|
|
from importlib.metadata import PackageNotFoundError
|
|
|
|
mock_version.side_effect = PackageNotFoundError()
|
|
assert version_command._get_package_version("missing-package") == "unknown"
|
|
|
|
# Test build info with mocked BUILD_INFO
|
|
mock_build_info = {
|
|
"git_commit": "abc123",
|
|
"git_commit_date": "2025-01-15",
|
|
"git_branch": "main",
|
|
"git_tag": "v0.2.12",
|
|
"build_timestamp": "2025-01-15T18:30:00+00:00",
|
|
}
|
|
with patch("llama_stack.cli.version.BUILD_INFO", mock_build_info):
|
|
result = version_command._get_build_info()
|
|
assert result["commit_hash"] == "abc123"
|
|
assert result["branch"] == "main"
|
|
|
|
# Test default JSON output (should only show version)
|
|
args_default = argparse.Namespace(output="json", build_info=False, dependencies=False, all=False)
|
|
with (
|
|
patch.object(version_command, "_get_package_version") as mock_get_version,
|
|
patch("builtins.print") as mock_print,
|
|
):
|
|
mock_get_version.return_value = "0.2.12"
|
|
|
|
version_command._run_version_command(args_default)
|
|
|
|
printed_output = mock_print.call_args[0][0]
|
|
json_output = json.loads(printed_output)
|
|
assert json_output["llama_stack_version"] == "0.2.12"
|
|
# Should not include build info in default output
|
|
assert "git_commit" not in json_output
|
|
|
|
# Test JSON output with build info
|
|
args_with_build_info = argparse.Namespace(output="json", build_info=True, dependencies=False, all=False)
|
|
with (
|
|
patch.object(version_command, "_get_package_version") as mock_get_version,
|
|
patch.object(version_command, "_get_build_info") as mock_get_build_info,
|
|
patch("builtins.print") as mock_print,
|
|
):
|
|
mock_get_version.return_value = "0.2.12"
|
|
mock_get_build_info.return_value = {
|
|
"commit_hash": "abc123",
|
|
"commit_date": "2025-01-15",
|
|
"branch": "main",
|
|
"tag": "v0.2.12",
|
|
"build_timestamp": "2025-01-15T18:30:00+00:00",
|
|
}
|
|
|
|
version_command._run_version_command(args_with_build_info)
|
|
|
|
printed_output = mock_print.call_args[0][0]
|
|
json_output = json.loads(printed_output)
|
|
assert json_output["llama_stack_version"] == "0.2.12"
|
|
assert json_output["git_commit"] == "abc123"
|