|
@@ 828-855 (lines=28) @@
|
| 825 |
|
alive_test_out, |
| 826 |
|
) |
| 827 |
|
|
| 828 |
|
@patch('ospd_openvas.db.KbDB') |
| 829 |
|
def test_prepare_alive_test_not_supplied_as_enum(self, mock_kb): |
| 830 |
|
w = DummyDaemon() |
| 831 |
|
|
| 832 |
|
alive_test_out = [ |
| 833 |
|
"1.3.6.1.4.1.25623.1.0.100315:1:checkbox:Do a TCP ping|||no", |
| 834 |
|
"1.3.6.1.4.1.25623.1.0.100315:2:checkbox:TCP ping tries also TCP-SYN ping|||no", |
| 835 |
|
"1.3.6.1.4.1.25623.1.0.100315:7:checkbox:TCP ping tries only TCP-SYN ping|||no", |
| 836 |
|
"1.3.6.1.4.1.25623.1.0.100315:3:checkbox:Do an ICMP ping|||yes", |
| 837 |
|
"1.3.6.1.4.1.25623.1.0.100315:4:checkbox:Use ARP|||no", |
| 838 |
|
"1.3.6.1.4.1.25623.1.0.100315:5:checkbox:Mark unrechable Hosts as dead (not scanning)|||yes", |
| 839 |
|
] |
| 840 |
|
|
| 841 |
|
t_opt = {'alive_test_methods': '1', 'icmp': '1'} |
| 842 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 843 |
|
|
| 844 |
|
ov_setting = {'some_setting': 1} |
| 845 |
|
|
| 846 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 847 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 848 |
|
|
| 849 |
|
p.scan_id = '456-789' |
| 850 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 851 |
|
p.prepare_alive_test_option_for_openvas() |
| 852 |
|
|
| 853 |
|
p.kbdb.add_scan_preferences.assert_called_with( |
| 854 |
|
p.scan_id, |
| 855 |
|
alive_test_out, |
| 856 |
|
) |
| 857 |
|
|
| 858 |
|
@patch('ospd_openvas.db.KbDB') |
|
@@ 798-825 (lines=28) @@
|
| 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): |
| 800 |
|
w = DummyDaemon() |
| 801 |
|
|
| 802 |
|
alive_test_out = [ |
| 803 |
|
"1.3.6.1.4.1.25623.1.0.100315:1:checkbox:Do a TCP ping|||no", |
| 804 |
|
"1.3.6.1.4.1.25623.1.0.100315:2:checkbox:TCP ping tries also TCP-SYN ping|||no", |
| 805 |
|
"1.3.6.1.4.1.25623.1.0.100315:7:checkbox:TCP ping tries only TCP-SYN ping|||no", |
| 806 |
|
"1.3.6.1.4.1.25623.1.0.100315:3:checkbox:Do an ICMP ping|||yes", |
| 807 |
|
"1.3.6.1.4.1.25623.1.0.100315:4:checkbox:Use ARP|||no", |
| 808 |
|
"1.3.6.1.4.1.25623.1.0.100315:5:checkbox:Mark unrechable Hosts as dead (not scanning)|||yes", |
| 809 |
|
] |
| 810 |
|
|
| 811 |
|
t_opt = {'alive_test': 2} |
| 812 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 813 |
|
|
| 814 |
|
ov_setting = {'some_setting': 1} |
| 815 |
|
|
| 816 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 817 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 818 |
|
|
| 819 |
|
p.scan_id = '456-789' |
| 820 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 821 |
|
p.prepare_alive_test_option_for_openvas() |
| 822 |
|
|
| 823 |
|
p.kbdb.add_scan_preferences.assert_called_with( |
| 824 |
|
p.scan_id, |
| 825 |
|
alive_test_out, |
| 826 |
|
) |
| 827 |
|
|
| 828 |
|
@patch('ospd_openvas.db.KbDB') |
greenbone /
ospd-openvas
