def test_generate_certificate_signed_with_ca(self):
manager = CertManager()
cert, key = manager.generate_certificate()
x509 = manager.load_certificate(filename=cert)
self.assertEquals(x509.get_issuer(), manager.CA_CERT.get_subject())
os.unlink(cert)
os.unlink(key)
def test_generate_certificate_signed_with_ca(self):
manager = CertManager()
cert, key = manager.generate_certificate()
x509 = manager.load_certificate(filename=cert)
self.assertEquals(x509.get_issuer(), manager.CA_CERT.get_subject())
os.unlink(cert)
os.unlink(key)
def test_generate_certificate_self_signed(self):
manager = CertManager()
attribs = {"self-signed": True}
cert, key = manager.generate_certificate(attributes=attribs)
x509 = manager.load_certificate(filename=cert)
self.assertEquals(x509.get_issuer(), x509.get_subject())
os.unlink(cert)
os.unlink(key)
def test_generate_certificate_self_signed(self):
manager = CertManager()
attribs = {"self-signed": True}
cert, key = manager.generate_certificate(attributes=attribs)
x509 = manager.load_certificate(filename=cert)
self.assertEquals(x509.get_issuer(), x509.get_subject())
os.unlink(cert)
os.unlink(key)
class TestFakeTLSServer(unittest.TestCase):
"""
Yeah.. these are not "unit" tests,, but whatever...
"""
def setUp(self):
self.cert_manager = CertManager()
self.callback_test = callbackTest()
def test_local_server_up_ca_signed(self):
crt_file, key_file = self.cert_manager.generate_certificate()
crt_file = os.path.join(Configuration().get_temp_certificate_folder(),
crt_file)
key_file = os.path.join(Configuration().get_temp_certificate_folder(),
key_file)
worker1 = WebServerSetup(keyfile=key_file,
certfile=crt_file,
server_address=("127.0.0.1", 4444),
callback=self.callback_test.increment)
worker1.start()
AuxiliarHTTPSClient.request("https://127.0.0.1:4444/")
# import time
# time.sleep(200)
worker1.kill()
self.assertEquals(1, self.callback_test.track)
def test_local_server_up_self_signed(self):
attribs = {"self-signed": True}
crt_file, key_file = self.cert_manager.generate_certificate(attribs)
crt_file = os.path.join(Configuration().get_temp_certificate_folder(),
crt_file)
key_file = os.path.join(Configuration().get_temp_certificate_folder(),
key_file)
worker1 = WebServerSetup(keyfile=key_file,
certfile=crt_file,
server_address=("127.0.0.1", 4444),
callback=self.callback_test.increment)
worker1.start()
AuxiliarHTTPSClient.request("https://127.0.0.1:4444/")
worker1.kill()
self.assertEquals(0, self.callback_test.track)
def tearDown(self):
target_folder = Configuration().get_temp_certificate_folder()
for f in os.listdir(target_folder):
os.remove(os.path.join(target_folder, f))
class TestFakeTLSServer(unittest.TestCase):
"""
Yeah.. these are not "unit" tests,, but whatever...
"""
def setUp(self):
self.cert_manager = CertManager()
self.callback_test = callbackTest()
def test_local_server_up_ca_signed(self):
crt_file, key_file = self.cert_manager.generate_certificate()
crt_file = os.path.join(Configuration().get_temp_certificate_folder(), crt_file)
key_file = os.path.join(Configuration().get_temp_certificate_folder(), key_file)
worker1 = WebServerSetup(keyfile=key_file,
certfile=crt_file,
server_address=("127.0.0.1", 4444),
callback=self.callback_test.increment)
worker1.start()
AuxiliarHTTPSClient.request("https://127.0.0.1:4444/")
# import time
# time.sleep(200)
worker1.kill()
self.assertEquals(1, self.callback_test.track)
def test_local_server_up_self_signed(self):
attribs = {"self-signed": True}
crt_file, key_file = self.cert_manager.generate_certificate(attribs)
crt_file = os.path.join(Configuration().get_temp_certificate_folder(), crt_file)
key_file = os.path.join(Configuration().get_temp_certificate_folder(), key_file)
worker1 = WebServerSetup(keyfile=key_file,
certfile=crt_file,
server_address=("127.0.0.1", 4444),
callback=self.callback_test.increment)
worker1.start()
AuxiliarHTTPSClient.request("https://127.0.0.1:4444/")
worker1.kill()
self.assertEquals(0, self.callback_test.track)
def tearDown(self):
target_folder = Configuration().get_temp_certificate_folder()
for f in os.listdir(target_folder):
os.remove(os.path.join(target_folder, f))
def test_generate_certificate_signed_with_ca_with_subject(self):
manager = CertManager()
x509_details = {"C": "US", "ST": "Boston",
"L": "Boston", "O": "CORE Security",
"OU": "SCS", "CN": "itsevart.com"}
attribs = {
"subject": manager.create_X509Name(x509_details)
}
cert, key = manager.generate_certificate(attribs)
x509 = manager.load_certificate(filename=cert)
self.assertEquals(x509.get_issuer(), manager.CA_CERT.get_subject())
self.assertEquals("itsevart.com", x509.get_subject().CN)
self.assertEquals("Boston", x509.get_subject().ST)
os.unlink(cert)
os.unlink(key)
def test_generate_certificate_signed_with_ca_with_subject(self):
manager = CertManager()
x509_details = {
"C": "US",
"ST": "Boston",
"L": "Boston",
"O": "CORE Security",
"OU": "SCS",
"CN": "itsevart.com"
}
attribs = {"subject": manager.create_X509Name(x509_details)}
cert, key = manager.generate_certificate(attribs)
x509 = manager.load_certificate(filename=cert)
self.assertEquals(x509.get_issuer(), manager.CA_CERT.get_subject())
self.assertEquals("itsevart.com", x509.get_subject().CN)
self.assertEquals("Boston", x509.get_subject().ST)
os.unlink(cert)
os.unlink(key)
