def test_custom_headers(self): test_headers = {'X-Test-Header': 'TEST'} base_headers = self.make_req.headers base_headers.update(test_headers) config = {'headers': test_headers} make_request = http.MakeRequest(config=config) self.assertEqual(base_headers, make_request.headers)
def setUp(self): self.url = 'http://example.com' self.logger_patched = mock.patch('cloudlib.http.logger.getLogger') self.logger = self.logger_patched.start() self.logger.return_value = tests.Logger() self.make_req = http.MakeRequest() self.fakehttp = tests.FakeHttp()
def __init__(self, job_args): """ :param job_args: """ self.job_args = job_args self.http = http.MakeRequest() self.shell = shell.ShellCommands(log_name='turbolift', debug=self.job_args.get('debug'))
def test_timeout_set(self): config = {'timeout': 120} make_request = http.MakeRequest(config=config) timeout = make_request.request_kwargs['timeout'] self.assertEqual(config['timeout'], timeout)
def test_enable_debug(self): config = {'debug': True} http.MakeRequest(config=config) debug_level = http.httplib.HTTPConnection.debuglevel self.assertEqual(1, debug_level) http.httplib.HTTPConnection.debuglevel = 0
def __init__(self, job_args): self.job_args = job_args self.req = http.MakeRequest()