mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-02-04 07:51:50 +00:00
2c6746ffa0
* Added ip2location.io for IP geolocation lookup. * Removed tab in last line. * Added "ip2location" as maintainer. * Update plugins/modules/ip2locationio_facts.py Co-authored-by: Felix Fontein <felix@fontein.de> * Update plugins/modules/ip2locationio_facts.py Co-authored-by: Felix Fontein <felix@fontein.de> * Update plugins/modules/ip2locationio_facts.py Co-authored-by: Felix Fontein <felix@fontein.de> * Update plugins/modules/ip2locationio_facts.py Co-authored-by: Felix Fontein <felix@fontein.de> * Update plugins/modules/ip2locationio_facts.py Co-authored-by: Felix Fontein <felix@fontein.de> * Update plugins/modules/ip2locationio_facts.py Co-authored-by: Felix Fontein <felix@fontein.de> * Update plugins/modules/ip2locationio_facts.py Co-authored-by: Felix Fontein <felix@fontein.de> * Added "typing" library. * Updated import position. * Reformatted. * Added unit test. * Updated documentation to add "ip" parameter. * Renamed module from "ip2location_facts" to "ip2location_info". * Updated version number. Co-authored-by: Felix Fontein <felix@fontein.de> * Update plugins/modules/ip2location_info.py Co-authored-by: Felix Fontein <felix@fontein.de> * Update plugins/modules/ip2location_info.py Co-authored-by: Felix Fontein <felix@fontein.de> * Updated return definition. * Update BOTMETA.yml to latest module name. * Update plugins/modules/ip2location_info.py Co-authored-by: Felix Fontein <felix@fontein.de> * Update plugins/modules/ip2location_info.py Co-authored-by: Felix Fontein <felix@fontein.de> * Removed extra parameter from "fetch_url". * Fixed "test_ip2location_info.py" with formatter. --------- Co-authored-by: Felix Fontein <felix@fontein.de>
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
# Copyright (c) 2025, IP2Location <support@ip2location.com>
|
|
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import unittest
|
|
from unittest.mock import Mock, patch
|
|
|
|
from ansible_collections.community.general.plugins.modules.ip2location_info import Ip2LocationInfo
|
|
|
|
|
|
IP2LOCATION_DATA = {
|
|
"ip": "8.8.8.8",
|
|
"country_code": "US",
|
|
"country_name": "United States of America",
|
|
"region_name": "California",
|
|
"city_name": "Mountain View",
|
|
"latitude": 37.386051,
|
|
"longitude": -122.083855,
|
|
"zip_code": "94035",
|
|
"time_zone": "-07:00",
|
|
"asn": "15169",
|
|
"as": "Google LLC",
|
|
"is_proxy": False,
|
|
}
|
|
|
|
|
|
class TestIp2LocationInfo(unittest.TestCase):
|
|
def test_get_geo_data_with_ip(self):
|
|
module = Mock()
|
|
module.params = {"timeout": 10, "ip": "8.8.8.8"}
|
|
module.from_json.side_effect = json.loads
|
|
|
|
with patch(
|
|
"ansible_collections.community.general.plugins.modules.ip2location_info.fetch_url"
|
|
) as mock_fetch_url:
|
|
mock_response = Mock()
|
|
mock_response.read.return_value = json.dumps(IP2LOCATION_DATA)
|
|
mock_fetch_url.return_value = (mock_response, {"status": 200})
|
|
|
|
ip2location_info = Ip2LocationInfo(module)
|
|
result = ip2location_info.get_geo_data()
|
|
|
|
self.assertEqual(result["country_code"], "US")
|
|
mock_fetch_url.assert_called_once_with(module, "https://api.ip2location.io/?ip=8.8.8.8", timeout=10)
|