[Feature] add cli command chat,complete (#4037)

This commit is contained in:
memoryCoderC
2025-09-11 19:53:14 +08:00
committed by GitHub
parent a47976e82d
commit 850465e8ed
8 changed files with 596 additions and 0 deletions

View File

@@ -0,0 +1,26 @@
import unittest
from unittest.mock import MagicMock, patch
from fastdeploy.entrypoints.cli.main import main as cli_main
class TestCliMain(unittest.TestCase):
@patch("fastdeploy.utils.FlexibleArgumentParser")
@patch("fastdeploy.entrypoints.cli.main.importlib.metadata")
def test_main_basic(self, mock_metadata, mock_parser):
# Setup mocks
mock_metadata.version.return_value = "1.0.0"
mock_args = MagicMock()
mock_args.subparser = None
mock_parser.return_value.parse_args.return_value = mock_args
# Test basic call
cli_main()
# Verify version check
mock_metadata.version.assert_called_once_with("fastdeploy")
mock_args.dispatch_function.assert_called_once()
if __name__ == "__main__":
unittest.main()