Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

LakeFormation: deregister_resource() should handle unknown resources (#7023)

Browse Source
This commit is contained in:
Bert Blommers 2023-11-13 22:21:09 -01:00 committed by GitHub
parent c14df137d5
commit d3efa2afb9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
moto/lakeformation/models.py
View File

@ -56,6 +56,8 @@ class LakeFormationBackend(BaseBackend):
return self.resources[resource_arn] return self.resources[resource_arn]
def deregister_resource(self, resource_arn: str) -> None: def deregister_resource(self, resource_arn: str) -> None:
if resource_arn not in self.resources:
raise EntityNotFound
del self.resources[resource_arn] del self.resources[resource_arn]
def register_resource(self, resource_arn: str, role_arn: str) -> None: def register_resource(self, resource_arn: str, role_arn: str) -> None:

5
tests/test_lakeformation/test_lakeformation.py
View File

@ -41,6 +41,11 @@ def test_deregister_resource():
err = exc.value.response["Error"] err = exc.value.response["Error"]
assert err["Code"] == "EntityNotFoundException" assert err["Code"] == "EntityNotFoundException"
with pytest.raises(ClientError) as exc:
client.deregister_resource(ResourceArn="some arn")
err = exc.value.response["Error"]
assert err["Code"] == "EntityNotFoundException"
@mock_lakeformation @mock_lakeformation
def test_list_resources(): def test_list_resources():