111 lines
3.1 KiB
Python
111 lines
3.1 KiB
Python
import boto3
|
|
from freezegun import freeze_time
|
|
import pytest
|
|
|
|
from moto import mock_opsworks
|
|
|
|
|
|
@freeze_time("2015-01-01")
|
|
@mock_opsworks
|
|
def test_create_layer_response():
|
|
client = boto3.client("opsworks", region_name="us-east-1")
|
|
stack_id = client.create_stack(
|
|
Name="test_stack_1",
|
|
Region="us-east-1",
|
|
ServiceRoleArn="service_arn",
|
|
DefaultInstanceProfileArn="profile_arn",
|
|
)["StackId"]
|
|
|
|
response = client.create_layer(
|
|
StackId=stack_id,
|
|
Type="custom",
|
|
Name="TestLayer",
|
|
Shortname="TestLayerShortName",
|
|
)
|
|
|
|
assert "LayerId" in response
|
|
|
|
second_stack_id = client.create_stack(
|
|
Name="test_stack_2",
|
|
Region="us-east-1",
|
|
ServiceRoleArn="service_arn",
|
|
DefaultInstanceProfileArn="profile_arn",
|
|
)["StackId"]
|
|
|
|
response = client.create_layer(
|
|
StackId=second_stack_id,
|
|
Type="custom",
|
|
Name="TestLayer",
|
|
Shortname="TestLayerShortName",
|
|
)
|
|
|
|
assert "LayerId" in response
|
|
|
|
# ClientError
|
|
with pytest.raises(Exception) as exc:
|
|
client.create_layer(
|
|
StackId=stack_id, Type="custom", Name="TestLayer", Shortname="_"
|
|
)
|
|
assert (
|
|
r'already a layer named "TestLayer"' in exc.value.response["Error"]["Message"]
|
|
)
|
|
|
|
# ClientError
|
|
with pytest.raises(Exception) as exc:
|
|
client.create_layer(
|
|
StackId=stack_id, Type="custom", Name="_", Shortname="TestLayerShortName"
|
|
)
|
|
assert (
|
|
r'already a layer with shortname "TestLayerShortName"'
|
|
) in exc.value.response["Error"]["Message"]
|
|
|
|
# ClientError
|
|
with pytest.raises(Exception) as exc:
|
|
client.create_layer(
|
|
StackId="nothere", Type="custom", Name="TestLayer", Shortname="_"
|
|
)
|
|
assert exc.value.response["Error"]["Message"] == "nothere"
|
|
|
|
|
|
@freeze_time("2015-01-01")
|
|
@mock_opsworks
|
|
def test_describe_layers():
|
|
client = boto3.client("opsworks", region_name="us-east-1")
|
|
stack_id = client.create_stack(
|
|
Name="test_stack_1",
|
|
Region="us-east-1",
|
|
ServiceRoleArn="service_arn",
|
|
DefaultInstanceProfileArn="profile_arn",
|
|
)["StackId"]
|
|
layer_id = client.create_layer(
|
|
StackId=stack_id,
|
|
Type="custom",
|
|
Name="TestLayer",
|
|
Shortname="TestLayerShortName",
|
|
)["LayerId"]
|
|
|
|
rv1 = client.describe_layers(StackId=stack_id)
|
|
rv2 = client.describe_layers(LayerIds=[layer_id])
|
|
assert rv1["Layers"] == rv2["Layers"]
|
|
|
|
assert rv1["Layers"][0]["Name"] == "TestLayer"
|
|
|
|
# ClientError
|
|
with pytest.raises(Exception) as exc:
|
|
client.describe_layers(StackId=stack_id, LayerIds=[layer_id])
|
|
assert exc.value.response["Error"]["Message"] == (
|
|
"Please provide one or more layer IDs or a stack ID"
|
|
)
|
|
|
|
# ClientError
|
|
with pytest.raises(Exception) as exc:
|
|
client.describe_layers(StackId="nothere")
|
|
assert exc.value.response["Error"]["Message"] == (
|
|
"Unable to find stack with ID nothere"
|
|
)
|
|
|
|
# ClientError
|
|
with pytest.raises(Exception) as exc:
|
|
client.describe_layers(LayerIds=["nothere"])
|
|
assert exc.value.response["Error"]["Message"] == "nothere"
|