def test_client_playback(self): s = flow.State() f = tutils.tflow(resp=True) pb = [tutils.tflow(resp=True), f] fm = flow.FlowMaster(DummyServer(ProxyConfig()), s) assert not fm.start_server_playback( pb, False, [], False, False, None, False, None, False) assert not fm.start_client_playback(pb, False) fm.client_playback.testing = True 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_load_flows_reverse(self): r = self._treader() s = flow.State() conf = ProxyConfig(mode="reverse", upstream_server=[True,True,"use-this-domain",80]) fm = flow.FlowMaster(DummyServer(conf), s) fm.load_flows(r) assert s._flow_list[0].request.host == "use-this-domain"
def test_err_2(self): conf = ProxyConfig(host="invalidhost") tutils.raises("error starting proxy server", ProxyServer, conf)
def test_err(self): conf = ProxyConfig(port=1) tutils.raises("error starting proxy server", ProxyServer, conf)
""" with open('blacklist', 'r') as ff: domains = ff.xreadlines() for domain in domains: domain = domain.strip().rstrip() if domain.find('#'): continue blacklist.append(domain) """ if __name__ == '__main__': whitelist = [] config = ProxyConfig( host=mitm_config['host'], port=mitm_config['port'], # use ~/.mitmproxy/mitmproxy-ca.pem as default CA file. cadir="~/.mitmproxy/" ) with open('whiteLists.txt', 'r') as ff: domains = ff.xreadlines() for domain in domains: domain = domain.strip().rstrip() if domain.find('#') != -1: continue whitelist.append(domain) print domain print "will be scan." state = flow.State()