1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
import unittest
from libnmap.parser import NmapParser
from libnmap.plugins.backendplugin import NmapBackendPlugin
from libnmap.plugins.backendpluginFactory import BackendPluginFactory
class TestNmapBackendPlugin(unittest.TestCase):
"""
This testing class will tests each plugins
The following test need to be done :
- test the factory
- test all the method of the class NmapBackendPlugin:
- Verify implmented/notImplemented
- Verify the behaviour (ie insert must insert)
To support a new plugin or a new way to instanciate a plugin, add a dict
with the necessary parameter in the urls table define in setUp
All testcase must loop thru theses urls to validate a plugins
"""
def setUp(self):
fdir = os.path.dirname(os.path.realpath(__file__))
self.flist_full = [
{"file": "{0}/{1}".format(fdir, "files/2_hosts.xml"), "hosts": 2},
{"file": "{0}/{1}".format(fdir, "files/1_hosts.xml"), "hosts": 1},
{
"file": "{0}/{1}".format(
fdir, "files/1_hosts_banner_ports_notsyn.xml"
),
"hosts": 1,
},
{
"file": "{0}/{1}".format(
fdir, "files/1_hosts_banner_ports.xml"
),
"hosts": 1,
},
{
"file": "{0}/{1}".format(fdir, "files/1_hosts_banner.xml"),
"hosts": 1,
},
{
"file": "{0}/{1}".format(fdir, "files/2_hosts_version.xml"),
"hosts": 2,
},
{
"file": "{0}/{1}".format(fdir, "files/2_tcp_hosts.xml"),
"hosts": 2,
},
{
"file": "{0}/{1}".format(fdir, "files/1_hosts_nohostname.xml"),
"hosts": 1,
},
]
self.flist = self.flist_full
# build a list of NmapReport
self.reportList = []
for testfile in self.flist:
fd = open(testfile["file"], "r")
s = fd.read()
fd.close()
nrp = NmapParser.parse(s)
self.reportList.append(nrp)
self.urls = [
{"plugin_name": "mongodb"},
{
"plugin_name": "sql",
"url": "sqlite:////tmp/reportdb.sql",
"echo": False,
},
{
"plugin_name": "sql",
"url": "mysql+pymysql://root@localhost/poulet",
"echo": False,
},
]
def test_backend_factory(self):
"""test_factory BackendPluginFactory.create(**url)
Invoke factory and test that the object is of the right classes
"""
for url in self.urls:
backend = BackendPluginFactory.create(**url)
self.assertEqual(isinstance(backend, NmapBackendPlugin), True)
className = "Nmap%sPlugin" % url["plugin_name"].title()
self.assertEqual(backend.__class__.__name__, className, True)
def test_backend_insert(self):
"""test_insert
best way to insert is to call save() of nmapreport :P
"""
for nrp in self.reportList:
for url in self.urls:
# create the backend factory object
backend = BackendPluginFactory.create(**url)
# save the report
returncode = nrp.save(backend)
# test return code
self.assertNotEqual(returncode, None)
def test_backend_get(self):
"""test_backend_get
inset all report and save the returned id in a list
then get each id and create a new list of report
compare each report (assume eq)
"""
id_list = []
result_list = []
for url in self.urls:
backend = BackendPluginFactory.create(**url)
for nrp in self.reportList:
id_list.append(nrp.save(backend))
for rep_id in id_list:
result_list.append(backend.get(rep_id))
self.assertEqual(len(result_list), len(self.reportList))
self.assertEqual((result_list), (self.reportList))
id_list = []
result_list = []
def test_backend_getall(self):
pass
def test_backend_delete(self):
"""test_backend_delete
inset all report and save the returned id in a list
for each id remove the item and test if not present
"""
id_list = []
result_list = []
for url in self.urls:
backend = BackendPluginFactory.create(**url)
for nrp in self.reportList:
id_list.append(nrp.save(backend))
for rep_id in id_list:
result_list.append(backend.delete(rep_id))
self.assertEqual(backend.get(rep_id), None)
id_list = []
result_list = []
if __name__ == "__main__":
test_suite = [
"test_backend_factory",
"test_backend_insert",
"test_backend_get",
"test_backend_getall",
"test_backend_delete",
]
suite = unittest.TestSuite(map(TestNmapBackendPlugin, test_suite))
test_result = unittest.TextTestRunner(verbosity=5).run(suite)
|
