#!/usr/bin/env python

from saml2.client import Saml2Client
from saml2 import samlp

XML_RESPONSE_FILE = "saml_response.xml"
XML_RESPONSE_FILE2 = "saml2_response.xml"
#XML_RESPONSE_FILE3 = "sun_saml2_response.xml"

def for_me(condition, me ):
    for restriction in condition.audience_restriction:
        audience = restriction.audience
        if audience.text.strip() == me:
            return True

def ava(attribute_statement):
    result = {}
    for attribute in attribute_statement.attribute:
        # Check name_format ??
        name = attribute.name.strip()
        result[name] = []
        for value in attribute.attribute_value:
            result[name].append(value.text.strip())
    return result

def test_verify_1():
    xml_response = open(XML_RESPONSE_FILE).read()
    client = Saml2Client({})
    (ava, came_from) = \
            client.verify(xml_response, "xenosmilus.umdc.umu.se",decode=False)
    assert ava == {'__userid': '_cddc88563d433f556d4cc70c3162deabddea3b5019', 
                    'eduPersonAffiliation': ['member', 'student'], 
                    'uid': ['student']}
    
def test_parse_1():
    xml_response = open(XML_RESPONSE_FILE).read()
    response = samlp.response_from_string(xml_response)
    client = Saml2Client({})
    (ava, name_id, real_uri) = \
            client.do_response(response, "xenosmilus.umdc.umu.se")
    assert ava == {'eduPersonAffiliation': ['member', 'student'], 'uid': ['student']}
    assert name_id == "_cddc88563d433f556d4cc70c3162deabddea3b5019"

def test_parse_2():
    xml_response = open(XML_RESPONSE_FILE2).read()
    response = samlp.response_from_string(xml_response)
    client = Saml2Client({})
    (ava, name_id, real_uri) = \
            client.do_response(response, "xenosmilus.umdc.umu.se")
    assert ava == {'uid': ['andreas'], 
                    'mobile': ['+4741107700'], 
                    'edupersonnickname': ['erlang'], 
                    'o': ['Feide RnD'], 
                    'edupersonentitlement': ['urn:mace:feide.no:entitlement:test'], 
                    'edupersonaffiliation': ['employee'], 
                    'eduPersonPrincipalName': ['andreas@rnd.feide.no'], 
                    'sn': ['Solberg'], 
                    'mail': ['andreas@uninett.no'], 
                    'ou': ['Guests'], 
                    'cn': ['Andreas Solberg']}
    assert name_id == "_242f88493449e639aab95dd9b92b1d04234ab84fd8"
        
# def test_parse_3():
#     xml_response = open(XML_RESPONSE_FILE3).read()
#     response = samlp.response_from_string(xml_response)
#     client = Saml2Client({})
#     (ava, name_id, real_uri) = \
#             client.do_response(response, "xenosmilus.umdc.umu.se")
#     print 40*"="
#     print ava
#     print 40*","
#     print name_id
#     assert False