def terr(req=None):
if not req:
req = treq()
f = req.flow
f.error = Error("error")
f.error.reply = controller.DummyReply()
return f.error
def test_all(self):
s = flow.State()
fm = flow.FlowMaster(None, s)
fm.anticache = True
fm.anticomp = True
req = tutils.treq()
fm.handle_clientconnect(req.flow.client_conn)
f = fm.handle_request(req)
assert s.flow_count() == 1
resp = tutils.tresp(req)
fm.handle_response(resp)
assert s.flow_count() == 1
rx = tutils.tresp()
rx.flow = None
assert not fm.handle_response(rx)
fm.handle_clientdisconnect(req.flow.client_conn)
f.error = Error("msg")
f.error.reply = controller.DummyReply()
fm.handle_error(f.error)
fm.load_script(tutils.test_data.path("scripts/a.py"))
fm.shutdown()
def test_getset_state(self):
f = tutils.tflow(resp=True)
state = f.get_state()
assert f.get_state() == protocol.http.HTTPFlow.from_state(state).get_state()
f.response = None
f.error = Error("error")
state = f.get_state()
assert f.get_state() == protocol.http.HTTPFlow.from_state(state).get_state()
f2 = f.copy()
assert f.get_state() == f2.get_state()
assert not f == f2
f2.error = Error("e2")
assert not f == f2
f.load_state(f2.get_state())
assert f.get_state() == f2.get_state()
def test_err(self):
c = flow.State()
req = tutils.treq()
f = c.add_request(req)
f.error = Error("message")
assert c.add_error(f.error)
e = Error("message")
assert not c.add_error(e)
c = flow.State()
req = tutils.treq()
f = c.add_request(req)
e = tutils.terr()
c.set_limit("~e")
assert not c.view
assert c.add_error(e)
assert c.view
def test_getset_state(self):
e = Error("Error")
state = e._get_state()
assert Error._from_state(state) == e
assert e.copy()
e2 = Error("bar")
assert not e == e2
e._load_state(e2._get_state())
assert e == e2
e3 = e.copy()
assert e3 == e
def test_err(self):
c = flow.State()
f = tutils.tflow()
c.add_request(f)
f.error = Error("message")
assert c.add_error(f)
c = flow.State()
f = tutils.tflow()
c.add_request(f)
c.set_limit("~e")
assert not c.view
f.error = tutils.terr()
assert c.add_error(f)
assert c.view
def test_client_playback(self):
s = flow.State()
f = tutils.tflow(resp=True)
pb = [tutils.tflow(resp=True), f]
fm = flow.FlowMaster(None, s)
assert not fm.start_server_playback(pb, False, [], False, False)
assert not fm.start_client_playback(pb, False)
q = Queue.Queue()
assert not fm.state.flow_count()
fm.tick(q, 0)
assert fm.state.flow_count()
f.error = Error("error")
fm.handle_error(f)
def test_client_playback(self):
s = flow.State()
f = tutils.tflow_full()
pb = [tutils.tflow_full(), f]
fm = flow.FlowMaster(None, s)
assert not fm.start_server_playback(pb, False, [], False, False)
assert not fm.start_client_playback(pb, False)
q = Queue.Queue()
assert not fm.state.flow_count()
fm.tick(q)
assert fm.state.flow_count()
f.error = Error("error")
f.error.reply = controller.DummyReply()
fm.handle_error(f.error)
def terr(content="error"):
"""
@return: libmproxy.protocol.primitives.Error
"""
err = Error(content)
return err
def test_getset_state(self):
e = Error("Error")
state = e._get_state()
assert Error._from_state(state) == e
assert e.copy()
e2 = Error("bar")
assert not e == e2
e._load_state(e2._get_state())
assert e == e2
e3 = e.copy()
assert e3 == e
def test_getset_state(self):
e = Error("Error")
state = e.get_state()
assert Error.from_state(state).get_state() == e.get_state()
assert e.copy()
e2 = Error("bar")
assert not e == e2
e.load_state(e2.get_state())
assert e.get_state() == e2.get_state()
e3 = e.copy()
assert e3.get_state() == e.get_state()
def err(self):
f = self.req()
f.error = Error("msg")
return f
def test_getset_state(self):
e = Error("Error")
state = e.get_state()
assert Error.from_state(state).get_state() == e.get_state()
assert e.copy()
e2 = Error("bar")
assert not e == e2
e.load_state(e2.get_state())
assert e.get_state() == e2.get_state()
e3 = e.copy()
assert e3.get_state() == e.get_state()
def _add_error(self, state):
req = tutils.treq()
f = state.add_request(req)
f.error = Error("msg")
