1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
import mock, socket, os, time
from libmproxy import dump
from netlib import certutils, tcp
from libpathod.pathoc import Pathoc
import tutils
def get_free_port():
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind(("127.0.0.1", 0))
port = s.getsockname()[1]
s.close()
return port
class AppTestMixin(object):
def request(self, spec):
t_start = time.time()
while (time.time() - t_start) < 5:
try:
p = Pathoc(("127.0.0.1", self.port))
p.connect() # might fail as the server might not be online yet.
return p.request(spec)
except tcp.NetLibError:
time.sleep(0.01)
assert False
def test_basic(self):
assert self.request("get:/").status_code == 200
assert self.request("get:/").status_code == 200 # Check for connection close
assert len(self.m.apps.apps) == 0
def test_cert(self):
with tutils.tmpdir() as d:
# Create Certs
path = os.path.join(d, "test")
assert certutils.dummy_ca(path)
self.m.server.config.cacert = path
for ext in ["pem", "p12"]:
resp = self.request("get:/cert/%s" % ext)
assert resp.status_code == 200
with open(path + "-cert.%s" % ext, "rb") as f:
assert resp.content == f.read()
class TestAppExternal(AppTestMixin):
@classmethod
def setupAll(cls):
cls.port = get_free_port()
o = dump.Options(app=True, app_external=True, app_host="127.0.0.1", app_port=cls.port)
s = mock.MagicMock()
cls.m = dump.DumpMaster(s, o, None)
@classmethod
def teardownAll(cls):
cls.m.shutdown()
|
