new Add test for ASNObject

MISP · Apr 11, 2019 · e8334be · e8334be
1 parent c37eabd
commit e8334be
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/tests/testlive_comprehensive.py b/tests/testlive_comprehensive.py
@@ -21,7 +21,7 @@
 
 try:
     from pymisp import ExpandedPyMISP, MISPEvent, MISPOrganisation, MISPUser, Distribution, ThreatLevel, Analysis, MISPObject
-    from pymisp.tools import CSVLoader, DomainIPObject
+    from pymisp.tools import CSVLoader, DomainIPObject, ASNObject
 except ImportError:
     if sys.version_info < (3, 6):
         print('This test suite requires Python 3.6+, breaking.')
@@ -957,6 +957,19 @@ def test_domain_ip_object(self):
             # Delete event
             self.admin_misp_connector.delete_event(first.id)
 
+    def test_asn_object(self):
+        first = self.create_simple_event()
+        try:
+            dom_ip_obj = ASNObject({'asn': '12345',
+                                    'first-seen': '20190101',
+                                    'last-seen': '2019-02-03'})
+            first.add_object(dom_ip_obj)
+            first = self.user_misp_connector.add_event(first)
+            self.assertEqual(len(first.objects[0].attributes), 3)
+        finally:
+            # Delete event
+            self.admin_misp_connector.delete_event(first.id)
+
     def test_object_template(self):
         r = self.admin_misp_connector.update_object_templates()
         self.assertEqual(type(r), list)