#! /usr/bin/env python
# -*- coding: UTF-8 -*-
#
# $Id: idwsf2_tests.py 3254 2007-06-05 21:23:57Z dlaniel $
#
# Python unit tests for Lasso library
#
# Copyright (C) 2004-2007 Entr'ouvert
# http://lasso.entrouvert.org
#
# Authors: See AUTHORS file in top-level directory.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA


import os
import unittest
import sys

if not '..' in sys.path:
    sys.path.insert(0, '..')
if not '../.libs' in sys.path:
    sys.path.insert(0, '../.libs')

import lasso


try:
    dataDir
except NameError:
    dataDir = os.path.join(os.environ['TOP_SRCDIR'], 'tests', 'data')

idpSoapEndpoint = 'http://idp1/soapEndpoint'
spSoapEndpoint = 'http://sp1/soapEndpoint'
spInteractionUrl = 'http://sp1/askMeAQuestion'

class IdWsf2TestCase(unittest.TestCase):
    def getWspServer(self):
        wsp_metadata = os.path.join(dataDir, 'sp5-saml2/metadata.xml')
        wsp_private_key = os.path.join(dataDir, 'sp5-saml2/private-key.pem')
        idp_metadata = os.path.join(dataDir, 'idp5-saml2/metadata.xml')

        server = lasso.Server(wsp_metadata, wsp_private_key, None, None)
        server.addProvider(lasso.PROVIDER_ROLE_IDP, idp_metadata, None, None)
        server.setEncryptionPrivateKey(wsp_private_key);

        return server;

    def getWscServer(self):
        wsc_metadata = os.path.join(dataDir, 'sp6-saml2/metadata.xml')
        wsc_private_key = os.path.join(dataDir, 'sp6-saml2/private-key.pem')
        idp_metadata = os.path.join(dataDir, 'idp5-saml2/metadata.xml')

        server = lasso.Server(wsc_metadata, wsc_private_key, None, None)
        server.addProvider(lasso.PROVIDER_ROLE_IDP, idp_metadata, None, None)

        return server;

    def getIdpServer(self):
        if hasattr(self, 'idp_server_dump') and self.idp_server_dump is not None:
            server = lasso.Server.newFromDump(self.idp_server_dump)
        else:
            idp_metadata = os.path.join(dataDir, 'idp5-saml2/metadata.xml')
            idp_private_key = os.path.join(dataDir, 'idp5-saml2/private-key.pem')
            wsp_metadata = os.path.join(dataDir, 'sp5-saml2/metadata.xml')
            wsc_metadata = os.path.join(dataDir, 'sp6-saml2/metadata.xml')

            server = lasso.Server(idp_metadata, idp_private_key, None, None)
            server.addProvider(lasso.PROVIDER_ROLE_SP, wsp_metadata, None, None)
            server.getProvider(server.providerIds[0]).setEncryptionMode(lasso.ENCRYPTION_MODE_NAMEID);
            server.addProvider(lasso.PROVIDER_ROLE_SP, wsc_metadata, None, None)
            self.idp_server_dump = server.dump()

        return server


    def metadataRegister(self, wsp, idp):
        wsp_disco = lasso.IdWsf2Discovery(wsp)
        abstract = 'Personal Profile service'
        wsp_disco.initMetadataRegister()
        wsp_disco.addSimpleServiceMetadata(
                service_types = ('urn:liberty:id-sis-pp:2005-05',),
                abstract = abstract, provider_id = wsp.providerIds[0],
                address = idpSoapEndpoint)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processRequestMsg(wsp_disco.msgBody)
        idp_disco.validateRequest()
        self.failUnlessEqual(len(idp_disco.metadatas), 1)
        idp_disco.buildResponseMsg()
        wsp_disco.processResponseMsg(idp_disco.msgBody)
        self.failUnlessEqual(len(wsp_disco.metadatas), 1)
        self.failUnlessEqual(wsp_disco.metadatas[0].svcMDID, wsp_disco.response.svcMDID[0])
        return idp, wsp_disco.metadatas[0]

    def login(self, sp, idp, sp_identity_dump=None, sp_session_dump=None,
            idp_identity_dump=None, idp_session_dump=None):
        sp_login = lasso.Login(sp)
        idp_provider_id = 'http://idp5/metadata'
        sp_login.initAuthnRequest(idp_provider_id, lasso.HTTP_METHOD_REDIRECT)
        sp_login.request.nameIDPolicy.format = lasso.SAML2_NAME_IDENTIFIER_FORMAT_PERSISTENT
        sp_login.request.nameIDPolicy.allowCreate = True
        sp_login.buildAuthnRequestMsg()

        idp_login = lasso.Login(idp)
        query = sp_login.msgUrl.split('?')[1]
        if idp_identity_dump is not None:
            idp_login.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_login.setSessionFromDump(idp_session_dump)
        idp_login.processAuthnRequestMsg(query)
        idp_login.validateRequestMsg(True, True)
        idp_login.buildAssertion(lasso.SAML_AUTHENTICATION_METHOD_PASSWORD, None, None, None, None)
        idp_login.idwsf2AddDiscoveryBootstrapEpr(url = idpSoapEndpoint, abstract = 'Discovery Service', security_mech_id = lasso.SECURITY_MECH_BEARER)
        idp_login.buildArtifactMsg(lasso.HTTP_METHOD_ARTIFACT_GET)
        artifact_message = idp_login.artifactMessage

        if idp_login.isIdentityDirty:
            idp_identity_dump = idp_login.identity.dump()
        if idp_login.isSessionDirty:
            idp_session_dump = idp_login.session.dump()

        sp_login = lasso.Login(sp)
        query = idp_login.msgUrl.split('?')[1]
        query = query.replace("%3D", "=")
        sp_login.initRequest(query, lasso.HTTP_METHOD_ARTIFACT_GET)
        sp_login.buildRequestMsg()

        idp_login = lasso.Login(idp)
        idp_login.processRequestMsg(sp_login.msgBody)
        idp_login.artifactMessage = artifact_message
        idp_login.buildResponseMsg(None)

        sp_login.processResponseMsg(idp_login.msgBody)
        sp_login.acceptSso()
        if sp_login.isIdentityDirty:
            sp_identity_dump = sp_login.identity.dump()
        if sp_login.isSessionDirty:
            sp_session_dump = sp_login.session.dump()

        return sp_identity_dump, sp_session_dump, idp_identity_dump, idp_session_dump, sp_login.idwsf2GetDiscoveryBootstrapEpr()


