Move exception to dedicated class

This commit is contained in:
gruebel 2019-11-21 22:03:25 +01:00
commit 158db1f5d6
3 changed files with 28 additions and 21 deletions

View file

@ -0,0 +1,15 @@
from __future__ import unicode_literals
import json
from werkzeug.exceptions import BadRequest
class InvalidInputError(BadRequest):
def __init__(self):
super(InvalidInputError, self).__init__()
self.description = json.dumps(
{
"message": "You provided a value that does not match the required pattern.",
"__type": "InvalidInputException",
}
)

View file

@ -8,6 +8,7 @@ from moto.core import BaseBackend, BaseModel
from moto.core.exceptions import RESTError
from moto.core.utils import unix_time
from moto.organizations import utils
from moto.organizations.exceptions import InvalidInputError
class FakeOrganization(BaseModel):
@ -447,10 +448,7 @@ class OrganizationsBackend(BaseBackend):
account = next((a for a in self.accounts if a.id == kwargs["ResourceId"]), None)
if account is None:
raise RESTError(
"InvalidInputException",
"You provided a value that does not match the required pattern.",
)
raise InvalidInputError
new_tags = {tag["Key"]: tag["Value"] for tag in kwargs["Tags"]}
account.tags.update(new_tags)
@ -459,10 +457,7 @@ class OrganizationsBackend(BaseBackend):
account = next((a for a in self.accounts if a.id == kwargs["ResourceId"]), None)
if account is None:
raise RESTError(
"InvalidInputException",
"You provided a value that does not match the required pattern.",
)
raise InvalidInputError
tags = [{"Key": key, "Value": value} for key, value in account.tags.items()]
return dict(Tags=tags)
@ -471,10 +466,7 @@ class OrganizationsBackend(BaseBackend):
account = next((a for a in self.accounts if a.id == kwargs["ResourceId"]), None)
if account is None:
raise RESTError(
"InvalidInputException",
"You provided a value that does not match the required pattern.",
)
raise InvalidInputError
for key in kwargs["TagKeys"]:
account.tags.pop(key, None)