zOMGdev/cppcms: https://github.com/artyom-beilis/cppcms.git

tests/cookie_test.py

44 строки1 KB

Artyom Beilis

Python2 & 3 compatibility code

22 авг 2020, 00:43

22 авг 2020, 00:430aad54c

#!/usr/bin/env python
# coding=utf-8
# vim: tabstop=4 expandtab shiftwidth=4 softtabstop=4

from __future__ import print_function

try:
    import httplib
except:
    import http.client as httplib
import sys
import traceback

def check(name,A,B):
    if A != B:
        print("-----Actual--")
        print(A,"---Expected--")
        print(B,"-------------")
        f1=open('actual.txt','wb')
        f2=open('expected.txt','wb')
        f1.write(A)
        f2.write(B)
        f1.close()
        f2.close()
        traceback.print_tb()
        sys.exit(1)
    else:
        print("Ok:",name)

class Nothing:
    pass

def test_valid(headers,expected):
    h=httplib.HTTPConnection('localhost:8080');
    h.request('GET','/test','',{ 'Cookie': headers } );
    r=h.getresponse()
    check(headers,r.read(),expected)

test_valid(b'test=foo',b'test:foo\n')
test_valid(b'test=',b'test:\n')
test_valid(b'test',b'test:\n')
test_valid(b'test="\xD7\xA9\xD7\x9C\xD7\x95\xD7\x9D"; z=x',b'test:\xD7\xA9\xD7\x9C\xD7\x95\xD7\x9D\nz:x\n');
test_valid(b'test="\xD7\xA9\\"\xD7\x9C\xD7\x95\xD7\x9D"; z=x',b'test:\xD7\xA9"\xD7\x9C\xD7\x95\xD7\x9D\nz:x\n');
test_valid(b'a=\xD7\xA9\xD7\x9C\xD7\x95\xD7\x9D; z=x',b'z:x\n');