ok
Direktori : /usr/lib/python2.7/site-packages/ndg/httpsclient/test/ |
Current File : //usr/lib/python2.7/site-packages/ndg/httpsclient/test/test_utils.py |
"""unit tests module for ndg.httpsclient.utils module PyOpenSSL utility to make a httplib-like interface suitable for use with urllib2 """ __author__ = "P J Kershaw (STFC)" __date__ = "06/01/12" __copyright__ = "(C) 2012 Science and Technology Facilities Council" __license__ = "BSD - see LICENSE file in top-level directory" __contact__ = "Philip.Kershaw@stfc.ac.uk" __revision__ = '$Id$' import unittest import os from OpenSSL import SSL from ndg.httpsclient.test import Constants from ndg.httpsclient.utils import (Configuration, fetch_from_url, open_url, _should_use_proxy) class TestUtilsModule(unittest.TestCase): '''Test ndg.httpsclient.utils module''' def test01_configuration(self): config = Configuration(SSL.Context(SSL.SSLv3_METHOD), True) self.assert_(config.ssl_context) self.assertEquals(config.debug, True) def test02_fetch_from_url(self): config = Configuration(SSL.Context(SSL.SSLv3_METHOD), True) res = fetch_from_url(Constants.TEST_URI, config) self.assert_(res) def test03_open_url(self): config = Configuration(SSL.Context(SSL.SSLv3_METHOD), True) res = open_url(Constants.TEST_URI, config) self.assertEqual(res[0], 200, 'open_url for %r failed' % Constants.TEST_URI) def test04__should_use_proxy(self): if 'no_proxy' in os.environ: no_proxy = os.environ['no_proxy'] del os.environ['no_proxy'] else: no_proxy = None self.assertTrue(_should_use_proxy(Constants.TEST_URI), 'Expecting use proxy = True') os.environ['no_proxy'] = 'localhost,localhost.localdomain' self.assertFalse(_should_use_proxy(Constants.TEST_URI), 'Expecting use proxy = False') if no_proxy is not None: os.environ['no_proxy'] = no_proxy else: del os.environ['no_proxy'] if __name__ == "__main__": unittest.main()