added GeneralHoneypotSerializer (#179)

* added GeneralHoneypot serializer * adjusted EnrichmentViewTestCase
intelowlproject · Jun 8, 2023 · 83dd080 · 83dd080
1 parent df9bcd1
commit 83dd080
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/api/serializers.py b/api/serializers.py
@@ -8,7 +8,17 @@
 logger = logging.getLogger(__name__)
 
 
+class GeneralHoneypotSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = GeneralHoneypot
+
+    def to_representation(self, value):
+        return value.name
+
+
 class IOCSerializer(serializers.ModelSerializer):
+    general_honeypot = GeneralHoneypotSerializer(many=True, read_only=True)
+
     class Meta:
         model = IOC
         exclude = [

diff --git a/tests/test_views.py b/tests/test_views.py
@@ -41,8 +41,8 @@ def test_for_vaild_registered_ip(self):
         self.assertEqual(response.json()["ioc"]["times_seen"], self.ioc.times_seen)
         self.assertEqual(response.json()["ioc"]["log4j"], self.ioc.log4j)
         self.assertEqual(response.json()["ioc"]["cowrie"], self.ioc.cowrie)
-        self.assertEqual(response.json()["ioc"]["general_honeypot"][0], self.heralding.pk)  # FEEDS
-        self.assertEqual(response.json()["ioc"]["general_honeypot"][1], self.ciscoasa.pk)  # FEEDS
+        self.assertEqual(response.json()["ioc"]["general_honeypot"][0], self.heralding.name)  # FEEDS
+        self.assertEqual(response.json()["ioc"]["general_honeypot"][1], self.ciscoasa.name)  # FEEDS
         self.assertEqual(response.json()["ioc"]["scanner"], self.ioc.scanner)
         self.assertEqual(response.json()["ioc"]["payload_request"], self.ioc.payload_request)