Merge remote-tracking branch 'nielslaukens/feature/mock_eb' into feature/mock_eb

tests/test_eb/test_eb.py

@@ -0,0 +1,152 @@
+import boto3
+import sure  # noqa
+from botocore.exceptions import ClientError
+
+from moto import mock_eb
+
+
+@mock_eb
+def test_create_application():
+    # Create Elastic Beanstalk Application
+    conn = boto3.client('elasticbeanstalk', region_name='us-east-1')
+    app = conn.create_application(
+        ApplicationName="myapp",
+    )
+    app['Application']['ApplicationName'].should.equal("myapp")
+
+
+@mock_eb
+def test_create_application_dup():
+    conn = boto3.client('elasticbeanstalk', region_name='us-east-1')
+    conn.create_application(
+        ApplicationName="myapp",
+    )
+    conn.create_application.when.called_with(
+        ApplicationName="myapp",
+    ).should.throw(ClientError)
+
+
+@mock_eb
+def test_describe_applications():
+    # Create Elastic Beanstalk Application
+    conn = boto3.client('elasticbeanstalk', region_name='us-east-1')
+    conn.create_application(
+        ApplicationName="myapp",
+    )
+
+    apps = conn.describe_applications()
+    len(apps['Applications']).should.equal(1)
+    apps['Applications'][0]['ApplicationName'].should.equal('myapp')
+
+
+@mock_eb
+def test_create_environment():
+    # Create Elastic Beanstalk Environment
+    conn = boto3.client('elasticbeanstalk', region_name='us-east-1')
+    app = conn.create_application(
+        ApplicationName="myapp",
+    )
+    env = conn.create_environment(
+        ApplicationName="myapp",
+        EnvironmentName="myenv",
+    )
+    env['EnvironmentName'].should.equal("myenv")
+
+
+@mock_eb
+def test_describe_environments():
+    # List Elastic Beanstalk Envs
+    conn = boto3.client('elasticbeanstalk', region_name='us-east-1')
+    conn.create_application(
+        ApplicationName="myapp",
+    )
+    conn.create_environment(
+        ApplicationName="myapp",
+        EnvironmentName="myenv",
+    )
+
+    envs = conn.describe_environments()
+    envs = envs['Environments']
+    len(envs).should.equal(1)
+    envs[0]['ApplicationName'].should.equal('myapp')
+    envs[0]['EnvironmentName'].should.equal('myenv')
+
+
+def tags_dict_to_list(tag_dict):
+    tag_list = []
+    for key, value in tag_dict.items():
+        tag_list.append({'Key': key, 'Value': value})
+    return tag_list
+
+
+def tags_list_to_dict(tag_list):
+    tag_dict = {}
+    for tag in tag_list:
+        tag_dict[tag['Key']] = tag['Value']
+    return tag_dict
+
+
+@mock_eb
+def test_create_environment_tags():
+    conn = boto3.client('elasticbeanstalk', region_name='us-east-1')
+    conn.create_application(
+        ApplicationName="myapp",
+    )
+    env_tags = {'initial key': 'initial value'}
+    env = conn.create_environment(
+        ApplicationName="myapp",
+        EnvironmentName="myenv",
+        Tags=tags_dict_to_list(env_tags),
+    )
+
+    tags = conn.list_tags_for_resource(
+        ResourceArn=env['EnvironmentArn'],
+    )
+    tags['ResourceArn'].should.equal(env['EnvironmentArn'])
+    tags_list_to_dict(tags['ResourceTags']).should.equal(env_tags)
+
+
+@mock_eb
+def test_update_tags():
+    conn = boto3.client('elasticbeanstalk', region_name='us-east-1')
+    conn.create_application(
+        ApplicationName="myapp",
+    )
+    env_tags = {
+        'initial key': 'initial value',
+        'to remove': 'delete me',
+        'to update': 'original',
+    }
+    env = conn.create_environment(
+        ApplicationName="myapp",
+        EnvironmentName="myenv",
+        Tags=tags_dict_to_list(env_tags),
+    )
+
+    extra_env_tags = {
+        'to update': 'new',
+        'extra key': 'extra value',
+    }
+    conn.update_tags_for_resource(
+        ResourceArn=env['EnvironmentArn'],
+        TagsToAdd=tags_dict_to_list(extra_env_tags),
+        TagsToRemove=['to remove'],
+    )
+
+    total_env_tags = env_tags.copy()
+    total_env_tags.update(extra_env_tags)
+    del total_env_tags['to remove']
+
+    tags = conn.list_tags_for_resource(
+        ResourceArn=env['EnvironmentArn'],
+    )
+    tags['ResourceArn'].should.equal(env['EnvironmentArn'])
+    tags_list_to_dict(tags['ResourceTags']).should.equal(total_env_tags)
+
+
+@mock_eb
+def test_list_available_solution_stacks():
+    conn = boto3.client('elasticbeanstalk', region_name='us-east-1')
+    stacks = conn.list_available_solution_stacks()
+    len(stacks['SolutionStacks']).should.be.greater_than(0)
+    len(stacks['SolutionStacks']).should.be.equal(len(stacks['SolutionStackDetails']))