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