def testGetCaParametersDefault(self):
"""Test GetCaParametersDefault()."""
self.mox.StubOutWithMock(util, 'GetCaParameters')
settings = {'whatever': 1}
util.GetCaParameters(settings, None).AndReturn('ok')
self.mox.ReplayAll()
self.assertEqual('ok', util.GetCaParametersDefault(settings))
self.mox.VerifyAll()
def LoadCaParameters(self, settings):
"""Load ca/cert parameters for default CA.
Args:
settings: object with attribute level access to settings.
Raises:
CaParametersError: if any errors occur loading keys/certs.
"""
try:
ca_params = util.GetCaParameters(settings)
except util.CaParametersError, e:
raise CaParametersError(*e.args)
def testGetCaParametersDefaultWithOmitServerPrivateKey(self):
"""Test GetCaParametersDefault()."""
self.mox.StubOutWithMock(util, 'GetCaParameters')
settings = {'whatever': 1}
util.GetCaParameters(
settings, None, omit_server_private_key=True).AndReturn('ok')
self.mox.ReplayAll()
self.assertEqual(
'ok',
util.GetCaParametersDefault(settings, omit_server_private_key=True))
self.mox.VerifyAll()
def LoadCaParameters(self, settings, ca_id=None):
"""Load ca/cert parameters for CA named ca_id.
Args:
settings: object with attribute level access to settings.
ca_id: str or None (default), identifies the CA/server cert/keys.
Raises:
CaParametersError: if any errors occur loading keys/certs for ca_id.
"""
try:
ca_params = util.GetCaParameters(settings, ca_id)
except util.CaParametersError, e:
raise CaParametersError(*e.args)
def testGetCaParametersWhenForcePrimaryCaSettings(self):
"""Test GetCaParameters()."""
# Mock to detect that it was NOT called:
self.mox.StubOutWithMock(util, 'GetCaId')
self.mox.ReplayAll()
p = util.GetCaParameters(self.settings, None) # Force primary CA settings
self.assertEqual(p.ca_public_cert_pem, self.settings.CA_PUBLIC_CERT_PEM)
self.assertEqual(
p.server_public_cert_pem, self.settings.SERVER_PUBLIC_CERT_PEM)
self.assertEqual(p.server_private_key_pem, 'foo_server_private_pem')
self.assertEqual(p.required_issuer, self.settings.REQUIRED_ISSUER)
self.assertTrue(p.ca_id is None)
self.mox.VerifyAll()
def testGetCaParametersWhenFollowCaIdValueNone(self):
"""Test GetCaParameters()."""
self.settings.CA_ID = None
self.mox.StubOutWithMock(util, 'GetCaId')
util.GetCaId(self.settings).AndReturn(None) # CA_ID=None
self.mox.ReplayAll()
p = util.GetCaParameters(self.settings)
self.assertEqual(p.ca_public_cert_pem, self.settings.CA_PUBLIC_CERT_PEM)
self.assertEqual(
p.server_public_cert_pem, self.settings.SERVER_PUBLIC_CERT_PEM)
self.assertEqual(p.server_private_key_pem, 'foo_server_private_pem')
self.assertEqual(p.required_issuer, self.settings.REQUIRED_ISSUER)
self.assertTrue(p.ca_id is None)
self.mox.VerifyAll()
def testGetCaParametersWhenCaIdFoo(self):
"""Test GetCaParameters()."""
ca_id = 'FOO'
self.mox.StubOutWithMock(util, 'GetCaId')
util.GetCaId(self.settings).AndReturn(ca_id) # CA_ID='FOO'
self.settings.CA_ID = ca_id
self.settings.FOO_CA_PUBLIC_CERT_PEM = '__ca!'
self.settings.FOO_SERVER_PUBLIC_CERT_PEM = '__pub!'
self.settings.FOO_SERVER_PRIVATE_KEY_PEM = '__priv!'
self.settings.FOO_REQUIRED_ISSUER = '__ri!'
self.mox.ReplayAll()
p = util.GetCaParameters(self.settings)
self.assertEqual(p.ca_public_cert_pem, '__ca!')
self.assertEqual(p.server_public_cert_pem, '__pub!')
self.assertEqual(p.server_private_key_pem, '__priv!')
self.assertEqual(p.required_issuer, '__ri!')
self.assertEqual(p.ca_id, ca_id)
self.mox.VerifyAll()
def _LoadCaParameters(self):
"""Load CA parameters from settings."""
logging.debug('LoadCaParameters')
self._ca_params = util.GetCaParameters(
settings, omit_server_private_key=True)
logging.debug('Loaded ca_params')
