from sys import argv
from time import sleep
import boto3
import botocore
import requests

try:
    stack = argv[1]
except IndexError:
    print("Please provide a stack name")
    exit(1)

cfn_client = boto3.client('cloudformation')

try:
    stack = cfn_client.describe_stacks(StackName=stack)
except botocore.exceptions.ClientError:
    print("This stack does not exist or region is incorrect")
    exit(1)

ELB_DNS = stack['Stacks'][0]['Outputs'][0]['OutputValue']
for _ in range(0, 2):
    resp = requests.get(f"http://{ELB_DNS}")
    if resp.status_code == 200:
        print("Test succeeded")
        exit(0)
    sleep(5)

print(f"Result of test: {resp.content}")
print(f"HTTP Response code: {resp.status_code}")
print("Test did not succeed")
exit(1)