Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues Projects Releases Wiki Activity

Factor out flow unit tests into speparate file.

Browse Source
This commit is contained in:
Aldo Cortesi 2011-01-31 11:48:53 +13:00
parent 152b97fa0b
commit edb8228dd2
3 changed files with 134 additions and 126 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

142
test/test_console.py
View File

@ -1,26 +1,7 @@
from libmproxy import console, proxy, utils, filt, flow from libmproxy import console, proxy, filt, flow
import utils
import libpry import libpry
def treq(conn=None):
if not conn:
conn = proxy.BrowserConnection("address", 22)
headers = utils.Headers()
headers["header"] = ["qvalue"]
return proxy.Request(conn, "host", 80, "http", "GET", "/path", headers, "content")
def tresp(req=None):
if not req:
req = treq()
headers = utils.Headers()
headers["header_response"] = ["svalue"]
return proxy.Response(req, 200, "HTTP/1.1", "message", headers, "content_response")
def tflow():
bc = proxy.BrowserConnection("address", 22)
return flow.Flow(bc)
class uState(libpry.AutoTree): class uState(libpry.AutoTree):
def test_backup(self): def test_backup(self):
@ -45,21 +26,21 @@ class uState(libpry.AutoTree):
assert c.lookup(bc) assert c.lookup(bc)
assert c.get_focus() == (f, 0) assert c.get_focus() == (f, 0)
req = treq(bc) req = utils.treq(bc)
assert c.add_request(req) assert c.add_request(req)
assert len(c.flow_list) == 1 assert len(c.flow_list) == 1
assert c.lookup(req) assert c.lookup(req)
newreq = treq() newreq = utils.treq()
assert not c.add_request(newreq) assert not c.add_request(newreq)
assert not c.lookup(newreq) assert not c.lookup(newreq)
resp = tresp(req) resp = utils.tresp(req)
assert c.add_response(resp) assert c.add_response(resp)
assert len(c.flow_list) == 1 assert len(c.flow_list) == 1
assert c.lookup(resp) assert c.lookup(resp)
newresp = tresp() newresp = utils.tresp()
assert not c.add_response(newresp) assert not c.add_response(newresp)
assert not c.lookup(newresp) assert not c.lookup(newresp)
@ -77,7 +58,7 @@ class uState(libpry.AutoTree):
def test_view(self): def test_view(self):
c = console.ConsoleState() c = console.ConsoleState()
f = tflow() f = utils.tflow()
c.add_browserconnect(f) c.add_browserconnect(f)
assert len(c.view) == 1 assert len(c.view) == 1
c.set_limit(filt.parse("~q")) c.set_limit(filt.parse("~q"))
@ -85,8 +66,8 @@ class uState(libpry.AutoTree):
c.set_limit(None) c.set_limit(None)
f = tflow() f = utils.tflow()
req = treq(f.connection) req = utils.treq(f.connection)
c.add_browserconnect(f) c.add_browserconnect(f)
c.add_request(req) c.add_request(req)
assert len(c.view) == 2 assert len(c.view) == 2
@ -130,15 +111,15 @@ class uState(libpry.AutoTree):
assert c.get_focus() == (None, None) assert c.get_focus() == (None, None)
def _add_request(self, state): def _add_request(self, state):
f = tflow() f = utils.tflow()
state.add_browserconnect(f) state.add_browserconnect(f)
q = treq(f.connection) q = utils.treq(f.connection)
state.add_request(q) state.add_request(q)
return f return f
def _add_response(self, state): def _add_response(self, state):
f = self._add_request(state) f = self._add_request(state)
r = tresp(f.request) r = utils.tresp(f.request)
state.add_response(r) state.add_response(r)
def test_focus_view(self): def test_focus_view(self):
@ -155,8 +136,8 @@ class uState(libpry.AutoTree):
def test_delete_last(self): def test_delete_last(self):
c = console.ConsoleState() c = console.ConsoleState()
f1 = tflow() f1 = utils.tflow()
f2 = tflow() f2 = utils.tflow()
c.add_browserconnect(f1) c.add_browserconnect(f1)
c.add_browserconnect(f2) c.add_browserconnect(f2)
c.set_focus(1) c.set_focus(1)
@ -165,14 +146,14 @@ class uState(libpry.AutoTree):
def test_kill_flow(self): def test_kill_flow(self):
c = console.ConsoleState() c = console.ConsoleState()
f = tflow() f = utils.tflow()
c.add_browserconnect(f) c.add_browserconnect(f)
c.kill_flow(f) c.kill_flow(f)
assert not c.flow_list assert not c.flow_list
def test_clear(self): def test_clear(self):
c = console.ConsoleState() c = console.ConsoleState()
f = tflow() f = utils.tflow()
c.add_browserconnect(f) c.add_browserconnect(f)
f.intercepting = True f.intercepting = True
@ -197,96 +178,6 @@ class uState(libpry.AutoTree):
assert isinstance(c.flow_list[0], flow.Flow) assert isinstance(c.flow_list[0], flow.Flow)
class uFlow(libpry.AutoTree):
def test_match(self):
f = tflow()
f.response = tresp()
f.request = f.response.request
assert not f.match(filt.parse("~b test"))
def test_backup(self):
f = tflow()
f.backup()
f.revert()
def test_getset_state(self):
f = tflow()
state = f.get_state()
assert f == flow.Flow.from_state(state)
f.response = tresp()
f.request = f.response.request
state = f.get_state()
assert f == flow.Flow.from_state(state)
def test_simple(self):
f = tflow()
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.request = treq()
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.response = tresp()
f.response.headers["content-type"] = ["text/html"]
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.response.code = 404
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.focus = True
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.connection = flow.ReplayConnection()
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.response = None
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.error = proxy.Error(200, "test")
assert console.format_flow(f, True)
assert console.format_flow(f, False)
def test_kill(self):
f = tflow()
f.request = treq()
f.intercept()
assert not f.request.acked
f.kill()
assert f.request.acked
f.intercept()
f.response = tresp()
f.request = f.response.request
f.request.ack()
assert not f.response.acked
f.kill()
assert f.response.acked
def test_accept_intercept(self):
f = tflow()
f.request = treq()
f.intercept()
assert not f.request.acked
f.accept_intercept()
assert f.request.acked
f.response = tresp()
f.request = f.response.request
f.intercept()
f.request.ack()
assert not f.response.acked
f.accept_intercept()
assert f.response.acked
def test_serialization(self):
f = flow.Flow(None)
f.request = treq()
class uformat_keyvals(libpry.AutoTree): class uformat_keyvals(libpry.AutoTree):
def test_simple(self): def test_simple(self):
assert console.format_keyvals( assert console.format_keyvals(
@ -327,7 +218,6 @@ class uPathCompleter(libpry.AutoTree):
tests = [ tests = [
uFlow(),
uformat_keyvals(), uformat_keyvals(),
uState(), uState(),
uPathCompleter() uPathCompleter()

96
test/test_flow.py Normal file
View File

@ -0,0 +1,96 @@
from libmproxy import console, proxy, filt, flow
import utils
import libpry
class uFlow(libpry.AutoTree):
def test_match(self):
f = utils.tflow()
f.response = utils.tresp()
f.request = f.response.request
assert not f.match(filt.parse("~b test"))
def test_backup(self):
f = utils.tflow()
f.backup()
f.revert()
def test_getset_state(self):
f = utils.tflow()
state = f.get_state()
assert f == flow.Flow.from_state(state)
f.response = utils.tresp()
f.request = f.response.request
state = f.get_state()
assert f == flow.Flow.from_state(state)
def test_simple(self):
f = utils.tflow()
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.request = utils.treq()
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.response = utils.tresp()
f.response.headers["content-type"] = ["text/html"]
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.response.code = 404
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.focus = True
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.connection = flow.ReplayConnection()
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.response = None
assert console.format_flow(f, True)
assert console.format_flow(f, False)
f.error = proxy.Error(200, "test")
assert console.format_flow(f, True)
assert console.format_flow(f, False)
def test_kill(self):
f = utils.tflow()
f.request = utils.treq()
f.intercept()
assert not f.request.acked
f.kill()
assert f.request.acked
f.intercept()
f.response = utils.tresp()
f.request = f.response.request
f.request.ack()
assert not f.response.acked
f.kill()
assert f.response.acked
def test_accept_intercept(self):
f = utils.tflow()
f.request = utils.treq()
f.intercept()
assert not f.request.acked
f.accept_intercept()
assert f.request.acked
f.response = utils.tresp()
f.request = f.response.request
f.intercept()
f.request.ack()
assert not f.response.acked
f.accept_intercept()
assert f.response.acked
def test_serialization(self):
f = flow.Flow(None)
f.request = utils.treq()
tests = [
uFlow()
]

22
test/utils.py Normal file
View File

@ -0,0 +1,22 @@
from libmproxy import proxy, utils, filt, flow
def treq(conn=None):
if not conn:
conn = proxy.BrowserConnection("address", 22)
headers = utils.Headers()
headers["header"] = ["qvalue"]
return proxy.Request(conn, "host", 80, "http", "GET", "/path", headers, "content")
def tresp(req=None):
if not req:
req = treq()
headers = utils.Headers()
headers["header_response"] = ["svalue"]
return proxy.Response(req, 200, "HTTP/1.1", "message", headers, "content_response")
def tflow():
bc = proxy.BrowserConnection("address", 22)
return flow.Flow(bc)