Skip to content

Commit

Permalink
Greengrass Implement get_resource_definition_version
Browse files Browse the repository at this point in the history
  • Loading branch information
cm-iwata committed Jun 17, 2022
1 parent 34b3bc5 commit 42dc58c
Show file tree
Hide file tree
Showing 4 changed files with 126 additions and 0 deletions.
19 changes: 19 additions & 0 deletions moto/greengrass/models.py
Expand Up @@ -466,6 +466,25 @@ def list_resource_definition_versions(self, resource_definition_id):

return self.resource_definition_versions[resource_definition_id].values()

def get_resource_definition_version(
self, resource_definition_id, resource_definition_version_id
):

if resource_definition_id not in self.resource_definition_versions:
raise IdNotFoundException("That resources definition does not exist.")

if (
resource_definition_version_id
not in self.resource_definition_versions[resource_definition_id]
):
raise VersionNotFoundException(
f"Version {resource_definition_version_id} of Resource List Definition {resource_definition_id} does not exist."
)

return self.resource_definition_versions[resource_definition_id][
resource_definition_version_id
]

@staticmethod
def _validate_resources(resources):
for resource in resources:
Expand Down
15 changes: 15 additions & 0 deletions moto/greengrass/responses.py
Expand Up @@ -327,6 +327,21 @@ def list_resource_definition_versions(self):
),
)

def resource_definition_version(self, request, full_url, headers):
self.setup_class(request, full_url, headers)

if self.method == "GET":
return self.get_resource_definition_version()

def get_resource_definition_version(self):
resource_definition_id = self.path.split("/")[-3]
resource_definition_version_id = self.path.split("/")[-1]
res = self.greengrass_backend.get_resource_definition_version(
resource_definition_id=resource_definition_id,
resource_definition_version_id=resource_definition_version_id,
)
return 200, {"status": 200}, json.dumps(res.to_dict())

def function_definitions(self, request, full_url, headers):
self.setup_class(request, full_url, headers)

Expand Down
1 change: 1 addition & 0 deletions moto/greengrass/urls.py
Expand Up @@ -24,4 +24,5 @@
"{0}/greengrass/definition/resources$": response.resource_definitions,
"{0}/greengrass/definition/resources/(?P<definition_id>[^/]+)/?$": response.resource_definition,
"{0}/greengrass/definition/resources/(?P<definition_id>[^/]+)/versions$": response.resource_definition_versions,
"{0}/greengrass/definition/resources/(?P<definition_id>[^/]+)/versions/(?P<definition_version_id>[^/]+)/?$": response.resource_definition_version,
}
91 changes: 91 additions & 0 deletions tests/test_greengrass/test_greengrass_resource.py
Expand Up @@ -551,3 +551,94 @@ def test_list_resource_definition_versions_with_invalid_id():
"That resources definition does not exist."
)
ex.value.response["Error"]["Code"].should.equal("IdNotFoundException")


@freezegun.freeze_time("2022-06-01 12:00:00")
@mock_greengrass
def test_get_resource_definition_version():

client = boto3.client("greengrass", region_name="ap-northeast-1")
resources = [
{
"Id": "123",
"Name": "test_directory",
"ResourceDataContainer": {
"LocalDeviceResourceData": {
"SourcePath": "/dev/null",
}
},
}
]

initial_version = {"Resources": resources}
create_res = client.create_resource_definition(
InitialVersion=initial_version, Name="TestResource"
)
resource_def_id = create_res["Id"]
resource_def_ver_id = create_res["LatestVersion"]

resource_def_ver_res = client.get_resource_definition_version(
ResourceDefinitionId=resource_def_id,
ResourceDefinitionVersionId=resource_def_ver_id,
)

resource_def_ver_res.should.have.key("Arn")
resource_def_ver_res.should.have.key("CreationTimestamp")
resource_def_ver_res.should.have.key("Definition").should.equal(initial_version)
resource_def_ver_res.should.have.key("Id").equals(resource_def_id)
resource_def_ver_res.should.have.key("Version")
resource_def_ver_res["ResponseMetadata"]["HTTPStatusCode"].should.equal(200)

if not TEST_SERVER_MODE:
resource_def_ver_res["CreationTimestamp"].should.equal(
"2022-06-01T12:00:00.000Z"
)


@mock_greengrass
def test_get_resource_definition_version_with_invalid_id():

client = boto3.client("greengrass", region_name="ap-northeast-1")
with pytest.raises(ClientError) as ex:
client.get_resource_definition_version(
ResourceDefinitionId="fe2392e9-e67f-4308-af1b-ff94a128b231",
ResourceDefinitionVersionId="cd2ea6dc-6634-4e89-8441-8003500435f9",
)
ex.value.response["Error"]["Message"].should.equal(
"That resources definition does not exist."
)
ex.value.response["Error"]["Code"].should.equal("IdNotFoundException")


@mock_greengrass
def test_get_resource_definition_version_with_invalid_version_id():

client = boto3.client("greengrass", region_name="ap-northeast-1")
resources = [
{
"Id": "123",
"Name": "test_directory",
"ResourceDataContainer": {
"LocalDeviceResourceData": {
"SourcePath": "/dev/null",
}
},
}
]

initial_version = {"Resources": resources}
create_res = client.create_resource_definition(
InitialVersion=initial_version, Name="TestResource"
)

resource_def_id = create_res["Id"]
invalid_resource_version_id = "6fbffc21-989e-4d29-a793-a42f450a78c6"
with pytest.raises(ClientError) as ex:
client.get_resource_definition_version(
ResourceDefinitionId=resource_def_id,
ResourceDefinitionVersionId=invalid_resource_version_id,
)
ex.value.response["Error"]["Message"].should.equal(
f"Version {invalid_resource_version_id} of Resource List Definition {resource_def_id} does not exist."
)
ex.value.response["Error"]["Code"].should.equal("VersionNotFoundException")

0 comments on commit 42dc58c

Please sign in to comment.