|
@@ 806-833 (lines=28) @@
|
| 803 |
|
alive_test_out, |
| 804 |
|
) |
| 805 |
|
|
| 806 |
|
@patch('ospd_openvas.db.KbDB') |
| 807 |
|
def test_prepare_alive_test_not_supplied_as_enum(self, mock_kb): |
| 808 |
|
w = DummyDaemon() |
| 809 |
|
|
| 810 |
|
alive_test_out = [ |
| 811 |
|
"1.3.6.1.4.1.25623.1.0.100315:1:checkbox:Do a TCP ping|||no", |
| 812 |
|
"1.3.6.1.4.1.25623.1.0.100315:2:checkbox:TCP ping tries also TCP-SYN ping|||no", |
| 813 |
|
"1.3.6.1.4.1.25623.1.0.100315:7:checkbox:TCP ping tries only TCP-SYN ping|||no", |
| 814 |
|
"1.3.6.1.4.1.25623.1.0.100315:3:checkbox:Do an ICMP ping|||yes", |
| 815 |
|
"1.3.6.1.4.1.25623.1.0.100315:4:checkbox:Use ARP|||no", |
| 816 |
|
"1.3.6.1.4.1.25623.1.0.100315:5:checkbox:Mark unrechable Hosts as dead (not scanning)|||yes", |
| 817 |
|
] |
| 818 |
|
|
| 819 |
|
t_opt = {'alive_test_methods': '1', 'icmp': '1'} |
| 820 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 821 |
|
|
| 822 |
|
ov_setting = {'some_setting': 1} |
| 823 |
|
|
| 824 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 825 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 826 |
|
|
| 827 |
|
p.scan_id = '456-789' |
| 828 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 829 |
|
p.prepare_alive_test_option_for_openvas() |
| 830 |
|
|
| 831 |
|
p.kbdb.add_scan_preferences.assert_called_with( |
| 832 |
|
p.scan_id, |
| 833 |
|
alive_test_out, |
| 834 |
|
) |
| 835 |
|
|
| 836 |
|
@patch('ospd_openvas.db.KbDB') |
|
@@ 776-803 (lines=28) @@
|
| 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): |
| 778 |
|
w = DummyDaemon() |
| 779 |
|
|
| 780 |
|
alive_test_out = [ |
| 781 |
|
"1.3.6.1.4.1.25623.1.0.100315:1:checkbox:Do a TCP ping|||no", |
| 782 |
|
"1.3.6.1.4.1.25623.1.0.100315:2:checkbox:TCP ping tries also TCP-SYN ping|||no", |
| 783 |
|
"1.3.6.1.4.1.25623.1.0.100315:7:checkbox:TCP ping tries only TCP-SYN ping|||no", |
| 784 |
|
"1.3.6.1.4.1.25623.1.0.100315:3:checkbox:Do an ICMP ping|||yes", |
| 785 |
|
"1.3.6.1.4.1.25623.1.0.100315:4:checkbox:Use ARP|||no", |
| 786 |
|
"1.3.6.1.4.1.25623.1.0.100315:5:checkbox:Mark unrechable Hosts as dead (not scanning)|||yes", |
| 787 |
|
] |
| 788 |
|
|
| 789 |
|
t_opt = {'alive_test': 2} |
| 790 |
|
w.scan_collection.get_target_options = MagicMock(return_value=t_opt) |
| 791 |
|
|
| 792 |
|
ov_setting = {'some_setting': 1} |
| 793 |
|
|
| 794 |
|
with patch.object(Openvas, 'get_settings', return_value=ov_setting): |
| 795 |
|
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, None) |
| 796 |
|
|
| 797 |
|
p.scan_id = '456-789' |
| 798 |
|
p.kbdb.add_scan_preferences = MagicMock() |
| 799 |
|
p.prepare_alive_test_option_for_openvas() |
| 800 |
|
|
| 801 |
|
p.kbdb.add_scan_preferences.assert_called_with( |
| 802 |
|
p.scan_id, |
| 803 |
|
alive_test_out, |
| 804 |
|
) |
| 805 |
|
|
| 806 |
|
@patch('ospd_openvas.db.KbDB') |
greenbone /
ospd-openvas
