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