def testGetCaIdWhenFoo(self):
"""Test GetCaId()."""
self.settings.CA_ID = 'FOO'
self.mox.ReplayAll()
ca_id = util.GetCaId(self.settings)
self.assertEqual(ca_id, 'FOO')
self.mox.VerifyAll()
def testGetCaIdWhenNone(self):
"""Test GetCaId()."""
self.settings.CA_ID = None
self.mox.ReplayAll()
ca_id = util.GetCaId(self.settings)
self.assertTrue(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 testGetCaParametersWhenCaIdBarAndMissingParam(self):
"""Test GetCaParameters()."""
ca_id = 'BAR'
self.mox.StubOutWithMock(util, 'GetCaId')
util.GetCaId(self.settings).AndReturn(ca_id) # CA_ID='FOO'
self.settings.CA_ID = ca_id
self.settings.BAR_CA_PUBLIC_CERT_PEM = '__ca!'
self.settings.BAR_SERVER_PUBLIC_CERT_PEM = '__pub!'
self.settings.BAR_SERVER_PRIVATE_KEY_PEM = '__priv!'
# intentionally omit:
# self.settings.BAR_REQUIRED_ISSUER = '__ri!'
self.mox.ReplayAll()
self.assertRaises(util.CaParametersError, util.GetCaParameters,
self.settings)
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()