|
@@ 858-874 (lines=17) @@
|
| 855 |
|
alive_test_out, |
| 856 |
|
) |
| 857 |
|
|
| 858 |
|
@patch('ospd_openvas.db.KbDB') |
| 859 |
|
def test_prepare_alive_test_no_enum_no_alive_test(self, mock_kb): |
| 860 |
|
w = DummyDaemon() |
| 861 |
|
|
| 862 |
|
t_opt = {'alive_test_methods': '1', 'icmp': '0'} |
| 863 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 864 |
|
|
| 865 |
|
ov_setting = {'some_setting': 1} |
| 866 |
|
|
| 867 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 868 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 869 |
|
|
| 870 |
|
p.scan_id = '456-789' |
| 871 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 872 |
|
p.prepare_alive_test_option_for_openvas() |
| 873 |
|
|
| 874 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 875 |
|
|
| 876 |
|
def test_alive_test_methods_to_bit_field(self): |
| 877 |
|
|
|
@@ 781-796 (lines=16) @@
|
| 778 |
|
|
| 779 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 780 |
|
|
| 781 |
|
@patch('ospd_openvas.db.KbDB') |
| 782 |
|
def test_set_alive_no_invalid_alive_test_no_enum(self, mock_kb): |
| 783 |
|
w = DummyDaemon() |
| 784 |
|
|
| 785 |
|
t_opt = {'alive_test_methods': '1', 'icmp': '-1'} |
| 786 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 787 |
|
|
| 788 |
|
ov_setting = {'some_setting': 1} |
| 789 |
|
|
| 790 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 791 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 792 |
|
p.scan_id = '456-789' |
| 793 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 794 |
|
p.prepare_alive_test_option_for_openvas() |
| 795 |
|
|
| 796 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 797 |
|
|
| 798 |
|
@patch('ospd_openvas.db.KbDB') |
| 799 |
|
def test_set_alive_pinghost(self, mock_kb): |
|
@@ 764-779 (lines=16) @@
|
| 761 |
|
|
| 762 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 763 |
|
|
| 764 |
|
@patch('ospd_openvas.db.KbDB') |
| 765 |
|
def test_set_alive_no_invalid_alive_test(self, mock_kb): |
| 766 |
|
w = DummyDaemon() |
| 767 |
|
|
| 768 |
|
t_opt = {'alive_test': -1} |
| 769 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 770 |
|
|
| 771 |
|
ov_setting = {'some_setting': 1} |
| 772 |
|
|
| 773 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 774 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 775 |
|
p.scan_id = '456-789' |
| 776 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 777 |
|
p.prepare_alive_test_option_for_openvas() |
| 778 |
|
|
| 779 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 780 |
|
|
| 781 |
|
@patch('ospd_openvas.db.KbDB') |
| 782 |
|
def test_set_alive_no_invalid_alive_test_no_enum(self, mock_kb): |
|
@@ 747-762 (lines=16) @@
|
| 744 |
|
|
| 745 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 746 |
|
|
| 747 |
|
@patch('ospd_openvas.db.KbDB') |
| 748 |
|
def test_set_alive_no_setting(self, mock_kb): |
| 749 |
|
w = DummyDaemon() |
| 750 |
|
|
| 751 |
|
t_opt = {} |
| 752 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 753 |
|
|
| 754 |
|
ov_setting = {} |
| 755 |
|
|
| 756 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 757 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 758 |
|
p.scan_id = '456-789' |
| 759 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 760 |
|
p.prepare_alive_test_option_for_openvas() |
| 761 |
|
|
| 762 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 763 |
|
|
| 764 |
|
@patch('ospd_openvas.db.KbDB') |
| 765 |
|
def test_set_alive_no_invalid_alive_test(self, mock_kb): |