class MetadataTestCase(IdWsf2TestCase):
    def test01(self):
        """Test metadata registration on the IdP"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump, dst_epr = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        wsp_disco.setEpr(dst_epr)

        abstract = 'Personal Profile service'
        wsp_disco.initMetadataRegister()
        self.failUnless(wsp_disco.request is not None)
        wsp_disco.addSimpleServiceMetadata(service_types = 
                (lasso.PP11_HREF,), abstract = abstract,
                provider_id = wsp.providerId, address = spSoapEndpoint,
                security_mech_ids = (lasso.SECURITY_MECH_BEARER,))
        self.failUnlessEqual(len(wsp_disco.metadatas), 1)
        metadata = wsp_disco.metadatas[0]
        self.failUnlessEqual(metadata.abstract, abstract)
        self.failUnlessEqual(metadata.providerId, wsp.providerId)
        self.failUnlessEqual(len(metadata.serviceContext), 1)
        self.failUnlessEqual(len(metadata.serviceContext[0].serviceType), 1)
        self.failUnlessEqual(metadata.serviceContext[0].serviceType[0],
                lasso.PP11_HREF)
        self.failUnlessEqual(len(metadata.serviceContext[0].endpointContext), 1)
        self.failUnlessEqual(
                len(metadata.serviceContext[0].endpointContext[0].address),
                1)
        self.failUnlessEqual(metadata.serviceContext[0].endpointContext[0].address[0],
                spSoapEndpoint)
        self.failUnlessEqual(
                len(metadata.serviceContext[0].endpointContext[0].securityMechId),
                1)
        self.failUnlessEqual(
                metadata.serviceContext[0].endpointContext[0].securityMechId[0],
                lasso.SECURITY_MECH_BEARER)
        self.failUnless(metadata.svcMDID is None)
        wsp_disco.buildRequestMsg()
        self.failUnlessEqual(wsp_disco.msgUrl, idpSoapEndpoint)
        self.failUnless(wsp_disco.msgBody is not None)

        idp_disco = lasso.IdWsf2Discovery(idp)
        self.failUnless(idp_disco is not None)
        idp_disco.processRequestMsg(wsp_disco.msgBody)
        self.failUnless(idp_disco.request is not None)
        self.failUnlessEqual(len(idp_disco.request.svcMD), 1)
        self.failUnless(idp_disco.request.svcMD[0].svcMDID is None)
        try:
            idp_disco.checkSecurityMechanism()
        except lasso.Error, e:
            self.fail(e)
        try:
            idp_disco.validateRequest()
        except lasso.Error, e:
            self.fail(e)
        self.failUnless(idp_disco.response is not None)
        self.failUnlessEqual(len(idp_disco.metadatas), 1)
        metadata = idp_disco.metadatas[0]
        self.failUnlessEqual(metadata.abstract, abstract)
        self.failUnlessEqual(metadata.providerId, wsp.providerId)
        self.failUnlessEqual(len(metadata.serviceContext), 1)
        self.failUnlessEqual(len(metadata.serviceContext[0].serviceType), 1)
        self.failUnlessEqual(metadata.serviceContext[0].serviceType[0],
                lasso.PP11_HREF)
        self.failUnlessEqual(len(metadata.serviceContext[0].endpointContext), 1)
        self.failUnlessEqual(
                len(metadata.serviceContext[0].endpointContext[0].address),
                1)
        self.failUnlessEqual(metadata.serviceContext[0].endpointContext[0].address[0],
                spSoapEndpoint)
        self.failUnlessEqual(
                len(metadata.serviceContext[0].endpointContext[0].securityMechId),
                1)
        self.failUnlessEqual(
                metadata.serviceContext[0].endpointContext[0].securityMechId[0],
                lasso.SECURITY_MECH_BEARER)
        idp_disco.buildResponseMsg()
        self.failUnless(metadata.svcMDID is not None)
        self.failUnless(idp_disco.msgUrl is None)
        self.failUnless(idp_disco.msgBody is not None)

        wsp_disco.processResponseMsg(idp_disco.msgBody)

        self.failUnless(len(wsp_disco.metadatas) == 1, 'missing svcMDID')
        self.failUnless(wsp_disco.metadatas[0].svcMDID, 'missing svcMDID')

    def test02(self):
        "Test failure by IdP for register request"
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump, dst_epr = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        wsp_disco.setEpr(dst_epr)

        abstract = 'Personal Profile service'
        wsp_disco.initMetadataRegister()
        self.failUnless(wsp_disco.request is not None)
        wsp_disco.addSimpleServiceMetadata(service_types = 
                (lasso.PP11_HREF,), abstract = abstract,
                provider_id = wsp.providerId, address = spSoapEndpoint,
                security_mech_ids= (lasso.SECURITY_MECH_BEARER,))
        self.failUnlessEqual(len(wsp_disco.metadatas), 1)
        metadata = wsp_disco.metadatas[0]
        self.failUnlessEqual(metadata.abstract, abstract)
        self.failUnlessEqual(metadata.providerId, wsp.providerId)
        self.failUnlessEqual(len(metadata.serviceContext[0].serviceType), 1)
        self.failUnlessEqual(metadata.serviceContext[0].serviceType[0],
                lasso.PP11_HREF)
        self.failUnlessEqual(len(metadata.serviceContext[0].endpointContext), 1)
        self.failUnlessEqual(
                len(metadata.serviceContext[0].endpointContext[0].address),
                1)
        self.failUnlessEqual(metadata.serviceContext[0].endpointContext[0].address[0],
                spSoapEndpoint)
        self.failUnlessEqual(
                len(metadata.serviceContext[0].endpointContext[0].securityMechId),
                1)
        self.failUnlessEqual(
                metadata.serviceContext[0].endpointContext[0].securityMechId[0],
                lasso.SECURITY_MECH_BEARER)
        self.failUnless(metadata.svcMDID is None)
        wsp_disco.buildRequestMsg()
        self.failUnlessEqual(wsp_disco.msgUrl, idpSoapEndpoint)
        self.failUnless(wsp_disco.msgBody is not None)

        idp_disco = lasso.IdWsf2Discovery(idp)
        self.failUnless(idp_disco is not None)
        try:
            idp_disco.processRequestMsg(wsp_disco.msgBody)
        except lasso.Error, e:
            self.fail(e)
        self.failUnless(idp_disco.request is not None)
        try:
            idp_disco.checkSecurityMechanism()
        except lasso.Error, e:
            self.fail(e)
        try:
            idp_disco.failRequest(lasso.IDWSF2_DISCOVERY_STATUS_CODE_FAILED, lasso.IDWSF2_DISCOVERY_STATUS_CODE_FORBIDDEN)
        except lasso.Error, e:
            self.fail(e)
        self.failUnless(idp_disco.response is not None)
        self.failUnless(idp_disco.response.status is not None)
        self.failUnless(idp_disco.response.status.code is not lasso.IDWSF2_DISCOVERY_STATUS_CODE_FAILED)
        self.failUnlessEqual(len(idp_disco.response.status.status), 1)
        self.failUnless(idp_disco.response.status.status[0].code is not lasso.IDWSF2_DISCOVERY_STATUS_CODE_FORBIDDEN)
        idp_disco.buildResponseMsg()
        self.failUnless(idp_disco.msgUrl is None)
        self.failUnless(idp_disco.msgBody is not None)

        try:
            wsp_disco.processResponseMsg(idp_disco.msgBody)
        except lasso.Idwsf2DiscoveryForbiddenError:
            pass
        except lasso.Error, e:
            self.fail(e)

    def test03(self):
        """Test metadata register with redirection"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump, dst_epr = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        wsp_disco.setEpr(dst_epr)
        print dst_epr.dump()

        abstract = 'Personal Profile service'
        wsp_disco.initMetadataRegister()
        soap_envelope = wsp_disco.getSoapEnvelopeRequest()
        soap_envelope.setSb2UserInteractionHint(lasso.IDWSF2_SB2_USER_INTERACTION_HINT_INTERACT_IF_NEEDED)
        self.failUnless(isinstance(soap_envelope.header, lasso.SoapHeader))
        self.failUnless(len(soap_envelope.header.other) > 0)
        self.failUnlessEqual(soap_envelope.getSb2UserInteractionHint(), lasso.IDWSF2_SB2_USER_INTERACTION_HINT_INTERACT_IF_NEEDED)
        self.failUnless(wsp_disco.request is not None)
        wsp_disco.addSimpleServiceMetadata(service_types = 
                (lasso.PP11_HREF,), abstract = abstract,
                provider_id = wsp.providerId, address = spSoapEndpoint,
                security_mech_ids = (lasso.SECURITY_MECH_BEARER,))
        self.failUnlessEqual(len(wsp_disco.metadatas), 1)
        metadata = wsp_disco.metadatas[0]
        self.failUnlessEqual(metadata.abstract, abstract)
        self.failUnlessEqual(metadata.providerId, wsp.providerId)
        self.failUnlessEqual(len(metadata.serviceContext), 1)
        self.failUnlessEqual(len(metadata.serviceContext[0].serviceType), 1)
        self.failUnlessEqual(metadata.serviceContext[0].serviceType[0],
                lasso.PP11_HREF)
        self.failUnlessEqual(len(metadata.serviceContext[0].endpointContext), 1)
        self.failUnlessEqual(
                len(metadata.serviceContext[0].endpointContext[0].address),
                1)
        self.failUnlessEqual(metadata.serviceContext[0].endpointContext[0].address[0],
                spSoapEndpoint)
        self.failUnlessEqual(
                len(metadata.serviceContext[0].endpointContext[0].securityMechId),
                1)
        self.failUnlessEqual(
                metadata.serviceContext[0].endpointContext[0].securityMechId[0],
                lasso.SECURITY_MECH_BEARER)
        self.failUnless(metadata.svcMDID is None)
        wsp_disco.buildRequestMsg()
        self.failUnlessEqual(wsp_disco.msgUrl, idpSoapEndpoint)
        self.failUnless(wsp_disco.msgBody is not None)

        idp_disco = lasso.IdWsf2Discovery(idp)
        self.failUnless(idp_disco is not None)
        idp_disco.processRequestMsg(wsp_disco.msgBody)
        self.failUnless(idp_disco.request is not None)
        self.failUnlessEqual(len(idp_disco.request.svcMD), 1)
        self.failUnless(idp_disco.request.svcMD[0].svcMDID is None)
        soap_envelope = idp_disco.getSoapEnvelopeRequest()
        self.failUnless(soap_envelope is not None)
        self.failUnless(soap_envelope.getMessageId() is not None)
        try:
            idp_disco.checkSecurityMechanism()
        except lasso.Error, e:
            self.fail(e)
        # redirect
        interactionUrl = spInteractionUrl
        idp_disco.redirectUserForInteraction(interactionUrl, False)
        response = idp_disco.response
        self.failUnless(isinstance(response, lasso.SoapFault))
        self.failUnless(response.detail is not None)
        self.failUnlessEqual(len(response.detail.any), 1)
        self.failUnless(isinstance(response.detail.any[0], lasso.IdWsf2Sb2RedirectRequest))
        self.failUnless(response.detail.any[0].redirectURL.startswith(interactionUrl + '?transactionID='))
        try:
            idp_disco.buildResponseMsg()
        except lasso.Error, e:
            self.fail(e)
        self.failUnless(idp_disco.msgBody is not None)


        self.failUnless(idp_disco.msgUrl is None)
        self.failUnless(idp_disco.msgBody is not None)

        try:
            wsp_disco.processResponseMsg(idp_disco.msgBody)
        except lasso.WsfprofileRedirectRequestError:
            pass
        except lasso.Error, e:
            self.fail(e)
        response_envelope = wsp_disco.getSoapEnvelopeResponse()
        self.failUnless(response_envelope.sb2GetRedirectRequestUrl().startswith(interactionUrl + '?transactionID='))
        # Here keep information about the request associated to ID: response_envelope.getMessageId().content
        wsp_disco_dump = wsp_disco.dump()
        wsp_disco = lasso.Node.newFromDump(wsp_disco_dump)
        request_envelope = wsp_disco.getSoapEnvelopeRequest()
        self.failUnless(request_envelope is not None)
        relates_to = request_envelope.getRelatesTo(True)
        self.failUnless(relates_to is not None)
        response_message_id = response_envelope.getMessageId().content
        relates_to.content = response_message_id
        wsp_disco.buildRequestMsg()
        # now redo as for test01 after request building
        self.failUnlessEqual(wsp_disco.msgUrl, idpSoapEndpoint)
        self.failUnless(wsp_disco.msgBody is not None)

        idp_disco = lasso.IdWsf2Discovery(idp)
        self.failUnless(idp_disco is not None)
        idp_disco.processRequestMsg(wsp_disco.msgBody)
        self.failUnless(idp_disco.request is not None)
        self.failUnlessEqual(len(idp_disco.request.svcMD), 1)
        self.failUnless(idp_disco.request.svcMD[0].svcMDID is None)
        try:
            idp_disco.checkSecurityMechanism()
        except lasso.Error, e:
            print wsp_disco.msgBody
            self.fail(e)
        try:
            idp_disco.validateRequest()
        except lasso.Error, e:
            self.fail(e)
        self.failUnless(idp_disco.response is not None)
        self.failUnlessEqual(len(idp_disco.metadatas), 1)
        metadata = idp_disco.metadatas[0]
        self.failUnlessEqual(metadata.abstract, abstract)
        self.failUnlessEqual(metadata.providerId, wsp.providerId)
        self.failUnlessEqual(len(metadata.serviceContext), 1)
        self.failUnlessEqual(len(metadata.serviceContext[0].serviceType), 1)
        self.failUnlessEqual(metadata.serviceContext[0].serviceType[0],
                lasso.PP11_HREF)
        self.failUnlessEqual(len(metadata.serviceContext[0].endpointContext), 1)
        self.failUnlessEqual(
                len(metadata.serviceContext[0].endpointContext[0].address),
                1)
        self.failUnlessEqual(metadata.serviceContext[0].endpointContext[0].address[0],
                spSoapEndpoint)
        self.failUnlessEqual(
                len(metadata.serviceContext[0].endpointContext[0].securityMechId),
                1)
        self.failUnlessEqual(
                metadata.serviceContext[0].endpointContext[0].securityMechId[0],
                lasso.SECURITY_MECH_BEARER)
        idp_disco.buildResponseMsg()
        self.failUnless(metadata.svcMDID is not None)
        self.failUnless(idp_disco.msgUrl is None)
        self.failUnless(idp_disco.msgBody is not None)

        wsp_disco.processResponseMsg(idp_disco.msgBody)

        self.failUnless(len(wsp_disco.metadatas) == 1, 'missing svcMDID')
        self.failUnless(wsp_disco.metadatas[0].svcMDID, 'missing svcMDID')

