def test_yatai_service_start():
runner = CliRunner()
cli = create_bento_service_cli()
add_yatai_service_sub_command(cli)
yatai_service_start_cmd = cli.commands["yatai-service-start"]
with mock.patch("bentoml.cli.yatai_service.start_yatai_service_grpc_server"
) as mocked_start_yatai_service_grpc_server:
runner.invoke(yatai_service_start_cmd)
mocked_start_yatai_service_grpc_server.assert_called()
mocked_start_yatai_service_grpc_server.assert_called_with(
db_url=SQLITE_DATABASE_URL,
grpc_port=50051,
ui_port=3000,
with_ui=True,
base_url=".",
repository_type="file_system",
file_system_directory=FILE_SYSTEM_REPOSITORY,
s3_url=None,
s3_endpoint_url=None,
gcs_url=None,
)
runner.invoke(yatai_service_start_cmd,
["--repo-base-url=s3://url_address"])
mocked_start_yatai_service_grpc_server.assert_called()
mocked_start_yatai_service_grpc_server.assert_called_with(
db_url=SQLITE_DATABASE_URL,
grpc_port=50051,
ui_port=3000,
with_ui=True,
base_url=".",
repository_type="s3",
file_system_directory=FILE_SYSTEM_REPOSITORY,
s3_url="s3://url_address",
s3_endpoint_url=None,
gcs_url=None,
)
runner.invoke(yatai_service_start_cmd,
["--repo-base-url=gs://url_address"])
mocked_start_yatai_service_grpc_server.assert_called()
mocked_start_yatai_service_grpc_server.assert_called_with(
db_url=SQLITE_DATABASE_URL,
grpc_port=50051,
ui_port=3000,
with_ui=True,
base_url=".",
repository_type="gcs",
file_system_directory=FILE_SYSTEM_REPOSITORY,
s3_url=None,
s3_endpoint_url=None,
gcs_url="gs://url_address",
)
def create_bentoml_cli():
# pylint: disable=unused-variable
_cli = create_bento_service_cli()
# Commands created here aren't mean to be used from generated BentoService CLI when
# installed as PyPI package. The are only used as part of BentoML cli command.
config_sub_command = get_configuration_sub_command()
aws_sagemaker_sub_command = get_aws_sagemaker_sub_command()
aws_lambda_sub_command = get_aws_lambda_sub_command()
deployment_sub_command = get_deployment_sub_command()
add_bento_sub_command(_cli)
add_yatai_service_sub_command(_cli)
_cli.add_command(config_sub_command)
_cli.add_command(aws_sagemaker_sub_command)
_cli.add_command(aws_lambda_sub_command)
_cli.add_command(deployment_sub_command)
return _cli
