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

tests/gzip_test.py

57 строк1 KB

Artyom Beilis

added setbuf API and blocking/nonblocking modes

19 ноя 2015, 19:13

19 ноя 2015, 19:13b83f40d

#!/usr/bin/env python
# coding=utf-8

# vim: tabstop=4 expandtab shiftwidth=4 softtabstop=4 

import httplib
import sys
import gzip
import StringIO

def test(name,A,B):
    if A != B:
        print "Error :" + name 
        print "-----Actual--"
        print A,"---Expected--"
        print B,"-------------"
        sys.exit(1)
    else:
        print "Ok:"+name

def decompress(content):
    virt = StringIO.StringIO(content)
    zstream = gzip.GzipFile(mode='r', fileobj=virt)
    result = zstream.read()
    return result

def big():
    slist=[]
    for x in xrange(0,100000):
        slist.append(str(x)+'\n')
    return ''.join(slist)


def test_valid(url,accepts,compressed,expected):
    h=httplib.HTTPConnection('localhost:8080');
    headers = {}
    if accepts:
        headers = { 'Accept-Encoding' : 'gzip'  }
    h.request('GET','/test' + url,'',headers)
    r=h.getresponse().read()
    if compressed:
        r=decompress(r)
    test(url,r,expected)

test_valid('/ca',True,True,'test a')
test_valid('/ca',False,False,'test a')

test_valid('/cb',True,True,'test b')
test_valid('/cb',False,False,'test b')
bg=big()
test_valid('/cbig',True,True,bg)
test_valid('/cbig',False,False,bg)

test_valid('/bin',True,False,'binary')
test_valid('/bin',False,False,'binary')
test_valid('/not',True,False,'not compressed')
test_valid('/not',False,False,'not compressed')