def test__get_sink__returns_stdout_sink(): # arrange env = LocalEnvironment() # act result = env.get_sink() # assert assert isinstance(result, StdoutSink)
def test_get_type_returns_unknown_if_not_specified(): # arrange env = LocalEnvironment() Config.service_type = None # act result = env.get_type() # assert assert result == "Unknown"
def test__get_log_group_name__returns_log_group_if_configured(): # arrange expected = fake.word() env = LocalEnvironment() Config.log_group_name = expected # act result = env.get_log_group_name() # assert assert result == expected
def test__get_type__returns_type_if_configured(): # arrange expected_type = fake.word() env = LocalEnvironment() Config.service_type = expected_type # act result = env.get_type() # assert assert result == expected_type
def test__get_log_group_name__returns_service_name_metrics_if_not_configured(): # arrange expected = fake.word() env = LocalEnvironment() Config.service_name = expected Config.log_group_name = None # act result = env.get_log_group_name() # assert assert result == f"{expected}-metrics"
async def test_probe_always_returns_false(): # arrange env = LocalEnvironment() # act result = await env.probe() # assert assert result is False
import logging from aws_embedded_metrics import config from aws_embedded_metrics.environment import Environment from aws_embedded_metrics.environment.default_environment import DefaultEnvironment from aws_embedded_metrics.environment.lambda_environment import LambdaEnvironment from aws_embedded_metrics.environment.local_environment import LocalEnvironment from aws_embedded_metrics.environment.ec2_environment import EC2Environment from typing import Optional log = logging.getLogger(__name__) lambda_environment = LambdaEnvironment() ec2_environment = EC2Environment() default_environment = DefaultEnvironment() local_environment = LocalEnvironment() environments = [lambda_environment, ec2_environment] Config = config.get_config() class EnvironmentCache: environment: Optional[Environment] = None async def resolve_environment() -> Environment: if EnvironmentCache.environment is not None: log.debug("Environment resolved from cache.") return EnvironmentCache.environment if Config.environment: lower_configured_enviroment = Config.environment.lower()