Re-initialize NSS database after otptoken plugin tests

OTP token tests do not properly reinitialize the NSS db, thus
making subsequent xmlrpc tests fail on SSL cert validation.

Make sure NSS db is re-initalized in the teardown method.

https://fedorahosted.org/freeipa/ticket/4748

Reviewed-By: Petr Viktorin <pviktori@redhat.com>
This commit is contained in:
Tomas Babej 2014-11-20 18:37:57 +01:00 committed by Petr Viktorin
parent b3a6701e73
commit 792ff0c0c4
2 changed files with 26 additions and 12 deletions

View File

@ -89,6 +89,26 @@ def strip_header(pem):
return pem return pem
def initialize_nss_database(dbdir=None):
"""
Initializes NSS database, if not initialized yet. Uses a proper database
directory (.ipa/alias or HTTPD_ALIAS_DIR), depending on the value of
api.env.in_tree.
"""
if not nss.nss_is_initialized():
if dbdir is None:
if 'in_tree' in api.env:
if api.env.in_tree:
dbdir = api.env.dot_ipa + os.sep + 'alias'
else:
dbdir = paths.HTTPD_ALIAS_DIR
nss.nss_init(dbdir)
else:
nss.nss_init_nodb()
else:
nss.nss_init(dbdir)
def load_certificate(data, datatype=PEM, dbdir=None): def load_certificate(data, datatype=PEM, dbdir=None):
""" """
Given a base64-encoded certificate, with or without the Given a base64-encoded certificate, with or without the
@ -103,18 +123,7 @@ def load_certificate(data, datatype=PEM, dbdir=None):
data = strip_header(data) data = strip_header(data)
data = base64.b64decode(data) data = base64.b64decode(data)
if not nss.nss_is_initialized(): initialize_nss_database(dbdir=dbdir)
if dbdir is None:
if 'in_tree' in api.env:
if api.env.in_tree:
dbdir = api.env.dot_ipa + os.sep + 'alias'
else:
dbdir = paths.HTTPD_ALIAS_DIR
nss.nss_init(dbdir)
else:
nss.nss_init_nodb()
else:
nss.nss_init(dbdir)
return nss.Certificate(buffer(data)) return nss.Certificate(buffer(data))

View File

@ -21,12 +21,17 @@ import os
import sys import sys
import nose import nose
from nss import nss from nss import nss
from ipalib.x509 import initialize_nss_database
from ipaserver.install.ipa_otptoken_import import PSKCDocument, ValidationError from ipaserver.install.ipa_otptoken_import import PSKCDocument, ValidationError
basename = os.path.join(os.path.dirname(__file__), "data") basename = os.path.join(os.path.dirname(__file__), "data")
class test_otptoken_import(object): class test_otptoken_import(object):
def teardown(self):
initialize_nss_database()
def test_figure3(self): def test_figure3(self):
doc = PSKCDocument(os.path.join(basename, "pskc-figure3.xml")) doc = PSKCDocument(os.path.join(basename, "pskc-figure3.xml"))
assert doc.keyname is None assert doc.keyname is None