def test_brain_with_proxy():
# Patch and Mock out requests/responses
set_proxies(http_proxy='pass',
https_proxy=None,
all_proxy=None)
config = Config([__name__])
brain = Brain(config)
assert brain._proxy_header() is not None
def test_requests_called_with_no_proxy(mock_get):
# Patch and Mock out requests/responses
response = Mock()
test_json = {'versions': [{'version': 1}], 'state': 'In Progress'}
response.json.return_value = test_json
mock_get.return_value = response
config = Config([__name__, '--predict=4'])
brain = Brain(config)
proxies = brain._proxy_header()
mock_get.assert_called_with(headers=ANY, proxies=proxies, url=ANY)
def test_brain_no_proxy():
config = Config([__name__])
brain = Brain(config)
assert brain._proxy_header() is None
def test_brain_no_proxy():
config = Config([__name__, '--disable-telemetry'])
brain = Brain(config)
assert brain._proxy_header() is None