526 lines
18 KiB
Python
526 lines
18 KiB
Python
import boto3
|
|
import freezegun
|
|
import pytest
|
|
from botocore.exceptions import ClientError
|
|
|
|
from moto import mock_aws
|
|
from moto.core import DEFAULT_ACCOUNT_ID as ACCOUNT_ID
|
|
from moto.settings import TEST_SERVER_MODE
|
|
|
|
|
|
@freezegun.freeze_time("2022-06-01 12:00:00")
|
|
@mock_aws
|
|
def test_create_group():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
init_core_ver = {
|
|
"Cores": [
|
|
{
|
|
"CertificateArn": f"arn:aws:iot:ap-northeast-1:{ACCOUNT_ID}:cert/36ed61be9c6271ae8da174e29d0e033c06af149d7b21672f3800fe322044554d",
|
|
"Id": "123456789",
|
|
"ThingArn": f"arn:aws:iot:ap-northeast-1:{ACCOUNT_ID}:thing/CoreThing",
|
|
}
|
|
]
|
|
}
|
|
|
|
create_core_def_res = client.create_core_definition(
|
|
InitialVersion=init_core_ver, Name="TestCore"
|
|
)
|
|
core_def_ver_arn = create_core_def_res["LatestVersionArn"]
|
|
|
|
init_grp = {"CoreDefinitionVersionArn": core_def_ver_arn}
|
|
|
|
grp_name = "TestGroup"
|
|
create_grp_res = client.create_group(Name=grp_name, InitialVersion=init_grp)
|
|
assert "Arn" in create_grp_res
|
|
assert "Id" in create_grp_res
|
|
assert "LastUpdatedTimestamp" in create_grp_res
|
|
assert "LatestVersion" in create_grp_res
|
|
assert "LatestVersionArn" in create_grp_res
|
|
assert create_grp_res["Name"] == grp_name
|
|
assert create_grp_res["ResponseMetadata"]["HTTPStatusCode"] == 201
|
|
|
|
if not TEST_SERVER_MODE:
|
|
assert create_grp_res["CreationTimestamp"] == "2022-06-01T12:00:00.000Z"
|
|
assert create_grp_res["LastUpdatedTimestamp"] == "2022-06-01T12:00:00.000Z"
|
|
|
|
|
|
@freezegun.freeze_time("2022-06-01 12:00:00")
|
|
@mock_aws
|
|
def test_list_groups():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
grp_name = "TestGroup"
|
|
create_grp_res = client.create_group(Name=grp_name)
|
|
group_id = create_grp_res["Id"]
|
|
|
|
list_res = client.list_groups()
|
|
|
|
group = list_res["Groups"][0]
|
|
assert group["Name"] == grp_name
|
|
assert "Arn" in group
|
|
assert group["Id"] == group_id
|
|
assert "LatestVersion" in group
|
|
assert "LatestVersionArn" in group
|
|
if not TEST_SERVER_MODE:
|
|
assert group["CreationTimestamp"] == "2022-06-01T12:00:00.000Z"
|
|
assert group["LastUpdatedTimestamp"] == "2022-06-01T12:00:00.000Z"
|
|
|
|
|
|
@freezegun.freeze_time("2022-06-01 12:00:00")
|
|
@mock_aws
|
|
def test_get_group():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
grp_name = "TestGroup"
|
|
create_res = client.create_group(Name=grp_name)
|
|
group_id = create_res["Id"]
|
|
group_arn = create_res["Arn"]
|
|
latest_version = create_res["LatestVersion"]
|
|
latest_version_arn = create_res["LatestVersionArn"]
|
|
|
|
get_res = client.get_group(GroupId=group_id)
|
|
|
|
assert get_res["Name"] == grp_name
|
|
assert get_res["Arn"] == group_arn
|
|
assert get_res["Id"] == group_id
|
|
assert get_res["LatestVersion"] == latest_version
|
|
assert get_res["LatestVersionArn"] == latest_version_arn
|
|
assert get_res["ResponseMetadata"]["HTTPStatusCode"] == 200
|
|
|
|
if not TEST_SERVER_MODE:
|
|
assert get_res["CreationTimestamp"] == "2022-06-01T12:00:00.000Z"
|
|
assert get_res["LastUpdatedTimestamp"] == "2022-06-01T12:00:00.000Z"
|
|
|
|
|
|
@mock_aws
|
|
def test_get_group_with_invalid_id():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
with pytest.raises(ClientError) as ex:
|
|
client.get_group(GroupId="b552443b-1888-469b-81f8-0ebc5ca92949")
|
|
err = ex.value.response["Error"]
|
|
assert err["Message"] == "That Group Definition does not exist."
|
|
assert err["Code"] == "IdNotFoundException"
|
|
|
|
|
|
@mock_aws
|
|
def test_delete_group():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
create_res = client.create_group(Name="TestGroup")
|
|
|
|
group_id = create_res["Id"]
|
|
del_res = client.delete_group(GroupId=group_id)
|
|
assert del_res["ResponseMetadata"]["HTTPStatusCode"] == 200
|
|
|
|
|
|
@mock_aws
|
|
def test_delete_group_with_invalid_id():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
|
|
with pytest.raises(ClientError) as ex:
|
|
client.delete_group(GroupId="6fbffc21-989e-4d29-a793-a42f450a78c6")
|
|
err = ex.value.response["Error"]
|
|
assert err["Message"] == "That group definition does not exist."
|
|
assert err["Code"] == "IdNotFoundException"
|
|
|
|
|
|
@mock_aws
|
|
def test_update_group():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
create_res = client.create_group(Name="TestGroup")
|
|
|
|
group_id = create_res["Id"]
|
|
updated_group_name = "UpdatedGroup"
|
|
update_res = client.update_group(GroupId=group_id, Name=updated_group_name)
|
|
assert update_res["ResponseMetadata"]["HTTPStatusCode"] == 200
|
|
|
|
get_res = client.get_group(GroupId=group_id)
|
|
assert get_res["Name"] == updated_group_name
|
|
|
|
|
|
@mock_aws
|
|
def test_update_group_with_empty_name():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
create_res = client.create_group(Name="TestGroup")
|
|
|
|
group_id = create_res["Id"]
|
|
|
|
with pytest.raises(ClientError) as ex:
|
|
client.update_group(GroupId=group_id, Name="")
|
|
err = ex.value.response["Error"]
|
|
assert err["Message"] == "Input does not contain any attributes to be updated"
|
|
assert err["Code"] == "InvalidContainerDefinitionException"
|
|
|
|
|
|
@mock_aws
|
|
def test_update_group_with_invalid_id():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
|
|
with pytest.raises(ClientError) as ex:
|
|
client.update_group(GroupId="6fbffc21-989e-4d29-a793-a42f450a78c6", Name="123")
|
|
err = ex.value.response["Error"]
|
|
assert err["Message"] == "That group definition does not exist."
|
|
assert err["Code"] == "IdNotFoundException"
|
|
|
|
|
|
@freezegun.freeze_time("2022-06-01 12:00:00")
|
|
@mock_aws
|
|
def test_create_group_version():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
create_grp_res = client.create_group(Name="TestGroup")
|
|
group_id = create_grp_res["Id"]
|
|
|
|
group_ver_res = client.create_group_version(GroupId=group_id)
|
|
assert "Arn" in group_ver_res
|
|
assert "CreationTimestamp" in group_ver_res
|
|
assert group_ver_res["Id"] == group_id
|
|
assert "Version" in group_ver_res
|
|
|
|
if not TEST_SERVER_MODE:
|
|
assert group_ver_res["CreationTimestamp"] == "2022-06-01T12:00:00.000Z"
|
|
|
|
|
|
@mock_aws
|
|
def test_create_group_version_with_invalid_id():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
with pytest.raises(ClientError) as ex:
|
|
client.create_group_version(GroupId="cd2ea6dc-6634-4e89-8441-8003500435f9")
|
|
|
|
err = ex.value.response["Error"]
|
|
assert err["Message"] == "That group does not exist."
|
|
assert err["Code"] == "IdNotFoundException"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"def_ver_key_name,arn,error_message",
|
|
[
|
|
(
|
|
"CoreDefinitionVersionArn",
|
|
"123",
|
|
"Cores definition reference does not exist",
|
|
),
|
|
(
|
|
"CoreDefinitionVersionArn",
|
|
"arn:aws:greengrass:ap-northeast-1:944137583148:/greengrass/definition/cores/fc3b3e5b-f1ce-4639-88d3-3ad897d95b2a/versions/dd0f800c-246c-4973-82cf-45b109cbd99b",
|
|
"Cores definition reference does not exist",
|
|
),
|
|
(
|
|
"DeviceDefinitionVersionArn",
|
|
"123",
|
|
"Devices definition reference does not exist",
|
|
),
|
|
(
|
|
"DeviceDefinitionVersionArn",
|
|
"arn:aws:greengrass:ap-northeast-1:944137583148:/greengrass/definition/devices/fc3b3e5b-f1ce-4639-88d3-3ad897d95b2a/versions/dd0f800c-246c-4973-82cf-45b109cbd99b",
|
|
"Devices definition reference does not exist",
|
|
),
|
|
(
|
|
"FunctionDefinitionVersionArn",
|
|
"123",
|
|
"Lambda definition reference does not exist",
|
|
),
|
|
(
|
|
"FunctionDefinitionVersionArn",
|
|
"arn:aws:greengrass:ap-northeast-1:944137583148:/greengrass/definition/functions/fc3b3e5b-f1ce-4639-88d3-3ad897d95b2a/versions/dd0f800c-246c-4973-82cf-45b109cbd99b",
|
|
"Lambda definition reference does not exist",
|
|
),
|
|
(
|
|
"ResourceDefinitionVersionArn",
|
|
"123",
|
|
"Resource definition reference does not exist",
|
|
),
|
|
(
|
|
"ResourceDefinitionVersionArn",
|
|
"arn:aws:greengrass:ap-northeast-1:944137583148:/greengrass/definition/resources/fc3b3e5b-f1ce-4639-88d3-3ad897d95b2a/versions/dd0f800c-246c-4973-82cf-45b109cbd99b",
|
|
"Resource definition reference does not exist",
|
|
),
|
|
(
|
|
"SubscriptionDefinitionVersionArn",
|
|
"123",
|
|
"Subscription definition reference does not exist",
|
|
),
|
|
(
|
|
"SubscriptionDefinitionVersionArn",
|
|
"arn:aws:greengrass:ap-northeast-1:944137583148:/greengrass/definition/subscriptions/fc3b3e5b-f1ce-4639-88d3-3ad897d95b2a/versions/dd0f800c-246c-4973-82cf-45b109cbd99b",
|
|
"Subscription definition reference does not exist",
|
|
),
|
|
],
|
|
)
|
|
@mock_aws
|
|
def test_create_group_version_with_invalid_version_arn(
|
|
def_ver_key_name, arn, error_message
|
|
):
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
create_grp_res = client.create_group(Name="TestGroup")
|
|
group_id = create_grp_res["Id"]
|
|
|
|
definition_versions = {def_ver_key_name: arn}
|
|
|
|
with pytest.raises(ClientError) as ex:
|
|
client.create_group_version(GroupId=group_id, **definition_versions)
|
|
|
|
err = ex.value.response["Error"]
|
|
assert (
|
|
err["Message"]
|
|
== f"The group is invalid or corrupted. (ErrorDetails: [{error_message}])"
|
|
)
|
|
assert err["Code"] == "400"
|
|
|
|
|
|
@freezegun.freeze_time("2022-06-01 12:00:00")
|
|
@mock_aws
|
|
def test_list_group_versions():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
create_res = client.create_group(Name="TestGroup")
|
|
group_id = create_res["Id"]
|
|
|
|
group_vers_res = client.list_group_versions(GroupId=group_id)
|
|
|
|
assert "Versions" in group_vers_res
|
|
group_ver = group_vers_res["Versions"][0]
|
|
assert "Arn" in group_ver
|
|
assert "CreationTimestamp" in group_ver
|
|
assert group_ver["Id"] == group_id
|
|
assert "Version" in group_ver
|
|
|
|
if not TEST_SERVER_MODE:
|
|
assert group_ver["CreationTimestamp"] == "2022-06-01T12:00:00.000Z"
|
|
|
|
|
|
@mock_aws
|
|
def test_list_group_versions_with_invalid_id():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
|
|
with pytest.raises(ClientError) as ex:
|
|
client.list_group_versions(GroupId="7b0bdeae-54c7-47cf-9f93-561e672efd9c")
|
|
err = ex.value.response["Error"]
|
|
assert err["Message"] == "That group definition does not exist."
|
|
assert err["Code"] == "IdNotFoundException"
|
|
|
|
|
|
@freezegun.freeze_time("2022-06-01 12:00:00")
|
|
@mock_aws
|
|
def test_get_group_version():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
init_core_ver = {
|
|
"Cores": [
|
|
{
|
|
"CertificateArn": f"arn:aws:iot:ap-northeast-1:{ACCOUNT_ID}:cert/36ed61be9c6271ae8da174e29d0e033c06af149d7b21672f3800fe322044554d",
|
|
"Id": "123456789",
|
|
"ThingArn": f"arn:aws:iot:ap-northeast-1:{ACCOUNT_ID}:thing/CoreThing",
|
|
}
|
|
]
|
|
}
|
|
|
|
create_core_def_res = client.create_core_definition(
|
|
InitialVersion=init_core_ver, Name="TestCore"
|
|
)
|
|
core_def_ver_arn = create_core_def_res["LatestVersionArn"]
|
|
|
|
init_device_ver = {
|
|
"Devices": [
|
|
{
|
|
"CertificateArn": f"arn:aws:iot:ap-northeast-1:{ACCOUNT_ID}:cert/36ed61be9c6271ae8da174e29d0e033c06af149d7b21672f3800fe322044554d",
|
|
"Id": "123",
|
|
"SyncShadow": True,
|
|
"ThingArn": f"arn:aws:iot:ap-northeast-1:{ACCOUNT_ID}:thing/CoreThing",
|
|
}
|
|
]
|
|
}
|
|
|
|
create_device_res = client.create_device_definition(
|
|
InitialVersion=init_device_ver, Name="TestDevice"
|
|
)
|
|
device_def_ver_arn = create_device_res["LatestVersionArn"]
|
|
|
|
init_func_ver = {
|
|
"Functions": [
|
|
{
|
|
"FunctionArn": "arn:aws:lambda:ap-northeast-1:123456789012:function:test-func:1",
|
|
"Id": "1234567890",
|
|
"FunctionConfiguration": {
|
|
"MemorySize": 16384,
|
|
"EncodingType": "binary",
|
|
"Pinned": True,
|
|
"Timeout": 3,
|
|
},
|
|
}
|
|
]
|
|
}
|
|
create_func_res = client.create_function_definition(
|
|
InitialVersion=init_func_ver, Name="TestFunc"
|
|
)
|
|
func_def_ver_arn = create_func_res["LatestVersionArn"]
|
|
|
|
init_resource_ver = {
|
|
"Resources": [
|
|
{
|
|
"Id": "123",
|
|
"Name": "test_directory",
|
|
"ResourceDataContainer": {
|
|
"LocalVolumeResourceData": {
|
|
"DestinationPath": "/test_dir",
|
|
"GroupOwnerSetting": {"AutoAddGroupOwner": True},
|
|
"SourcePath": "/home/ggc_user/test_dir",
|
|
}
|
|
},
|
|
}
|
|
]
|
|
}
|
|
|
|
create_resource_res = client.create_resource_definition(
|
|
InitialVersion=init_resource_ver, Name="TestResource"
|
|
)
|
|
resource_def_ver_arn = create_resource_res["LatestVersionArn"]
|
|
|
|
init_subscription_ver = {
|
|
"Subscriptions": [
|
|
{
|
|
"Id": "123456",
|
|
"Source": "arn:aws:lambda:ap-northeast-1:123456789012:function:test_func:1",
|
|
"Subject": "foo/bar",
|
|
"Target": "cloud",
|
|
}
|
|
]
|
|
}
|
|
create_subscription_res = client.create_subscription_definition(
|
|
InitialVersion=init_subscription_ver, Name="TestSubscription"
|
|
)
|
|
subscription_def_ver_arn = create_subscription_res["LatestVersionArn"]
|
|
|
|
definition = {
|
|
"CoreDefinitionVersionArn": core_def_ver_arn,
|
|
"DeviceDefinitionVersionArn": device_def_ver_arn,
|
|
"FunctionDefinitionVersionArn": func_def_ver_arn,
|
|
"ResourceDefinitionVersionArn": resource_def_ver_arn,
|
|
"SubscriptionDefinitionVersionArn": subscription_def_ver_arn,
|
|
}
|
|
|
|
grp_name = "TestGroup"
|
|
create_res = client.create_group(Name=grp_name, InitialVersion=definition)
|
|
group_id = create_res["Id"]
|
|
group_ver_id = create_res["LatestVersion"]
|
|
|
|
group_ver_res = client.get_group_version(
|
|
GroupId=group_id, GroupVersionId=group_ver_id
|
|
)
|
|
|
|
assert "Arn" in group_ver_res
|
|
assert "CreationTimestamp" in group_ver_res
|
|
assert group_ver_res["Definition"] == definition
|
|
assert group_ver_res["Id"] == group_id
|
|
assert group_ver_res["Version"] == group_ver_id
|
|
|
|
if not TEST_SERVER_MODE:
|
|
assert group_ver_res["CreationTimestamp"] == "2022-06-01T12:00:00.000Z"
|
|
|
|
|
|
@mock_aws
|
|
def test_get_group_version_with_invalid_id():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
|
|
with pytest.raises(ClientError) as ex:
|
|
client.get_group_version(
|
|
GroupId="7b0bdeae-54c7-47cf-9f93-561e672efd9c",
|
|
GroupVersionId="7b0bdeae-54c7-47cf-9f93-561e672efd9c",
|
|
)
|
|
err = ex.value.response["Error"]
|
|
assert err["Message"] == "That group definition does not exist."
|
|
assert err["Code"] == "IdNotFoundException"
|
|
|
|
|
|
@mock_aws
|
|
def test_get_group_version_with_invalid_version_id():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
create_res = client.create_group(Name="TestGroup")
|
|
|
|
group_id = create_res["Id"]
|
|
invalid_group_ver_id = "7b0bdeae-54c7-47cf-9f93-561e672efd9c"
|
|
|
|
with pytest.raises(ClientError) as ex:
|
|
client.get_group_version(
|
|
GroupId=group_id,
|
|
GroupVersionId=invalid_group_ver_id,
|
|
)
|
|
err = ex.value.response["Error"]
|
|
assert (
|
|
err["Message"]
|
|
== f"Version {invalid_group_ver_id} of Group Definition {group_id} does not exist."
|
|
)
|
|
assert err["Code"] == "VersionNotFoundException"
|
|
|
|
|
|
@freezegun.freeze_time("2022-06-01 12:00:00")
|
|
@mock_aws
|
|
def test_associate_role_to_group():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
res = client.associate_role_to_group(
|
|
GroupId="abc002c8-1093-485e-9324-3baadf38e582",
|
|
RoleArn=f"arn:aws:iam::{ACCOUNT_ID}:role/greengrass-role",
|
|
)
|
|
|
|
assert "AssociatedAt" in res
|
|
assert res["ResponseMetadata"]["HTTPStatusCode"] == 200
|
|
|
|
|
|
@freezegun.freeze_time("2022-06-01 12:00:00")
|
|
@mock_aws
|
|
def test_get_associated_role():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
group_id = "abc002c8-1093-485e-9324-3baadf38e582"
|
|
role_arn = f"arn:aws:iam::{ACCOUNT_ID}:role/greengrass-role"
|
|
client.associate_role_to_group(GroupId=group_id, RoleArn=role_arn)
|
|
|
|
res = client.get_associated_role(GroupId=group_id)
|
|
assert "AssociatedAt" in res
|
|
assert res["RoleArn"] == role_arn
|
|
assert res["ResponseMetadata"]["HTTPStatusCode"] == 200
|
|
|
|
if not TEST_SERVER_MODE:
|
|
assert res["AssociatedAt"] == "2022-06-01T12:00:00.000Z"
|
|
|
|
|
|
@mock_aws
|
|
def test_get_associated_role_with_invalid_id():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
with pytest.raises(ClientError) as ex:
|
|
client.get_associated_role(GroupId="abc002c8-1093-485e-9324-3baadf38e582")
|
|
|
|
err = ex.value.response["Error"]
|
|
assert err["Message"] == "You need to attach an IAM role to this deployment group."
|
|
assert err["Code"] == "404"
|
|
|
|
|
|
@freezegun.freeze_time("2022-06-01 12:00:00")
|
|
@mock_aws
|
|
def test_disassociate_role_from_group():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
group_id = "abc002c8-1093-485e-9324-3baadf38e582"
|
|
role_arn = f"arn:aws:iam::{ACCOUNT_ID}:role/greengrass-role"
|
|
client.associate_role_to_group(GroupId=group_id, RoleArn=role_arn)
|
|
client.get_associated_role(GroupId=group_id)
|
|
|
|
res = client.disassociate_role_from_group(GroupId=group_id)
|
|
assert "DisassociatedAt" in res
|
|
assert res["ResponseMetadata"]["HTTPStatusCode"] == 200
|
|
|
|
if not TEST_SERVER_MODE:
|
|
assert res["DisassociatedAt"] == "2022-06-01T12:00:00.000Z"
|
|
|
|
with pytest.raises(ClientError) as ex:
|
|
client.get_associated_role(GroupId=group_id)
|
|
|
|
err = ex.value.response["Error"]
|
|
assert err["Message"] == "You need to attach an IAM role to this deployment group."
|
|
assert err["Code"] == "404"
|
|
|
|
|
|
@freezegun.freeze_time("2022-06-01 12:00:00")
|
|
@mock_aws
|
|
def test_disassociate_role_from_group_with_none_exists_group_id():
|
|
client = boto3.client("greengrass", region_name="ap-northeast-1")
|
|
group_id = "abc002c8-1093-485e-9324-3baadf38e582"
|
|
res = client.disassociate_role_from_group(GroupId=group_id)
|
|
assert "DisassociatedAt" in res
|
|
assert res["ResponseMetadata"]["HTTPStatusCode"] == 200
|
|
|
|
if not TEST_SERVER_MODE:
|
|
assert res["DisassociatedAt"] == "2022-06-01T12:00:00.000Z"
|