mkdir testenvpinger
cd testenvpinger

import os
from aws_cdk.core import Stack, Construct, Environment

# we need default values here since aws-cdk-examples build synthesizes the app
ACCOUNT= os.environ.get('TESTENV_ACCOUNT', '123456789012')
REGION = os.environ.get('TESTENV_REGION', 'ap-northeast-1')
VPC_ID = os.environ.get('TESTENV_VPC_ID', 'vpc-0e680e107cb16f0c9')

AWS_ENV = Environment(account=ACCOUNT, region=REGION)

class TestEnvStack(Stack):
    """
    A base CDK stack class for all stacks defined in our fun little company.
    """

    def __init__(self, scope: Construct, id: str, **kwargs):
        super().__init__(scope, id, env=AWS_ENV, **kwargs)

        # lookup our pre-created VPC by ID
        self._vpc = aws_ec2.Vpc.from_lookup(self, "vpc", vpc_id=VPC_ID)

    @property
    def testenv_vpc(self) -> aws_ec2.IVpc:
        """
        :return: The test env vpc
        """
        return self._vpc

__all__ = ["TestEnvStack"]

from aws_cdk.core import Construct
from aws_cdk import aws_ecs, aws_ec2


# a user-defined construct
# just a class the inherits from the core.Construct base class
class Traffic(Construct):
    """
    An HTTP traffic generator.
    Hits a specified URL at some TPS.
    """

    def __init__(self, scope: Construct, id: str, *, vpc: aws_ec2.IVpc, url: str, tps: int):
        """
        Defines an instance of the traffic generator.
        :param scope: construct scope
        :param id:    construct id
        :param vpc:   the VPC in which to host the traffic generator
        :param url:   the URL to hit
        :param tps:   the number of transactions per second
        """
        super().__init__(scope, id)

        # define an ECS cluster hosted within the requested VPC
        cluster = aws_ecs.Cluster(self, 'cluster', vpc=vpc)

        # define our task definition with a single container
        # the image is built & published from a local asset directory
        task_definition = aws_ecs.FargateTaskDefinition(self, 'PingTask')
        task_definition.add_container('Pinger',
                                      image=aws_ecs.ContainerImage.from_asset("pinger"),
                                     "pinger"), environment={'URL': url})

        # define our fargate service. TPS determines how many instances we
        # want from our task (each task produces a single TPS)
        aws_ecs.FargateService(self, 'service',
 cluster=cluster,                              cluster=cluster,
                               task_definition=task_definition,
                               desired_count=tps)

task_definition=task_definition, desired_count=tps)

#!/usr/bin/env python3

from aws_cdk import core

from url_short.url_short_stack import UrlShortStack, TrafficStack

app = core.App()
UrlShortStack(app, "url-short")
TrafficStack(app,  "test-trafficping")

app.synth()