Merge pull request #2729 from grodzik/add-update-organizational-unit-endpoint

Add update organizational unit endpoint

moto/organizations/exceptions.py

@@ -10,3 +10,13 @@ class InvalidInputException(JsonRESTError):
             "InvalidInputException",
             "You provided a value that does not match the required pattern.",
         )
+
+
+class DuplicateOrganizationalUnitException(JsonRESTError):
+    code = 400
+
+    def __init__(self):
+        super(DuplicateOrganizationalUnitException, self).__init__(
+            "DuplicateOrganizationalUnitException",
+            "An OU with the same name already exists.",
+        )

moto/organizations/models.py

@@ -8,7 +8,10 @@ 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 InvalidInputException
+from moto.organizations.exceptions import (
+    InvalidInputException,
+    DuplicateOrganizationalUnitException,
+)
 
 
 class FakeOrganization(BaseModel):
@@ -222,6 +225,14 @@ class OrganizationsBackend(BaseBackend):
         self.attach_policy(PolicyId=utils.DEFAULT_POLICY_ID, TargetId=new_ou.id)
         return new_ou.describe()
 
+    def update_organizational_unit(self, **kwargs):
+        for ou in self.ou:
+            if ou.name == kwargs["Name"]:
+                raise DuplicateOrganizationalUnitException
+        ou = self.get_organizational_unit_by_id(kwargs["OrganizationalUnitId"])
+        ou.name = kwargs["Name"]
+        return ou.describe()
+
     def get_organizational_unit_by_id(self, ou_id):
         ou = next((ou for ou in self.ou if ou.id == ou_id), None)
         if ou is None:

moto/organizations/responses.py

@@ -36,6 +36,11 @@ class OrganizationsResponse(BaseResponse):
             self.organizations_backend.create_organizational_unit(**self.request_params)
         )
 
+    def update_organizational_unit(self):
+        return json.dumps(
+            self.organizations_backend.update_organizational_unit(**self.request_params)
+        )
+
     def describe_organizational_unit(self):
         return json.dumps(
             self.organizations_backend.describe_organizational_unit(