class MetadataAssociationAddTestCase(IdWsf2TestCase):
    def test01(self):
        """Init metadata association add request"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMD = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump, dst_epr = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        try:
            wsp_disco.initMetadataAssociationAdd()
        except lasso.Error, e:
            self.fail(e)
        try:
            wsp_disco.addServiceMetadata(svcMD)
        except lasso.Error, e:
            self.fail(e)

    def test03(self):
        """Init metadata association add request - msgUrl construction"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMD = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump, dst_epr = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        wsp_disco.setEpr(dst_epr)
        wsp_disco.initMetadataAssociationAdd()
        wsp_disco.addServiceMetadata(svcMD)
        wsp_disco.buildRequestMsg()

        self.failUnless(wsp_disco.msgUrl, 'missing msgUrl')
        self.failUnless(wsp_disco.msgBody, 'missing msgBody')

    def test05(self):
        """Process metadata association add request"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMD = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump, dst_epr = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        wsp_disco.setEpr(dst_epr)
        wsp_disco.initMetadataAssociationAdd()
        wsp_disco.addServiceMetadata(svcMD)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)

        try:
            idp_disco.processRequestMsg(wsp_disco.msgBody)
        except lasso.Error, e:
            self.fail(e)

        try:
            idp_disco.checkSecurityMechanism()
        except lasso.Error, e:
            self.fail(e)

        try:
            idp_disco.validateRequest()
        except lasso.Error, e:
            self.fail(e)

    def test06(self):
        """Register metadata association"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        try:
            idp_disco.registerMetadata()
        except lasso.Error, e:
            self.fail(e)

    def test07(self):
        """Check metadata association registration"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)
        idp_disco.registerMetadata()

        self.failUnless(idp_disco.isIdentityDirty, 'identity has not changed, it should contain a svcMDID')
        self.failUnless(idp_disco.identity.dump() != idp_identity_dump,
            'identity dump has not changed, it should contain a svcMDID')

    def test08(self):
        """Build metadata association add response"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()

        idp_disco.buildResponseMsg()

        self.failUnless(idp_disco.msgBody)

    def test09(self):
        """Process metadata association add response"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        try:
            wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)
        except lasso.Error, e:
            self.fail(e)


class DiscoveryQueryTestCase(IdWsf2TestCase):
    def test01(self):
        """Init discovery query"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        try:
            wsc_disco.initQuery()
        except lasso.Error, e:
            self.fail(e)

    def test02(self):
        """Init discovery query without login"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()

        wsc_disco = lasso.IdWsf2Discovery(wsc)

        try:
            wsc_disco.initQuery()
        except lasso.Error, e:
            if e[0] != lasso.PROFILE_ERROR_SESSION_NOT_FOUND:
                self.fail(e)
        else:
            self.fail('Should have a "session not found" exception')

    def test03(self):
        """Init discovery query - check msg url"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        wsc_disco.initQuery()

        self.failUnless(wsc_disco.msgUrl, 'missing msgUrl')

    def test04(self):
        """Add requested service type to discovery query"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        wsc_disco.initQuery()

        try:
            wsc_disco.addRequestedServiceType('urn:liberty:id-sis-pp:2005-05')
        except lasso.Error, e:
            self.fail(e)

    def test05(self):
        """Build discovery query"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        wsc_disco.initQuery()
        wsc_disco.addRequestedServiceType('urn:liberty:id-sis-pp:2005-05')
        wsc_disco.buildRequestMsg()

        self.failUnless(wsc_disco.msgBody)

    def test06(self):
        """Process discovery query"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        wsc_disco.initQuery()
        wsc_disco.addRequestedServiceType('urn:liberty:id-sis-pp:2005-05')
        wsc_disco.buildRequestMsg()

        try:
            idp_disco.processQueryMsg(wsc_disco.msgBody)
        except lasso.Error, e:
            self.fail(e)

    def test07(self):
        """Process discovery query and check name identifier"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        wsc_disco.initQuery()
        wsc_disco.addRequestedServiceType('urn:liberty:id-sis-pp:2005-05')
        wsc_disco.buildRequestMsg()

        idp_disco.processQueryMsg(wsc_disco.msgBody)

    def test08(self):
        """Build discovery query response EPRs"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        wsc_disco.initQuery()
        wsc_disco.addRequestedServiceType('urn:liberty:id-sis-pp:2005-05')
        wsc_disco.buildRequestMsg()

        idp_disco.processQueryMsg(wsc_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        try:
            idp_disco.buildQueryResponseEprs()
        except lasso.Error, e:
            self.fail(e)

    def test09(self):
        """Build discovery query response"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        wsc_disco.initQuery()
        wsc_disco.addRequestedServiceType('urn:liberty:id-sis-pp:2005-05')
        wsc_disco.buildRequestMsg()

        idp_disco.processQueryMsg(wsc_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)
        idp_disco.buildQueryResponseEprs()
        idp_disco.buildResponseMsg()

        self.failUnless(idp_disco.msgBody, 'missing msgBody')

    def test10(self):
        """Process discovery query response"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        wsc_disco.initQuery()
        wsc_disco.addRequestedServiceType('urn:liberty:id-sis-pp:2005-05')
        wsc_disco.buildRequestMsg()

        idp_disco.processQueryMsg(wsc_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)
        idp_disco.buildQueryResponseEprs()
        idp_disco.buildResponseMsg()

        try:
            wsc_disco.processQueryResponseMsg(idp_disco.msgBody)
        except lasso.Error, e:
            self.fail(e)

    def test11(self):
        """Check discovery query result"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        wsc_disco.initQuery()
        wsc_disco.addRequestedServiceType('urn:liberty:id-sis-pp:2005-05')
        wsc_disco.buildRequestMsg()

        idp_disco.processQueryMsg(wsc_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)
        idp_disco.buildQueryResponseEprs()
        idp_disco.buildResponseMsg()

        wsc_disco.processQueryResponseMsg(idp_disco.msgBody)

        self.failUnless(wsc_disco.getService(), 'missing service after discovery query')


