def test_head(self):
http = RequestsWrapper({}, {})
with mock.patch('requests.head'):
http.head('https://www.google.com')
requests.head.assert_called_once_with('https://www.google.com',
**http.options)
def test_head_session_option_override(self):
http = RequestsWrapper({}, {})
options = {'auth': ('user', 'pass')}
with mock.patch('datadog_checks.base.utils.http.RequestsWrapper.session'):
http.head('https://www.google.com', persist=True, auth=options['auth'])
http.session.head.assert_called_once_with('https://www.google.com', **options)
def test_head_session(self):
http = RequestsWrapper({'persist_connections': True}, {})
with mock.patch(
'datadog_checks.base.utils.http.RequestsWrapper.session'):
http.head('https://www.google.com')
http.session.head.assert_called_once_with('https://www.google.com')
def test_head_option_override(self):
http = RequestsWrapper({}, {})
options = http.options.copy()
options['auth'] = ('user', 'pass')
with mock.patch('requests.head'):
http.head('https://www.google.com', auth=options['auth'])
requests.head.assert_called_once_with('https://www.google.com', **options)
