From 83dd080d471ccacd421c771ddb98821a1e8a1b44 Mon Sep 17 00:00:00 2001 From: Martina Carella Date: Thu, 8 Jun 2023 10:55:34 +0200 Subject: [PATCH] added GeneralHoneypotSerializer (#179) * added GeneralHoneypot serializer * adjusted EnrichmentViewTestCase --- api/serializers.py | 10 ++++++++++ tests/test_views.py | 4 ++-- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/api/serializers.py b/api/serializers.py index 73549c41..b738dcc1 100644 --- 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 index eb5ea5ae..3c703177 100644 --- 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)