def test_filter_by_body(): f = context.gen_filter_by_body(['ct', 'sexy']) fn = context.gen_filter_by_body(['nct', 'sexy']) # Test request bodies r = Request() r.start_line = 'GET /sexy HTTP/1.1' r.headers['Header'] = 'sexy' r.body = 'foo' assert not f(r) assert fn(r) r.body = 'sexy' assert f(r) assert not fn(r) # Test response bodies r = Request() rsp = Response() rsp.start_line = 'HTTP/1.1 200 OK' rsp.headers['sexy'] = 'sexy' r.start_line = 'GET /sexy HTTP/1.1' r.headers['Header'] = 'sexy' r.response = rsp assert not f(r) assert fn(r) rsp.body = 'sexy' assert f(r) assert not fn(r)
def test_filter_by_params_post(): f1 = context.gen_filter_by_params(['ct', 'Session']) f2 = context.gen_filter_by_params(['ct', 'Cookie', 'ct', 'CookieVal']) r = Request(('GET / HTTP/1.1\r\n' 'Content-Type: application/x-www-form-urlencoded\r\n\r\n')) r.body = 'foo=bar' assert not f1(r) assert not f2(r) r = Request(('GET / HTTP/1.1\r\n' 'Content-Type: application/x-www-form-urlencoded\r\n\r\n')) r.body = 'Session=bar' assert f1(r) assert not f2(r) r = Request(('GET / HTTP/1.1\r\n' 'Content-Type: application/x-www-form-urlencoded\r\n\r\n')) r.body = 'Session=bar&Cookie=foo' assert f1(r) assert not f2(r) r = Request(('GET / HTTP/1.1\r\n' 'Content-Type: application/x-www-form-urlencoded\r\n\r\n')) r.body = 'Session=bar&CookieThing=CookieValue' assert f1(r) assert f2(r)
def test_filter_by_params_post(): f1 = context.gen_filter_by_params(['ct', 'Session']) f2 = context.gen_filter_by_params(['ct', 'Cookie', 'ct', 'CookieVal']) r = Request(('GET / HTTP/1.1\r\n' 'Content-Type: application/x-www-form-urlencoded\r\n\r\n')) r.body = 'foo=bar' assert not f1(r) assert not f2(r) r = Request(('GET / HTTP/1.1\r\n' 'Content-Type: application/x-www-form-urlencoded\r\n\r\n')) r.body = 'Session=bar' assert f1(r) assert not f2(r) r = Request(('GET / HTTP/1.1\r\n' 'Content-Type: application/x-www-form-urlencoded\r\n\r\n')) r.body = 'Session=bar&Cookie=foo' assert f1(r) assert not f2(r) r = Request(('GET / HTTP/1.1\r\n' 'Content-Type: application/x-www-form-urlencoded\r\n\r\n')) r.body = 'Session=bar&CookieThing=CookieValue' assert f1(r) assert f2(r)
def test_filter_by_body(): f = context.gen_filter_by_body(['ct', 'sexy']) fn = context.gen_filter_by_body(['nct', 'sexy']) # Test request bodies r = Request() r.start_line = 'GET /sexy HTTP/1.1' r.headers['Header'] = 'sexy' r.body = 'foo' assert not f(r) assert fn(r) r.body = 'sexy' assert f(r) assert not fn(r) # Test response bodies r = Request() rsp = Response() rsp.start_line = 'HTTP/1.1 200 OK' rsp.headers['sexy'] = 'sexy' r.start_line = 'GET /sexy HTTP/1.1' r.headers['Header'] = 'sexy' r.response = rsp assert not f(r) assert fn(r) rsp.body = 'sexy' assert f(r) assert not fn(r)
def test_gen_filter_by_all_request(): f = context.gen_filter_by_all(['ct', 'hello']) fn = context.gen_filter_by_all(['nct', 'hello']) # Nowhere r = Request('GET / HTTP/1.1\r\n') assert not f(r) assert fn(r) # Verb r = Request('hello / HTTP/1.1\r\n') assert f(r) assert not fn(r) # Path r = Request('GET /hello HTTP/1.1\r\n') assert f(r) assert not fn(r) # Data r = Request('GET / HTTP/1.1\r\n') r.body = 'hello' assert f(r) assert not fn(r) # Header key r = Request('GET / HTTP/1.1\r\n') r.headers['hello'] = 'goodbye' assert f(r) assert not fn(r) # Header value r = Request('GET / HTTP/1.1\r\n') r.headers['goodbye'] = 'hello' assert f(r) assert not fn(r) # Nowhere in headers r = Request('GET / HTTP/1.1\r\n') r.headers['goodbye'] = 'for real' assert not f(r) assert fn(r) # Cookie key r = Request('GET / HTTP/1.1\r\n') r.cookies['hello'] = 'world' assert f(r) assert not fn(r) # Cookie value r = Request('GET / HTTP/1.1\r\n') r.cookies['world'] = 'hello' assert f(r) assert not fn(r) # Nowhere in cookie r = Request('GET / HTTP/1.1\r\n') r.cookies['world'] = 'sucks' assert not f(r) assert fn(r)
def test_gen_filter_by_all_request(): f = context.gen_filter_by_all(['ct', 'hello']) fn = context.gen_filter_by_all(['nct', 'hello']) # Nowhere r = Request('GET / HTTP/1.1\r\n') assert not f(r) assert fn(r) # Verb r = Request('hello / HTTP/1.1\r\n') assert f(r) assert not fn(r) # Path r = Request('GET /hello HTTP/1.1\r\n') assert f(r) assert not fn(r) # Data r = Request('GET / HTTP/1.1\r\n') r.body = 'hello' assert f(r) assert not fn(r) # Header key r = Request('GET / HTTP/1.1\r\n') r.headers['hello'] = 'goodbye' assert f(r) assert not fn(r) # Header value r = Request('GET / HTTP/1.1\r\n') r.headers['goodbye'] = 'hello' assert f(r) assert not fn(r) # Nowhere in headers r = Request('GET / HTTP/1.1\r\n') r.headers['goodbye'] = 'for real' assert not f(r) assert fn(r) # Cookie key r = Request('GET / HTTP/1.1\r\n') r.cookies['hello'] = 'world' assert f(r) assert not fn(r) # Cookie value r = Request('GET / HTTP/1.1\r\n') r.cookies['world'] = 'hello' assert f(r) assert not fn(r) # Nowhere in cookie r = Request('GET / HTTP/1.1\r\n') r.cookies['world'] = 'sucks' assert not f(r) assert fn(r)
def req(): r = Request() r.start_line = "GET / HTTP/1.1" r.host = "www.ffffff.eeeeee" r.body = "AAAA" return r
def req(): r = Request() r.start_line = 'GET / HTTP/1.1' r.host = 'www.ffffff.eeeeee' r.body = 'AAAA' return r