ansible/test/units/plugins/lookup/test_url.py
Sam Doran 8f9cf456b0
url lookup - set default user agent (#72324)
* Add unit tests
* Add note about when default changed
2020-10-28 09:34:06 -04:00

26 lines
899 B
Python

# -*- coding: utf-8 -*-
# Copyright: (c) 2020, Sam Doran <sdoran@redhat.com>
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
import pytest
from ansible.plugins.loader import lookup_loader
@pytest.mark.parametrize(
('kwargs', 'agent'),
(
({}, 'ansible-httpget'),
({'http_agent': 'SuperFox'}, 'SuperFox'),
)
)
def test_user_agent(mocker, kwargs, agent):
mock_open_url = mocker.patch('ansible.plugins.lookup.url.open_url', side_effect=AttributeError('raised intentionally'))
url_lookup = lookup_loader.get('url')
with pytest.raises(AttributeError):
url_lookup.run(['https://nourl'], **kwargs)
assert 'http_agent' in mock_open_url.call_args.kwargs
assert mock_open_url.call_args.kwargs['http_agent'] == agent