moto/moto/cognitoidentity/responses.py

from __future__ import unicode_literals

from moto.core.responses import BaseResponse
from .models import cognitoidentity_backends
from .utils import get_random_identity_id


class CognitoIdentityResponse(BaseResponse):

    def create_identity_pool(self):
        identity_pool_name = self._get_param('IdentityPoolName')
        allow_unauthenticated_identities = self._get_param('AllowUnauthenticatedIdentities')
        supported_login_providers = self._get_param('SupportedLoginProviders')
        developer_provider_name = self._get_param('DeveloperProviderName')
        open_id_connect_provider_arns = self._get_param('OpenIdConnectProviderARNs')
        cognito_identity_providers = self._get_param('CognitoIdentityProviders')
        saml_provider_arns = self._get_param('SamlProviderARNs')

        return cognitoidentity_backends[self.region].create_identity_pool(
            identity_pool_name=identity_pool_name,
            allow_unauthenticated_identities=allow_unauthenticated_identities,
            supported_login_providers=supported_login_providers,
            developer_provider_name=developer_provider_name,
            open_id_connect_provider_arns=open_id_connect_provider_arns,
            cognito_identity_providers=cognito_identity_providers,
            saml_provider_arns=saml_provider_arns)

    def get_id(self):
        return cognitoidentity_backends[self.region].get_id()

    def describe_identity_pool(self):
        return cognitoidentity_backends[self.region].describe_identity_pool(self._get_param('IdentityPoolId'))

    def get_credentials_for_identity(self):
        return cognitoidentity_backends[self.region].get_credentials_for_identity(self._get_param('IdentityId'))

    def get_open_id_token_for_developer_identity(self):
        return cognitoidentity_backends[self.region].get_open_id_token_for_developer_identity(
            self._get_param('IdentityId') or get_random_identity_id(self.region)
        )

    def get_open_id_token(self):
        return cognitoidentity_backends[self.region].get_open_id_token(
            self._get_param("IdentityId") or get_random_identity_id(self.region)
        )