|
@@ 836-852 (lines=17) @@
|
| 833 |
|
alive_test_out, |
| 834 |
|
) |
| 835 |
|
|
| 836 |
|
@patch('ospd_openvas.db.KbDB') |
| 837 |
|
def test_prepare_alive_test_no_enum_no_alive_test(self, mock_kb): |
| 838 |
|
w = DummyDaemon() |
| 839 |
|
|
| 840 |
|
t_opt = {'alive_test_methods': '1', 'icmp': '0'} |
| 841 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 842 |
|
|
| 843 |
|
ov_setting = {'some_setting': 1} |
| 844 |
|
|
| 845 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 846 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 847 |
|
|
| 848 |
|
p.scan_id = '456-789' |
| 849 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 850 |
|
p.prepare_alive_test_option_for_openvas() |
| 851 |
|
|
| 852 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 853 |
|
|
| 854 |
|
def test_alive_test_methods_to_bit_field(self): |
| 855 |
|
|
|
@@ 759-774 (lines=16) @@
|
| 756 |
|
|
| 757 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 758 |
|
|
| 759 |
|
@patch('ospd_openvas.db.KbDB') |
| 760 |
|
def test_set_alive_no_invalid_alive_test_no_enum(self, mock_kb): |
| 761 |
|
w = DummyDaemon() |
| 762 |
|
|
| 763 |
|
t_opt = {'alive_test_methods': '1', 'icmp': '-1'} |
| 764 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 765 |
|
|
| 766 |
|
ov_setting = {'some_setting': 1} |
| 767 |
|
|
| 768 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 769 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 770 |
|
p.scan_id = '456-789' |
| 771 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 772 |
|
p.prepare_alive_test_option_for_openvas() |
| 773 |
|
|
| 774 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 775 |
|
|
| 776 |
|
@patch('ospd_openvas.db.KbDB') |
| 777 |
|
def test_set_alive_pinghost(self, mock_kb): |
|
@@ 742-757 (lines=16) @@
|
| 739 |
|
|
| 740 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 741 |
|
|
| 742 |
|
@patch('ospd_openvas.db.KbDB') |
| 743 |
|
def test_set_alive_no_invalid_alive_test(self, mock_kb): |
| 744 |
|
w = DummyDaemon() |
| 745 |
|
|
| 746 |
|
t_opt = {'alive_test': -1} |
| 747 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 748 |
|
|
| 749 |
|
ov_setting = {'some_setting': 1} |
| 750 |
|
|
| 751 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 752 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 753 |
|
p.scan_id = '456-789' |
| 754 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 755 |
|
p.prepare_alive_test_option_for_openvas() |
| 756 |
|
|
| 757 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 758 |
|
|
| 759 |
|
@patch('ospd_openvas.db.KbDB') |
| 760 |
|
def test_set_alive_no_invalid_alive_test_no_enum(self, mock_kb): |
|
@@ 725-740 (lines=16) @@
|
| 722 |
|
|
| 723 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 724 |
|
|
| 725 |
|
@patch('ospd_openvas.db.KbDB') |
| 726 |
|
def test_set_alive_no_setting(self, mock_kb): |
| 727 |
|
w = DummyDaemon() |
| 728 |
|
|
| 729 |
|
t_opt = {} |
| 730 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 731 |
|
|
| 732 |
|
ov_setting = {} |
| 733 |
|
|
| 734 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 735 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 736 |
|
p.scan_id = '456-789' |
| 737 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 738 |
|
p.prepare_alive_test_option_for_openvas() |
| 739 |
|
|
| 740 |
|
p.kbdb.add_scan_preferences.assert_not_called() |
| 741 |
|
|
| 742 |
|
@patch('ospd_openvas.db.KbDB') |
| 743 |
|
def test_set_alive_no_invalid_alive_test(self, mock_kb): |