Nice fix from Lorenzo for testing valid_until

This commit is contained in:
Roland Hedberg
2010-12-09 08:41:50 +01:00
parent 305e534d34
commit 09dc404f09

View File

@@ -1,3 +1,5 @@
import datetime
import re
import os
from saml2 import metadata, make_vals, make_instance
@@ -32,6 +34,12 @@ def _read_lines(name):
except IOError:
name = "tests/"+name
return open(name).readlines()
def _fix_valid_until(xmlstring):
new_date = datetime.datetime.now() + datetime.timedelta(days=1)
new_date = new_date.strftime("%Y-%m-%dT%H:%M:%SZ")
return re.sub(r' validUntil=".*?"', ' validUntil="%s"' % new_date,
xmlstring)
ATTRCONV = ac_factory("attributemaps")
@@ -131,7 +139,7 @@ def test_switch_1():
def test_sp_metadata():
md = metadata.MetaData(attrconv=ATTRCONV)
md.import_metadata(_read_file(SP_METADATA), "-")
md.import_metadata(_fix_valid_until(_read_file(SP_METADATA)), "-")
print md.entity
assert len(md.entity) == 1
@@ -417,4 +425,4 @@ def test_attributes():
assert ra[i].is_required == "True"
assert ra[0].friendly_name == "surname"
assert ra[0].name == 'urn:oid:2.5.4.4'