def test_getCertsFromCell_invalid_count(self): lc = test_cert_der link_cert = CertsCellPayloadItem(LINK_CERT_TYPE, len(lc), lc) cell = CertsCell.make(0, [link_cert]) self.assertRaises(ValueError, connectionbuildtask._getCertsFromCell, cell)
def test_getCertsFromCell_malformed_cert(self): lc = test_cert_der link_cert = CertsCellPayloadItem(LINK_CERT_TYPE, len(lc), lc) ic = test_cert_der[:len(test_cert_der)-1] id_cert = CertsCellPayloadItem(ID_CERT_TYPE, len(ic), ic) cell = CertsCell.make(0, [link_cert, id_cert]) self.assertRaises(ValueError, connectionbuildtask._getCertsFromCell, cell)
def test_getCertsFromCell(self): lc = test_cert_der link_cert = CertsCellPayloadItem(LINK_CERT_TYPE, len(lc), lc) ic = test_cert_der id_cert = CertsCellPayloadItem(ID_CERT_TYPE, len(ic), ic) cell = CertsCell.make(0, [link_cert, id_cert]) res1 = crypto.load_certificate(crypto.FILETYPE_ASN1, lc) res2 = crypto.load_certificate(crypto.FILETYPE_ASN1, ic) l, i = connectionbuildtask._getCertsFromCell(cell) self.assertEqual(crypto.dump_certificate(crypto.FILETYPE_ASN1, l), lc) self.assertEqual(crypto.dump_certificate(crypto.FILETYPE_ASN1, i), ic)