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')