[python] [test] Mock HTTP response with pre-defined string

mock_http.py

import urllib2
from StringIO import StringIO

SERVER_LIST="""test1.pvt
test2.pvt
test5.pvt
test8.pvt
test7.pvt
"""
ADD_HEADERS="""X-Header-Mock-1: asd"""

class MockHTTPHandler(urllib2.HTTPHandler):
    def http_open(self, req):
        resp = urllib2.addinfourl(StringIO(SERVER_LIST), ADD_HEADERS, req.get_full_url())
        resp.code = 200
        resp.msg = "OK"
        return resp

mock_url_opener = urllib2.build_opener(MockHTTPHandler)
urllib2.install_opener(mock_url_opener)