Fix list assumption in virtual org tests

This commit is contained in:
Clint Byrum 2015-05-26 15:20:51 -07:00
parent 0e68174a9a
commit 133501e69f

View File

@ -33,7 +33,7 @@ class TestVirtualOrg():
conf.load_file("server_conf") conf.load_file("server_conf")
self.sp = Saml2Client(conf) self.sp = Saml2Client(conf)
vo_name = conf.vorg.keys()[0] vo_name = list(conf.vorg.keys())[0]
self.vo = conf.vorg[vo_name] self.vo = conf.vorg[vo_name]
add_derek_info(self.sp) add_derek_info(self.sp)
@ -62,7 +62,7 @@ class TestVirtualOrg_2():
def setup_class(self): def setup_class(self):
conf = config.SPConfig() conf = config.SPConfig()
conf.load_file("server_conf") conf.load_file("server_conf")
vo_name = conf.vorg.keys()[0] vo_name = list(conf.vorg.keys())[0]
self.sp = Saml2Client(conf, virtual_organization=vo_name) self.sp = Saml2Client(conf, virtual_organization=vo_name)
add_derek_info(self.sp) add_derek_info(self.sp)