Add SWF endpoints: RegisterDomain, DeprecateDomain, ListDomains, DescribeDomain

moto/swf/responses.py

@@ -0,0 +1,103 @@
+import json
+import logging
+import six
+
+from moto.core.responses import BaseResponse
+from werkzeug.exceptions import HTTPException
+from moto.core.utils import camelcase_to_underscores, method_names_from_class
+
+from .models import swf_backends
+
+
+class SWFResponse(BaseResponse):
+
+    @property
+    def swf_backend(self):
+        return swf_backends[self.region]
+
+    # SWF actions are not dispatched via URLs but via a specific header called
+    # "x-amz-target", in the form of com.amazonaws.swf.service.model.SimpleWorkflowService.<action>
+    # This is not supported directly in BaseResponse sor for now we override
+    # the call_action() method
+    # See: http://docs.aws.amazon.com/amazonswf/latest/developerguide/UsingJSON-swf.html
+    def call_action(self):
+        headers = self.response_headers
+        # Headers are case-insensitive. Probably a better way to do this.
+        match = self.headers.get('x-amz-target') or self.headers.get('X-Amz-Target')
+        if match:
+            # TODO: see if we can call "[-1]" in BaseResponse, which would
+            # allow to remove that
+            action = match.split(".")[-1]
+
+        action = camelcase_to_underscores(action)
+        method_names = method_names_from_class(self.__class__)
+        if action in method_names:
+            method = getattr(self, action)
+            try:
+                response = method()
+            except HTTPException as http_error:
+                response = http_error.description, dict(status=http_error.code)
+            if isinstance(response, six.string_types):
+                return 200, headers, response
+            else:
+                body, new_headers = response
+                status = new_headers.get('status', 200)
+                headers.update(new_headers)
+                return status, headers, body
+        raise NotImplementedError("The {0} action has not been implemented".format(action))
+
+    # SWF parameters are passed through a JSON body, so let's ease retrieval
+    @property
+    def _params(self):
+        return json.loads(self.body)
+
+    def list_domains(self):
+        status = self._params.get("registrationStatus")
+        domains = self.swf_backend.list_domains(status)
+        template = self.response_template(LIST_DOMAINS_TEMPLATE)
+        return template.render(domains=domains)
+
+    def register_domain(self):
+        name = self._params.get("name")
+        description = self._params.get("description")
+        retention = self._params.get("workflowExecutionRetentionPeriodInDays")
+        domain = self.swf_backend.register_domain(name, retention,
+                                                  description=description)
+        template = self.response_template("")
+        return template.render()
+
+    def deprecate_domain(self):
+        name = self._params.get("name")
+        domain = self.swf_backend.deprecate_domain(name)
+        template = self.response_template("")
+        return template.render()
+
+    def describe_domain(self):
+        name = self._params.get("name")
+        domain = self.swf_backend.describe_domain(name)
+        template = self.response_template(DESCRIBE_DOMAIN_TEMPLATE)
+        return template.render(domain=domain)
+
+
+LIST_DOMAINS_TEMPLATE = """{
+    "domainInfos": [
+        {% for domain in domains %}
+        {
+            "description": "{{ domain.description }}",
+            "name": "{{ domain.name }}",
+            "status": "{{ domain.status }}"
+        }
+        {% endfor %}
+    ]
+}"""
+
+DESCRIBE_DOMAIN_TEMPLATE = """{
+    "configuration": {
+        "workflowExecutionRetentionPeriodInDays": "{{ domain.retention }}"
+    },
+    "domainInfo": {
+        "description": "{{ domain.description }}",
+        "name": "{{ domain.name }}",
+        "status": "{{ domain.status }}"
+    }
+}"""