fix tests

Author: Lng88

tests/unit/test_unit.py

@@ -6,7 +6,7 @@
 from unittest.mock import patch
 from utils import process_and_return_exclusions, is_json, add_lines, rgignore_cleanup
 from utils import get_batch_info, truncate_line, get_source_type, open_csv, get_colors
-from utils import write_file, grab_repo_name, get_hannibal_hec_info, TimeFunction, BiasedLanguageLogger
+from utils import write_file, grab_repo_name, get_hec_info, TimeFunction, BiasedLanguageLogger
 from utils import get_line_count
 from run_json import main, rg_search, build_args_dict, process_word_occurrences, process_biased_word_line
 from tools.event2splunk import Event2Splunk
@@ -115,7 +115,7 @@ def test_build_args_dict():
     assert args['path'] == mock_repo_path
     assert args['mode'] == 'check'
     assert args['err_file'] == constants.ERR_FILE
-    assert len(args) == 10
+    assert len(args) == 12
 
 
 def test_process_word_occurrences(batch_info):
@@ -183,14 +183,14 @@ def test_exclusions_if_no_exclude_file():
 
 def test_get_splunk_hec_info():
     with pytest.raises(Exception) as no_hec:
-        get_hannibal_hec_info(None)
-    assert "No Splunk HEC token provided" in str(no_hec.value)
+        get_hec_info(None, None)
+    assert "Missing Splunk HEC token" in str(no_hec.value)
 
 
 def test_event2splunk(mocker):
     logger = BiasedLanguageLogger(
         name='UnitTesting', filename='UnitTesting.log')
-    hec = get_hannibal_hec_info('valid-token')
+    hec = get_hec_info('valid-token', 'https://fakeurl.com:1234')
     event2splunk = Event2Splunk(hec, logger)
     mocker.patch('tools.event2splunk.Event2Splunk._send_batch')
     event2splunk.post_event(payload={}, source='testing', sourcetype='testing')
@@ -237,7 +237,9 @@ def test_main(mock_post_event, mock_close_event):
         'splunk_flag': True,
         'enable_logs': False,
         'err_file': constants.ERR_FILE,
+        'h_endpoint': 'valid-endpoint:1234',
         'splunk_token': 'valid-token',
+        'pz_endpoint': 'valid-endpoint:1234',
         'pzero_token': 'valid-token',
         'github_repo': None
     }