def test_check_columns_exists(mocker):
"""
Given:
A mock response of a call to https://ipinfo.io/1.1.1.1/json,
And a json of the expected output objects
When:
Calling ip on ip=1.1.1.1
Then:
Validate that the output have the desired columns that were missing from thr readable output.
related issue: https://github.com/demisto/etc/issues/46061
"""
from ipinfo_v2 import Client, ipinfo_ip_command
ip = '1.1.1.1'
mock_response = util_load_json('test_data/ip_1.1.1.1_response.json')
client = Client(api_key='',
base_url='https://ipinfo.io',
verify_certificate=False,
proxy=False,
reliability=DBotScoreReliability.C)
mocker.patch.object(client, 'http_request', return_value=mock_response)
command_results = mocker.patch('ipinfo_v2.CommandResults')
ipinfo_ip_command(client, ip)
assert "lat|lng" in command_results.call_args[1].get("readable_output")
def test_ipinfo_ip_command(mocker):
"""
Given:
A mock response of a call to https://ipinfo.io/1.1.1.1/json,
And a json of the expected output objects
When:
Calling ip on ip=1.1.1.1
Then:
Validate the output compared to the mock output
"""
from ipinfo_v2 import Client, ipinfo_ip_command
ip = '1.1.1.1'
mock_response = util_load_json('test_data/ip_1.1.1.1_response.json')
client = Client(api_key='',
base_url='https://ipinfo.io',
verify_certificate=False,
proxy=False,
reliability=DBotScoreReliability.C)
mocker.patch.object(client, 'http_request', return_value=mock_response)
command_results = mocker.patch('ipinfo_v2.CommandResults')
ipinfo_ip_command(client, ip)
expected_parsed_context = util_load_json(
'test_data/ip_1.1.1.1_command_results.json')
assert command_results.call_args[1].get(
"readable_output") == expected_parsed_context[1].get("HumanReadable")
assert command_results.call_args[1].get("outputs").get(
"Address") == "1.1.1.1"
assert command_results.call_args[1].get("outputs").get(
"Hostname") == "one.one.one.one"
def test_ipinfo_ip_command(requests_mock):
"""
Given:
A mock response of a call to https://ipinfo.io/1.1.1.1/json,
And a json of the expected output objects
When:
Calling ip on ip=1.1.1.1
Then:
Validate the output compared to the mock output
"""
from ipinfo_v2 import Client, ipinfo_ip_command
ip = '1.1.1.1'
mock_response = util_load_json('test_data/ip_1.1.1.1_response.json')
requests_mock.get(f'https://ipinfo.io/{ip}/json', json=mock_response)
client = Client(api_key='',
base_url='https://ipinfo.io',
verify_certificate=False,
proxy=False,
reliability=DBotScoreReliability.C)
command_results = ipinfo_ip_command(client, ip)
parsed_context = [
command_result.to_context() for command_result in command_results
]
expected_parsed_context = util_load_json(
'test_data/ip_1.1.1.1_command_results.json')
assert parsed_context == expected_parsed_context