def test_user_secrets():
from AutoFocusTagsFeed import LOG
client = Client(api_key='%%This_is_API_key%%',
base_url='url',
verify=False,
proxy=False)
res = LOG(client.headers)
assert "%%This_is_API_key%%" not in res
def test_build_iterator(mocker):
"""
Given:
- Output of the feed API
When:
- When calling fetch_indicators or get_indicators
Then:
- Returns a list of the indicators parsed from the API's response
"""
client = Client(api_key='1234', base_url='url', verify=False, proxy=False)
mocker.patch.object(
client,
'get_tags',
return_value=util_load_json('test_data/all_tags_result.json'))
mocker.patch.object(client, 'get_tag_details', return_value={'tag': []})
indicators = client.build_iterator(is_get_command=True)
assert len(indicators) == 2
def test_incremental_level_fetch(mocker):
context = {'tags_need_to_be_fetched': [], 'time_of_first_fetch': 133033333}
client = Client(api_key='1234', base_url='url', verify=False, proxy=False)
mocker.patch('AutoFocusTagsFeed.get_integration_context',
return_value=context)
mocker.patch('AutoFocusTagsFeed.get_all_updated_tags_since_last_fetch',
return_value=[{
'publictag_name': 'Mock.mocktag1'
}])
mocker.patch.object(
client,
'get_tag_details',
return_value=util_load_json('test_data/tag_details_result.json'))
actual_result = incremental_level_fetch(client)
assert actual_result == [
util_load_json('test_data/tag_details_result.json')
]
def test_fetch_indicators(mocker):
"""
Given:
- Output of the feed API as list
When:
- Fetching indicators from the API
Then:
- Create indicator objects list
"""
client = Client(api_key='1234', base_url='url', verify=False, proxy=False)
mocker.patch.object(
client,
'build_iterator',
return_value=util_load_json('test_data/build_iterator_results.json'))
actual_results = fetch_indicators_command(client,
params={'tlp_color': 'RED'})[0]
expected_results = util_load_json(
'test_data/fetch_indicators_results.json')[0]
assert actual_results["type"] == expected_results["type"]
assert actual_results["value"] == expected_results["value"]
assert actual_results["fields"] == expected_results["fields"]