added unit test for os fingerprint class

Author: savon-noir

libnmap/objects.py

@@ -101,7 +101,10 @@ def __init__(self, starttime='', endtime='', address=None, status=None,
         self._address = address if address is not None else {}
         self._services = services if services is not None else []
         self._extras = extras if extras is not None else {}
-        self.os = self.NmapOSFingerprint(self._extras['os'])
+        self._osfingerprinted = False
+        if 'os' in self._extras:
+            self.os = self.NmapOSFingerprint(self._extras['os'])
+            self._osfingerprinted = True
 
     def __eq__(self, other):
         """
@@ -312,6 +315,15 @@ def os_match_probabilities(self):
             pass
         return rval
 
+    @property
+    def os_fingerprinted(self):
+        """
+            Specify if the host has OS fingerprint data available
+
+            :return: Boolean
+        """
+        return self._osfingerprinted
+
     @property
     def os_fingerprint(self):
         """

libnmap/test/test_fp.py

@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+
+import unittest
+import os
+from libnmap.parser import NmapParser, NmapParserException
+
+
+class TestNmapFP(unittest.TestCase):
+    def setUp(self):
+        fdir = os.path.dirname(os.path.realpath(__file__))
+        self.flist_full = [{ 'file': "%s/%s" % (fdir, 'files/1_os_banner_scripts.xml'), 'os': 1},
+                { 'file': "%s/%s" % (fdir, 'files/2_hosts_version.xml'), 'os': 1},
+                { 'file': "%s/%s" % (fdir, 'files/1_hosts_banner_ports_notsyn.xml'), 'os': 0},
+                { 'file': "%s/%s" % (fdir, 'files/1_hosts_banner.xml'), 'os': 0},
+                { 'file': "%s/%s" % (fdir, 'files/1_hosts_down.xml'), 'os': 0}]
+        self.flist = self.flist_full
+
+    def test_fp(self):
+        for file_e in self.flist_full:
+            rep = NmapParser.parse_fromfile(file_e['file'])
+            for _host in rep.hosts:
+                if file_e['os'] != 0:
+                    self.assertTrue(_host.os_fingerprinted)
+                elif file_e['os'] == 0:
+                    self.assertFalse(_host.os_fingerprinted)
+                else:
+                    raise Exception
+
+if __name__ == '__main__':
+    test_suite = ['test_fp']
+    suite = unittest.TestSuite(map(TestNmapFP, test_suite))
+    test_result = unittest.TextTestRunner(verbosity=2).run(suite)