mirror of https://github.com/docker/docker-py.git
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import unittest
|
|
import six
|
|
|
|
from docker.utils.proxy import ProxyConfig
|
|
|
|
HTTP = 'http://test:80'
|
|
HTTPS = 'https://test:443'
|
|
FTP = 'ftp://user:password@host:23'
|
|
NO_PROXY = 'localhost,.localdomain'
|
|
CONFIG = ProxyConfig(http=HTTP, https=HTTPS, ftp=FTP, no_proxy=NO_PROXY)
|
|
ENV = {
|
|
'http_proxy': HTTP,
|
|
'HTTP_PROXY': HTTP,
|
|
'https_proxy': HTTPS,
|
|
'HTTPS_PROXY': HTTPS,
|
|
'ftp_proxy': FTP,
|
|
'FTP_PROXY': FTP,
|
|
'no_proxy': NO_PROXY,
|
|
'NO_PROXY': NO_PROXY,
|
|
}
|
|
|
|
|
|
class ProxyConfigTest(unittest.TestCase):
|
|
|
|
def test_from_dict(self):
|
|
config = ProxyConfig.from_dict({
|
|
'httpProxy': HTTP,
|
|
'httpsProxy': HTTPS,
|
|
'ftpProxy': FTP,
|
|
'noProxy': NO_PROXY
|
|
})
|
|
self.assertEqual(CONFIG.http, config.http)
|
|
self.assertEqual(CONFIG.https, config.https)
|
|
self.assertEqual(CONFIG.ftp, config.ftp)
|
|
self.assertEqual(CONFIG.no_proxy, config.no_proxy)
|
|
|
|
def test_new(self):
|
|
config = ProxyConfig()
|
|
self.assertIsNone(config.http)
|
|
self.assertIsNone(config.https)
|
|
self.assertIsNone(config.ftp)
|
|
self.assertIsNone(config.no_proxy)
|
|
|
|
config = ProxyConfig(http='a', https='b', ftp='c', no_proxy='d')
|
|
self.assertEqual(config.http, 'a')
|
|
self.assertEqual(config.https, 'b')
|
|
self.assertEqual(config.ftp, 'c')
|
|
self.assertEqual(config.no_proxy, 'd')
|
|
|
|
def test_truthiness(self):
|
|
assert not ProxyConfig()
|
|
assert ProxyConfig(http='non-zero')
|
|
assert ProxyConfig(https='non-zero')
|
|
assert ProxyConfig(ftp='non-zero')
|
|
assert ProxyConfig(no_proxy='non-zero')
|
|
|
|
def test_environment(self):
|
|
self.assertDictEqual(CONFIG.get_environment(), ENV)
|
|
empty = ProxyConfig()
|
|
self.assertDictEqual(empty.get_environment(), {})
|
|
|
|
def test_inject_proxy_environment(self):
|
|
# Proxy config is non null, env is None.
|
|
self.assertSetEqual(
|
|
set(CONFIG.inject_proxy_environment(None)),
|
|
set(['{}={}'.format(k, v) for k, v in six.iteritems(ENV)]))
|
|
|
|
# Proxy config is null, env is None.
|
|
self.assertIsNone(ProxyConfig().inject_proxy_environment(None), None)
|
|
|
|
env = ['FOO=BAR', 'BAR=BAZ']
|
|
|
|
# Proxy config is non null, env is non null
|
|
actual = CONFIG.inject_proxy_environment(env)
|
|
expected = ['{}={}'.format(k, v) for k, v in six.iteritems(ENV)] + env
|
|
# It's important that the first 8 variables are the ones from the proxy
|
|
# config, and the last 2 are the ones from the input environment
|
|
self.assertSetEqual(set(actual[:8]), set(expected[:8]))
|
|
self.assertSetEqual(set(actual[-2:]), set(expected[-2:]))
|
|
|
|
# Proxy is null, and is non null
|
|
self.assertListEqual(ProxyConfig().inject_proxy_environment(env), env)
|