diff options
Diffstat (limited to 'test/units/plugins/lookup/test_url.py')
-rw-r--r-- | test/units/plugins/lookup/test_url.py | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/test/units/plugins/lookup/test_url.py b/test/units/plugins/lookup/test_url.py new file mode 100644 index 0000000..2aa77b3 --- /dev/null +++ b/test/units/plugins/lookup/test_url.py @@ -0,0 +1,26 @@ +# -*- 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 |