def testLoadPemGenericWhenSpacesNoLastNewline(self): """Test LoadPemGeneric().""" header = 'BEGIN' footer = 'END' input = ' \n\n\n-----BEGIN----- \nhello \n-----END-----' expected = [ '-----BEGIN-----', 'hello', '-----END-----', ] self.assertEqual(expected, x509.LoadPemGeneric(input, header, footer))
def testLoadPemGenericWhenInfo(self): """Test LoadPemGeneric().""" header = 'BEGIN' footer = 'END' input = ('\n\n\n-----BEGIN-----\n' 'Proc-Type: foo\nhello\n-----END-----\n\n\n') expected = [ '-----BEGIN-----', 'hello', '-----END-----', ] self.assertEqual(expected, x509.LoadPemGeneric(input, header, footer))
def testLoadRSAPrivateKeyFromPEM(self): """Test LoadRSAPrivateKeyFromPEM().""" header = 'BEGIN RSA PRIVATE KEY' footer = 'END RSA PRIVATE KEY' pem_input = 'pem_input' pem_output = ['---header---', 'base64', '---footer---'] self.mox.StubOutWithMock(x509, 'LoadPemGeneric') self.mox.StubOutWithMock(x509.tlslite_bridge, 'parsePEMKey') x509.LoadPemGeneric(pem_input, header, footer).AndReturn(pem_output) x509.tlslite_bridge.parsePEMKey('\n'.join(pem_output)).AndReturn('ok') self.mox.ReplayAll() self.assertEqual(x509.LoadRSAPrivateKeyFromPEM(pem_input), 'ok') self.mox.VerifyAll()
def testLoadCertificateFromPEM(self): """Test LoadCertificateFromPEM().""" header = 'BEGIN CERTIFICATE' footer = 'END CERTIFICATE' pem_input = 'pem_input' pem_output = ['---header---', 'base64', '---footer---'] self.mox.StubOutWithMock(x509, 'LoadPemGeneric') self.mox.StubOutWithMock(x509, 'LoadCertificateFromBase64') x509.LoadPemGeneric(pem_input, header, footer).AndReturn(pem_output) x509.LoadCertificateFromBase64('base64').AndReturn('ok') self.mox.ReplayAll() self.assertEqual(x509.LoadCertificateFromPEM(pem_input), 'ok') self.mox.VerifyAll()
def testLoadRSAPrivateKeyFromPEMWhenSyntaxError(self): """Test LoadRSAPrivateKeyFromPEM().""" header = 'BEGIN RSA PRIVATE KEY' footer = 'END RSA PRIVATE KEY' pem_input = 'pem_input' pem_output = ['---header---', 'base64', '---footer---'] self.mox.StubOutWithMock(x509, 'LoadPemGeneric') self.mox.StubOutWithMock(x509.tlslite_bridge, 'parsePEMKey') x509.LoadPemGeneric(pem_input, header, footer).AndReturn(pem_output) x509.tlslite_bridge.parsePEMKey( '\n'.join(pem_output)).AndRaise(SyntaxError) self.mox.ReplayAll() self.assertRaises(x509.RSAPrivateKeyPEMFormatError, x509.LoadRSAPrivateKeyFromPEM, pem_input) self.mox.VerifyAll()