mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-30 11:19:23 +00:00
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
import os
|
|
from netlib import certutils
|
|
import tutils
|
|
|
|
|
|
def test_dummy_ca():
|
|
with tutils.tmpdir() as d:
|
|
path = os.path.join(d, "foo/cert.cnf")
|
|
assert certutils.dummy_ca(path)
|
|
assert os.path.exists(path)
|
|
|
|
path = os.path.join(d, "foo/cert2.pem")
|
|
assert certutils.dummy_ca(path)
|
|
assert os.path.exists(path)
|
|
assert os.path.exists(os.path.join(d, "foo/cert2-cert.pem"))
|
|
assert os.path.exists(os.path.join(d, "foo/cert2-cert.p12"))
|
|
|
|
|
|
class TestDummyCert:
|
|
def test_with_ca(self):
|
|
with tutils.tmpdir() as d:
|
|
cacert = os.path.join(d, "foo/cert.cnf")
|
|
assert certutils.dummy_ca(cacert)
|
|
p = certutils.dummy_cert(
|
|
os.path.join(d, "foo"),
|
|
cacert,
|
|
"foo.com",
|
|
["one.com", "two.com", "*.three.com"]
|
|
)
|
|
assert os.path.exists(p)
|
|
|
|
# Short-circuit
|
|
assert certutils.dummy_cert(
|
|
os.path.join(d, "foo"),
|
|
cacert,
|
|
"foo.com",
|
|
[]
|
|
)
|
|
|
|
def test_no_ca(self):
|
|
with tutils.tmpdir() as d:
|
|
p = certutils.dummy_cert(
|
|
d,
|
|
None,
|
|
"foo.com",
|
|
[]
|
|
)
|
|
assert os.path.exists(p)
|
|
|
|
|
|
class TestSSLCert:
|
|
def test_simple(self):
|
|
c = certutils.SSLCert.from_pem(file(tutils.test_data.path("data/text_cert"), "r").read())
|
|
assert c.cn == "google.com"
|
|
assert len(c.altnames) == 436
|
|
|
|
c = certutils.SSLCert.from_pem(file(tutils.test_data.path("data/text_cert_2"), "r").read())
|
|
assert c.cn == "www.inode.co.nz"
|
|
assert len(c.altnames) == 2
|
|
assert c.digest("sha1")
|
|
assert c.notbefore
|
|
assert c.notafter
|
|
assert c.subject
|
|
assert c.keyinfo == ("RSA", 2048)
|
|
assert c.serial
|
|
assert c.issuer
|
|
assert c.to_pem()
|
|
c.has_expired
|
|
|
|
def test_err_broken_sans(self):
|
|
c = certutils.SSLCert.from_pem(file(tutils.test_data.path("data/text_cert_weird1"), "r").read())
|
|
# This breaks unless we ignore a decoding error.
|
|
c.altnames
|
|
|
|
def test_der(self):
|
|
d = file(tutils.test_data.path("data/dercert")).read()
|
|
s = certutils.SSLCert.from_der(d)
|
|
assert s.cn
|