"""Lambda Function Error Response."""


class LambdaFunctionErrorState:
    """Stores the state of a function error."""

    def __init__(self, response, payload=None):
        """Set defaults and parse the lambda and function responses."""
        if not payload:
            payload = {}
        self.trace = payload.get('stackTrace')
        self.handler = response.get('FunctionError')
        self.type = payload.get('errorType')
        self.message = payload.get('errorMessage')

    def dump(self):
        """Return function error state as a dict for testing and debugging."""
        return {
            'message': self.message,
            'trace': self.trace,
            'handler': self.handler,
            'type': self.type
        }