You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
3.8 KiB
86 lines
3.8 KiB
# coding=utf-8
|
|
"""
|
|
Created on Jun 14, 2015
|
|
|
|
@author: danimar
|
|
"""
|
|
import unittest
|
|
import os
|
|
import os.path
|
|
from pytrustnfe.certificado import Certificado
|
|
from pytrustnfe.certificado import save_cert_key
|
|
from pytrustnfe.certificado import extract_cert_and_key_from_pfx
|
|
|
|
CHAVE = (
|
|
"-----BEGIN PRIVATE KEY-----\n"
|
|
"MIICdQIBADANBgkqhkiG9w0BAQEFAASCAl8wggJbAgEAAoGBAJONRp6l1y2ojgv8\n"
|
|
"tP3AOLW0vjWQqiPseBLM7YAxbzz5R7LYlWHC0ZJ4uIvd4Cvc6AuoNJoeuhzFcwHx\n"
|
|
"PL0TcFuW+5up1ktUohwaJ+/zKrMODCKt0gvif302yqasMnwLh9mGZQIkLkHPOX8p\n"
|
|
"ZQDC4dlqwOyYDi0f+bRd5C7aWx3RAgMBAAECgYADqASP+dwTLZIXifOSNikxl4D/\n"
|
|
"Is6UhU+UZ6+a9Z6kDClSrTtGaOV4k7U/AgiEDb1STKDBEPHbtKjc63Vt2gV2teem\n"
|
|
"ohU0Giv+gD42uuwy2DM31OfYrpR46mzOK9JrpQc78b36ealL3AWJ1gyBbbcOWbAb\n"
|
|
"KmP742V7pcD07EEp4QJBAM/e7M8VdLgOyaQzH9KHekU6fJlI4vy1UwgRUwx3/1W6\n"
|
|
"zlBYo1qXfc7NSVG8ZaSrJwW4rPn393u31CpXv+oc/OMCQQC1txS6nxM9+p/641HX\n"
|
|
"CHXiWJRn0Wv7rT1FyF2dHO+OQOkCCnHCsGDMf3bacTNb7iyaPbXEDac8od5uF/3h\n"
|
|
"aUy7AkBDPGoAeYItXqseL2Mlp6iG5+oRcp/o+YWH4IKqT84JHslI98KutL1+vKvw\n"
|
|
"gi2mW63djeR1Xh1wqP85SvTKduHdAkAIJLlIF8Lr/yRWQQO06EsoJqIX+Pmm4L+j\n"
|
|
"NfSECvztWhlXHxK0D+V2pKu15GbR0t2q1+Micx4wiGyIcIjPJkHrAkAvlbXGFcGT\n"
|
|
"pk9bQ8nl7EYqlvVn1TejzTLfBhBYOse/xT/NI4Kwjkan9R+EJ1cOc9EE8gm1W3jv\n"
|
|
"fMw/Bh2wC5kj\n"
|
|
"-----END PRIVATE KEY-----\n"
|
|
)
|
|
|
|
CERTIFICADO = (
|
|
"-----BEGIN CERTIFICATE-----\n"
|
|
"MIICMTCCAZqgAwIBAgIQfYOsIEVuAJ1FwwcTrY0t1DANBgkqhkiG9w0BAQUFADBX\n"
|
|
"MVUwUwYDVQQDHkwAewA1ADkARgAxAEUANAA2ADEALQBEAEQARQA1AC0ANABEADIA\n"
|
|
"RgAtAEEAMAAxAEEALQA4ADMAMwAyADIAQQA5AEUAQgA4ADMAOAB9MB4XDTE1MDYx\n"
|
|
"NTA1NDc1N1oXDTE2MDYxNDExNDc1N1owVzFVMFMGA1UEAx5MAHsANQA5AEYAMQBF\n"
|
|
"ADQANgAxAC0ARABEAEUANQAtADQARAAyAEYALQBBADAAMQBBAC0AOAAzADMAMgAy\n"
|
|
"AEEAOQBFAEIAOAAzADgAfTCBnzANBgkqhkiG9w0BAQEFAAOBjQAwgYkCgYEAk41G\n"
|
|
"nqXXLaiOC/y0/cA4tbS+NZCqI+x4EsztgDFvPPlHstiVYcLRkni4i93gK9zoC6g0\n"
|
|
"mh66HMVzAfE8vRNwW5b7m6nWS1SiHBon7/Mqsw4MIq3SC+J/fTbKpqwyfAuH2YZl\n"
|
|
"AiQuQc85fyllAMLh2WrA7JgOLR/5tF3kLtpbHdECAwEAATANBgkqhkiG9w0BAQUF\n"
|
|
"AAOBgQArdh+RyT6VxKGsXk1zhHsgwXfToe6GpTF4W8PHI1+T0WIsNForDhvst6nm\n"
|
|
"QtgAhuZM9rxpOJuNKc+pM29EixpAiZZiRMCSWEItNyEVdUIi+YnKBcAHd88TwO86\n"
|
|
"d126MWQ2O8cu5W1VoDp7hYBYKOnLbYi11/StO+0rzK+oPYAvIw==\n"
|
|
"-----END CERTIFICATE-----\n"
|
|
)
|
|
|
|
|
|
class test_assinatura(unittest.TestCase):
|
|
|
|
caminho = os.path.dirname(__file__)
|
|
|
|
def test_preparar_pfx(self):
|
|
dir_pfx = open(os.path.join(self.caminho, "teste.pfx"), "rb").read()
|
|
cert, key = extract_cert_and_key_from_pfx(dir_pfx, "123456")
|
|
self.assertEqual(key, CHAVE, "Chave gerada inválida")
|
|
self.assertEqual(cert, CERTIFICADO, "Certificado inválido")
|
|
|
|
def test_save_pfx(self):
|
|
pfx_source = open(os.path.join(self.caminho, "teste.pfx"), "rb").read()
|
|
pfx = Certificado(pfx_source, "123")
|
|
path = pfx.save_pfx()
|
|
saved = open(path, "rb").read()
|
|
self.assertEqual(
|
|
pfx_source, saved, "Arquivo pfx salvo não bate com arquivo lido"
|
|
)
|
|
|
|
def test_save_cert_and_key(self):
|
|
dir_pfx = open(os.path.join(self.caminho, "teste.pfx"), "rb").read()
|
|
cert, key = extract_cert_and_key_from_pfx(dir_pfx, "123456")
|
|
cert_path, key_path = save_cert_key(cert, key)
|
|
cert_saved = open(cert_path, "r").read()
|
|
key_saved = open(key_path, "r").read()
|
|
self.assertEqual(cert, cert_saved, "Certificado não corresponde ao original")
|
|
self.assertEqual(key, key_saved, "Chave não corresponde ao original")
|
|
|
|
def test_pfx_nao_existe(self):
|
|
self.assertRaises(
|
|
Exception, extract_cert_and_key_from_pfx, "file.pfx", "123456"
|
|
)
|
|
|
|
def test_pfx_senha_invalida(self):
|
|
dir_pfx = os.path.join(self.caminho, "teste.pfx")
|
|
self.assertRaises(Exception, extract_cert_and_key_from_pfx, dir_pfx, "123")
|