Esempio n. 1
0
def test_create_sink_creates_LambdaSink():
    # arrange
    env = LambdaEnvironment()

    # act
    result = env.get_sink()

    # assert
    assert isinstance(result, StdoutSink)
Esempio n. 2
0
def test_get_type_returns_cfn_lambda_name():
    # arrange
    env = LambdaEnvironment()

    # act
    result = env.get_type()

    # assert
    assert result == "AWS::Lambda::Function"
Esempio n. 3
0
def test_get_log_group_name_returns_function_name():
    # arrange
    expected_name = fake.word()
    os.environ["AWS_LAMBDA_FUNCTION_NAME"] = expected_name
    env = LambdaEnvironment()

    # act
    result = env.get_log_group_name()

    # assert
    assert result == expected_name
Esempio n. 4
0
async def test_probe_returns_true_if_fcn_name_in_env():
    # arrange
    os.environ["AWS_LAMBDA_FUNCTION_NAME"] = fake.word()
    env = LambdaEnvironment()

    # act
    result = await env.probe()

    # assert
    assert result is True
Esempio n. 5
0
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

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.ec2_environment import EC2Environment
from typing import Optional

log = logging.getLogger(__name__)

lambda_environment = LambdaEnvironment()
ec2_environment = EC2Environment()
default_environment = DefaultEnvironment()
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
Esempio n. 6
0
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import logging
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.ec2_environment import EC2Environment
from typing import Optional

log = logging.getLogger(__name__)

environments = [LambdaEnvironment(), EC2Environment()]


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

    for env_under_test in environments:
        is_environment = False
        try:
            log.info("Testing environment: %s", env_under_test.__class__.__name__)