def test_url(requests_mock):
from CTIX import Client, url_details_command
from CommonServerPython import Common
url_to_check = 'https://www.ibm.com/support/mynotifications/'
mock_response = util_load_json('test_data/url_details.json')
requests_mock.get(f'http://test.com/objects/indicator/?q={url_to_check}',
json=mock_response)
client = Client(base_url=BASE_URL,
access_id=ACCESS_ID,
secret_key=SECRET_KEY,
verify=False,
proxies={})
args = {'url': url_to_check, 'enhanced': False}
response = url_details_command(client, args)
assert response[0].outputs == mock_response["results"][0]
assert response[0].outputs_prefix == 'CTIX.URL'
assert response[0].outputs_key_field == 'name2'
assert isinstance(response, list)
assert len(response) == 1
assert isinstance(response[0].indicator, Common.URL)
assert response[0].indicator.url == url_to_check
def test_domain(requests_mock):
from CTIX import Client, domain_details_command
from CommonServerPython import Common
domain_to_check = 'testing.com'
mock_response = util_load_json('test_data/domain_details.json')
requests_mock.get(
f'http://test.com/objects/indicator/?q={domain_to_check}',
json=mock_response)
client = Client(base_url=BASE_URL,
access_id=ACCESS_ID,
secret_key=SECRET_KEY,
verify=False,
proxies={})
args = {'domain': domain_to_check, 'enhanced': False}
response = domain_details_command(client, args)
assert response[0].outputs == mock_response["results"][0]
assert response[0].outputs_prefix == 'CTIX.Domain'
assert response[0].outputs_key_field == 'name2'
assert isinstance(response, list)
assert len(response) == 1
assert isinstance(response[0].indicator, Common.Domain)
assert response[0].indicator.domain == domain_to_check
def test_file(requests_mock):
from CTIX import Client, file_details_command
from CommonServerPython import Common
file_to_check = '4d552241543b8176a3189864a16b6052f9d163a124291ec9552e1b77'
mock_response = util_load_json('test_data/file_details.json')
requests_mock.get(f'http://test.com/objects/indicator/?q={file_to_check}',
json=mock_response)
client = Client(base_url=BASE_URL,
access_id=ACCESS_ID,
secret_key=SECRET_KEY,
verify=False,
proxies={})
args = {'file': file_to_check, 'enhanced': False}
response = file_details_command(client, args)
assert response[0].outputs == mock_response["results"][0]
assert response[0].outputs_prefix == 'CTIX.File'
assert response[0].outputs_key_field == 'name2'
assert isinstance(response, list)
assert len(response) == 1
assert isinstance(response[0].indicator, Common.File)
assert response[0].indicator.name == file_to_check
def test_create_intel(requests_mock):
from CTIX import Client, create_intel_command
mock_response = util_load_json('test_data/create_intel.json')
requests_mock.post(f'http://test.com/create-intel/', json=mock_response)
client = Client(base_url=BASE_URL,
access_id=ACCESS_ID,
secret_key=SECRET_KEY,
verify=False,
proxies={})
post_data = {
"ips": "1.2.3.4,3.45.56.78",
"urls": "https://abc_test.com,https://test_abc.com"
}
response = create_intel_command(client, post_data)
assert "data", "status" in response.keys()
assert response["status"] == 200
def test_ip_not_found(requests_mock):
from CTIX import Client, ip_details_command
ip_to_check = '1.1.1.1'
mock_response = {"results": []}
requests_mock.get(f'http://test.com/objects/indicator/?q={ip_to_check}',
json=mock_response)
client = Client(base_url=BASE_URL,
access_id=ACCESS_ID,
secret_key=SECRET_KEY,
verify=False,
proxies={})
args = {'ip': ip_to_check, 'enhanced': False}
response = ip_details_command(client, args)
assert response[0].outputs == []
assert response[
0].readable_output == f"No matches found for IP {ip_to_check}"
def test_file_not_found(requests_mock):
from CTIX import Client, file_details_command
file_to_check = '6AD8334857B3F054A9F93BA380B5555B'
mock_response = {"results": []}
requests_mock.get(f'http://test.com/objects/indicator/?q={file_to_check}',
json=mock_response)
client = Client(base_url=BASE_URL,
access_id=ACCESS_ID,
secret_key=SECRET_KEY,
verify=False,
proxies={})
args = {'file': file_to_check, 'enhanced': False}
response = file_details_command(client, args)
assert response[0].outputs == []
assert response[
0].readable_output == f"No matches found for FILE {file_to_check}"