modified: objects.py

bmx0r · bmx0r · commit 6ac396017e14 · 2013-05-26T12:06:21.000+02:00
modified:   test/test_backend_plugin_factory.py
diff --git a/libnmap/objects.py b/libnmap/objects.py
@@ -873,6 +873,42 @@ def id(self):
         """
         return hash(1)
 
+    def __eq__(self, other):
+        """
+            Compare eq NmapReport based on :
+
+                - create a diff obj and check the result
+                report are equal if added&changed&removed are empty
+
+            :return: boolean
+        """
+        rval = False
+        if(self.__class__ == other.__class__ and self.id == other.id):
+            diffobj = self.diff(other)
+            rval = (len(diffobj.changed()) == 0 and
+                    len(diffobj.added()) == 0 and
+                    len(diffobj.removed()) == 0
+                    )
+        return rval
+
+    def __ne__(self, other):
+        """
+            Compare ne NmapReport based on:
+
+                - create a diff obj and check the result
+                report are ne if added|changed|removed are not empty
+
+            :return: boolean
+        """
+        rval = True
+        if(self.__class__ == other.__class__ and self.id == other.id):
+            diffobj = self.diff(other)
+            rval = (len(diffobj.changed()) != 0 or
+                    len(diffobj.added()) != 0 or
+                    len(diffobj.removed()) != 0
+                    )
+        return rval
+
     def __repr__(self):
         return "{0} {1} hosts: {2} {3}".format(self._nmaprun, self._scaninfo,
                                                len(self._hosts),
diff --git a/libnmap/test/test_backend_plugin_factory.py b/libnmap/test/test_backend_plugin_factory.py
@@ -103,7 +103,8 @@ def test_backend_get(self):
                 result_list.append(backend.get(rep_id))
             #print result_list[0]
             #print self.reportList[0]
-            #self.assertEqual(len(result_list), len(self.reportList))
+            self.assertEqual(len(result_list), len(self.reportList))
+            self.assertEqual((result_list), (self.reportList))
             #for index in range(len(result_list)):
             #    self.assertEqual(result_list[index], self.reportList[index])
             id_list = []