class DataServiceQueryTestCase(IdWsf2TestCase):
    def getProfileService(self):
        """Check discovery query result"""
        idp = self.getIdpServer()
        wsp = self.getWspServer()
        idp, svcMDID = self.metadataRegister(wsp, idp)
        wsp_identity_dump, wsp_session_dump, idp_identity_dump, idp_session_dump = self.login(wsp, idp)

        wsp_disco = lasso.IdWsf2Discovery(wsp)
        if wsp_identity_dump is not None:
            wsp_disco.setIdentityFromDump(wsp_identity_dump)
        if wsp_session_dump is not None:
            wsp_disco.setSessionFromDump(wsp_session_dump)
        wsp_disco.initMetadataAssociationAdd(svcMDID)
        wsp_disco.buildRequestMsg()

        idp_disco = lasso.IdWsf2Discovery(idp)
        idp_disco.processMetadataAssociationAddMsg(wsp_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)

        idp_disco.registerMetadata()
        if idp_disco.isIdentityDirty:
            idp_identity_dump = idp_disco.identity.dump()
        if idp_disco.isSessionDirty:
            idp_session_dump = idp_disco.session.dump()
        idp_disco.buildResponseMsg()

        wsp_disco.processMetadataAssociationAddResponseMsg(idp_disco.msgBody)

        wsc = self.getWscServer()
        wsc_identity_dump, wsc_session_dump, idp_identity_dump, idp_session_dump = \
            self.login(wsc, idp, None, None, idp_identity_dump, idp_session_dump)

        wsc_disco = lasso.IdWsf2Discovery(wsc)
        if wsc_identity_dump is not None:
            wsc_disco.setIdentityFromDump(wsc_identity_dump)
        if wsc_session_dump is not None:
            wsc_disco.setSessionFromDump(wsc_session_dump)

        wsc_disco.initQuery()
        wsc_disco.addRequestedServiceType('urn:liberty:id-sis-pp:2005-05')
        wsc_disco.buildRequestMsg()

        idp_disco.processQueryMsg(wsc_disco.msgBody)
        if idp_identity_dump is not None:
            idp_disco.setIdentityFromDump(idp_identity_dump)
        if idp_session_dump is not None:
            idp_disco.setSessionFromDump(idp_session_dump)
        idp_disco.buildQueryResponseEprs()
        idp_disco.buildResponseMsg()

        wsc_disco.processQueryResponseMsg(idp_disco.msgBody)

        return wsc_disco.getService(), wsp

    def test01(self):
        """Data service init query"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')

        try:
            service.initQuery()
        except lasso.Error, e:
            self.fail(e)

    def test02(self):
        """Data service init query - msgUrl construction"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()

        self.failUnless(service.msgUrl, 'missing msgUrl')

    def test03(self):
        """Data service add query item"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()

        try:
            service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        except lasso.Error, e:
            self.fail(e)

    def test04(self):
        """Data service build query"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.buildRequestMsg()

        self.failUnless(service.msgBody, 'missing msgBody')

    def test05(self):
        """Data service build query with multiple items"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        self.failUnless(service.msgBody, 'missing msgBody')

    def test06(self):
        """Data service process query"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)

        try:
            wsp_service.processQueryMsg(service.msgBody)
        except lasso.Error, e:
            self.fail(e)

    def test07(self):
        """Data service check service type"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        self.failUnless(wsp_service.type, 'service type is not set')
        self.failUnless(wsp_service.type == 'urn:liberty:id-sis-pp:2005-05', 'wrong service type')

    def test08(self):
        """Data service get query items"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        self.failUnless(wsp_service.queryItems, 'queryItems list is None or empty')

    def test09(self):
        """Data service check query items"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        items = [ '/pp2:PP/pp2:InformalName', 'not existing attribute', '/pp2:PP/pp2:MsgContact' ]
        for i in range(3):
            self.failUnless(wsp_service.queryItems[i] == items[i],
                "query items don't match : %s != %s" % (wsp_service.queryItems[i], items[i]))

    def test10(self):
        """Data service check name identifier"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

    def test11(self):
        """Data service parse query items - success"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        try:
            wsp_service.parseQueryItems()
        except lasso.Error, e:
            self.fail(e)

    def test12(self):
        """Data service parse query items - failure - no item"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        try:
            wsp_service.parseQueryItems()
        except lasso.DstEmptyRequestError:
            pass
        except Exception, e:
            self.fail(e)

    def test13(self):
        """Data service parse query items - failure - wrong item"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        try:
            wsp_service.parseQueryItems()
        except lasso.Error, e:
            if e[0] != lasso.DST_ERROR_QUERY_FAILED:
                self.fail(e)
        else:
             self.fail('query items parsing should have failed because a wrong query item was requested')

    def test14(self):
        """Data service parse query items - failure - no data"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        try:
            wsp_service.parseQueryItems()
        except lasso.Error, e:
            if e[0] != lasso.DST_ERROR_MISSING_SERVICE_DATA:
                self.fail(e)
        else:
             self.fail('query items parsing should have failed because no data was provided')


    def test15(self):
        """Data service parse query items - partial failure"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
            </PP>"""

        # No exception should be raised here but one will be raised on the WSC
        # when parsing response status
        try:
            wsp_service.parseQueryItems()
        except lasso.Error, e:
            self.fail(e)

    def test16(self):
        """Data service build query response"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        wsp_service.parseQueryItems()
        wsp_service.buildResponseMsg()

        self.failUnless(wsp_service.msgBody, 'missing msgBody')

    def test17(self):
        """Data service process query response - success"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        wsp_service.parseQueryItems()
        wsp_service.buildResponseMsg()

        try:
            service.processQueryResponseMsg(wsp_service.msgBody)
        except lasso.Error, e:
            self.fail(e)

    def test18(self):
        """Data service process query response - failure"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        try:
            wsp_service.parseQueryItems()
        except lasso.Error, e:
            pass
        wsp_service.buildResponseMsg()

        try:
            service.processQueryResponseMsg(wsp_service.msgBody)
        except lasso.Error, e:
            if e[0] != lasso.DST_ERROR_QUERY_FAILED:
                self.fail(e)
        else:
             self.fail('response should have a "failed" status because a wrong query item was requested')

    def test19(self):
        """Data service process query response - partial failure"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        try:
            wsp_service.parseQueryItems()
        except lasso.DstQueryPartiallyFailedError:
            pass
        except:
             self.fail('parseQueryItems should emit a "partially failed" error because a wrong query item was requested')
        wsp_service.buildResponseMsg()

        try:
            service.processQueryResponseMsg(wsp_service.msgBody)
        except lasso.DstQueryPartiallyFailedError:
            pass
        except:
             self.fail('response should have a "partially failed" status because a wrong query item was requested')

    def test20(self):
        """Data service get first attribute"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        wsp_service.parseQueryItems()
        wsp_service.buildResponseMsg()

        service.processQueryResponseMsg(wsp_service.msgBody)
        informal_name = service.getAttributeNode()

        self.failUnlessEqual(informal_name, """<pp2:InformalName xmlns="urn:liberty:id-sis-pp:2005-05" xmlns:pp2="urn:liberty:id-sis-pp:2005-05">User name</pp2:InformalName>""", 'first attribute node is wrong')

    def test21(self):
        """Data service get attribute string"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        wsp_service.parseQueryItems()
        wsp_service.buildResponseMsg()

        service.processQueryResponseMsg(wsp_service.msgBody)
        informal_name = service.getAttributeString('name')

        self.failUnlessEqual(informal_name, 'User name', 'attribute string is wrong')

    def test22(self):
        """Data service get attribute node"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        wsp_service.parseQueryItems()
        wsp_service.buildResponseMsg()

        service.processQueryResponseMsg(wsp_service.msgBody)
        email = service.getAttributeNode('email')

        expected_result = """<pp2:MsgContact xmlns="urn:liberty:id-sis-pp:2005-05" xmlns:pp2="urn:liberty:id-sis-pp:2005-05">.*?<pp2:MsgAccount>Email account</pp2:MsgAccount>.*?<pp2:MsgProvider>Email server</pp2:MsgProvider>.*?</pp2:MsgContact>"""

        import re
        result = re.findall(expected_result, email, re.DOTALL)

        self.failUnless(len(result) == 1, 'attribute node is wrong')

    def test23(self):
        """Data service get attribute node - partial failure"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <InformalName>User name</InformalName>
                <MsgContact>
                    <MsgAccount>Email account</MsgAccount>
                    <MsgProvider>Email server</MsgProvider>
                </MsgContact>
            </PP>"""

        try:
            wsp_service.parseQueryItems()
        except lasso.DstQueryPartiallyFailedError:
            pass
        except:
             self.fail('parseQueryItems should emit a "partially failed" error because a wrong query item was requested')
        wsp_service.buildResponseMsg()

        try:
            service.processQueryResponseMsg(wsp_service.msgBody)
        except lasso.Error, e:
            if e[0] == lasso.DST_ERROR_QUERY_PARTIALLY_FAILED:
                pass
        informal_name = service.getAttributeString('name')
        email = service.getAttributeNode('email')

        self.failUnlessEqual(informal_name, 'User name', 'attribute string is wrong')
        self.failUnlessEqual(email, None, 'attribute node should be None')

    def test24(self):
        """Data service redirect request"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        if '/pp2:PP/pp2:MsgContact' in wsp_service.queryItems:
            wsp_service.initRedirectUserForConsent('http://sp5/consent');
        wsp_service.buildResponseMsg()

        try:
            service.processQueryResponseMsg(wsp_service.msgBody)
        except lasso.SoapRedirectRequestFaultError:
            pass
        except Exception, e:
            self.fail(e)
        else:
            self.fail('a "soap fault redirect request" exception should have been raised')

    def test25(self):
        """Data service redirect request - check redirectUrl"""
        service, wsp = self.getProfileService()
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:InformalName', 'name')
        service.addQueryItem('not existing attribute', 'not existing attribute')
        service.addQueryItem('/pp2:PP/pp2:MsgContact', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        if '/pp2:PP/pp2:MsgContact' in wsp_service.queryItems:
            wsp_service.initRedirectUserForConsent('http://sp5/consent');
        wsp_service.buildResponseMsg()

        try:
            service.processQueryResponseMsg(wsp_service.msgBody)
        except lasso.SoapRedirectRequestFaultError:
            pass
        except Exception, e:
            self.fail(e)

        self.failUnlessEqual(service.redirectUrl, 'http://sp5/consent', 'redirectUrl is not set or wrong')

    def test26(self):
        """Data service get multiple attribute nodes with the same name"""
        lasso.registerIdWsf2DstService('pp2', 'urn:liberty:id-sis-pp:2005-05')
        service, wsp = self.getProfileService()
        service.initQuery()
        service.addQueryItem('/pp2:PP/pp2:MsgContact/pp2:MsgAccount', 'email')
        service.buildRequestMsg()

        wsp_service = lasso.IdWsf2DataService(wsp)
        wsp_service.processQueryMsg(service.msgBody)

        email1 = 'Email account 1'
        email2 = 'Email account 2'
        wsp_service.data = """<PP xmlns="urn:liberty:id-sis-pp:2005-05">
                <MsgContact>
                    <MsgAccount>%s</MsgAccount>
                </MsgContact>
                <MsgContact>
                    <MsgAccount>%s</MsgAccount>
                </MsgContact>
            </PP>""" % (email1, email2)

        wsp_service.parseQueryItems()
        wsp_service.buildResponseMsg()

        service.processQueryResponseMsg(wsp_service.msgBody)

        email_nodes = service.getAttributeNodes('email')
        self.failUnless(email_nodes[0] ==
            '<pp2:MsgAccount xmlns="urn:liberty:id-sis-pp:2005-05" xmlns:pp2="urn:liberty:id-sis-pp:2005-05">%s</pp2:MsgAccount>' % email1)
        self.failUnless(email_nodes[1] ==
            '<pp2:MsgAccount xmlns="urn:liberty:id-sis-pp:2005-05" xmlns:pp2="urn:liberty:id-sis-pp:2005-05">%s</pp2:MsgAccount>' % email2)

        email_strings = service.getAttributeStrings('email')
        self.failUnless(email_strings[0] == email1)
        self.failUnless(email_strings[1] == email2)


metadataSuite = unittest.makeSuite(MetadataTestCase, 'test')
#metadataAssociationAddSuite = unittest.makeSuite(MetadataAssociationAddTestCase, 'test')
#discoveryQuerySuite = unittest.makeSuite(DiscoveryQueryTestCase, 'test')
#dataServiceQuerySuite = unittest.makeSuite(DataServiceQueryTestCase, 'test')

allTests = unittest.TestSuite((metadataSuite,))
#    metadataAssociationAddSuite,)) # discoveryQuerySuite, dataServiceQuerySuite))

if __name__ == '__main__':
    sys.exit(not unittest.TextTestRunner(verbosity = 2).run(allTests).wasSuccessful())