26 lines
505 B
Python
26 lines
505 B
Python
import cStringIO, textwrap
|
|
from cStringIO import StringIO
|
|
import libpry
|
|
from libmproxy import proxy, flow
|
|
import tutils
|
|
|
|
|
|
class TestProxyError:
|
|
def test_simple(self):
|
|
p = proxy.ProxyError(111, "msg")
|
|
assert repr(p)
|
|
|
|
|
|
class TestAppRegistry:
|
|
def test_add_get(self):
|
|
ar = proxy.AppRegistry()
|
|
ar.add("foo", "domain", 80)
|
|
|
|
r = tutils.treq()
|
|
r.host = "domain"
|
|
r.port = 80
|
|
assert ar.get(r)
|
|
|
|
r.port = 81
|
|
assert not ar.get(r)
|