def test_log_to_cli(iam_client_stub, ec2_client_stub):
config = helpers.load_aws_example_config_file("example-full.yaml")
head_node_config = config["available_node_types"]["ray.head.default"][
"node_config"]
worker_node_config = config["available_node_types"]["ray.worker.default"][
"node_config"]
# Pass in SG for stub to work
head_node_config["SecurityGroupIds"] = ["sg-1234abcd"]
worker_node_config["SecurityGroupIds"] = ["sg-1234abcd"]
stubs.configure_iam_role_default(iam_client_stub)
stubs.configure_key_pair_default(ec2_client_stub)
stubs.describe_a_security_group(ec2_client_stub, DEFAULT_SG)
stubs.configure_subnet_default(ec2_client_stub)
config = helpers.bootstrap_aws_config(config)
# Only side-effect is to print logs to cli, called just to
# check that it runs without error
log_to_cli(config)
iam_client_stub.assert_no_pending_responses()
ec2_client_stub.assert_no_pending_responses()
def try_logging_config(config):
if config["provider"]["type"] == "aws":
from ray.autoscaler._private.aws.config import log_to_cli
log_to_cli(config)
