fix:issue-312 change validation logic in vul serializers

iast/serializers/vul.py

@@ -33,11 +33,11 @@ def split_container_name(name):
         if name is None:
             return ""
         if '/' in name:
-            result = name.split('/')[0].lower().strip()
+            return name.split('/')[0].lower().strip()
         elif ' ' in name:
             names = name.split(' ')[:-1]
-            result = ' '.join(names).lower().strip()
-        return result
+            return ' '.join(names).lower().strip()
+        return name
 
     def get_language(self, obj):
         if obj['agent_id'] not in self.AGENT_LANGUAGE_MAP:

iast/utils.py

@@ -72,7 +72,7 @@ def extend_schema_with_envcheck(querys: list = [],
                                 **kwargs):
     def myextend_schema(func):
         import os
-        if os.getenv('environment', None) in ('TEST', 'DOC'):
+        if os.getenv('environment', None) in ('TEST', 'DOC') or os.getenv('DOC', None) == 'TRUE' :
             from drf_spectacular.utils import extend_schema, OpenApiParameter, OpenApiExample, OpenApiTypes
             from drf_spectacular.utils import OpenApiResponse
             parameters = list(filter(lambda x: x, map(_filter_query, querys)))

test/iast/seriliazer/test_vul_details.py

@@ -0,0 +1,37 @@
+######################################################################
+# @author      : bidaya0 (bidaya0@$HOSTNAME)
+# @file        : test_vul_details
+# @created     : 星期四 12月 09, 2021 10:23:10 CST
+#
+# @description :
+######################################################################
+
+
+
+from django.test import TestCase
+from django.urls import include, path, reverse
+from dongtai.models.user import User
+from dongtai.models.vulnerablity import IastVulnerabilityModel
+from dongtai.models.hook_type import HookType
+import time
+from ddt import ddt, data, unpack
+from iast.serializers.vul import VulSerializer
+
+TEST_DATA = ('', None, 'Django', 'Apache Tomcat/9.0.37', 'Tomcat/8.x',
+             'php-fpm', 'WebLogic')
+
+@ddt
+class VulTestCase(TestCase):
+    def setUp(self):
+        pass
+
+    @data(*TEST_DATA)
+    def test_vul(self, value):
+        try:
+            res = VulSerializer.split_container_name(value)
+        except Exception as e:
+            self.fail("raised Exception:{}".format(e))
+        assert isinstance(res, str)
+
+    def tearDown(self):
+        pass