test_functional.py

from pathlib import Path

import pytest

from cloud_radar.cf.e2e._stack import Stack


@pytest.fixture(scope="session")
def template_dir():
    return Path(__file__).parent / "../../templates"


@pytest.fixture()
def default_params():
    parameters = {
        "BucketPrefix": "taskcat-$[taskcat_random-string]",
        "KeepBucket": "FALSE",
    }

    return parameters


@pytest.mark.e2e
def test_log_bucket(template_dir, default_params):
    template = template_dir / "log_bucket" / "log_bucket.yaml"

    regions = ["us-west-1", "us-west-2"]

    with Stack(template, default_params, regions) as stacks:
        for stack in stacks:
            print(f"Testing {stack.name}")

            bucket_name = ""

            for output in stack.outputs:
                if output.key == "LogsBucketName":
                    bucket_name = output.value
                    break

            assert "logs" in bucket_name

            assert stack.region.name in bucket_name

            print(f"Created bucket: {bucket_name}")

        assert len(stacks) == 2


@pytest.mark.e2e
def test_retain_bucket(template_dir, default_params):
    default_params["KeepBucket"] = "TRUE"

    template = template_dir / "log_bucket" / "log_bucket.yaml"

    with Stack(template, default_params) as stacks:
        stack_count = len(stacks)

    for stack in stacks:
        session = stack.region.session

        s3 = session.resource("s3")

        for output in stack.outputs:
            if output.key == "LogsBucketName":
                bucket = s3.Bucket(output.value)
                bucket.wait_until_exists()
                bucket.delete()
                bucket.wait_until_not_exists()
                break

    assert stack_count == 1