fix serilizer fields No indication of range

iast/views/sensitive_info_rule.py

@@ -48,7 +48,10 @@ class SensitiveInfoRuleSerializer(serializers.ModelSerializer):
     pattern_type_name = serializers.SerializerMethodField()
     class Meta:
         model = IastSensitiveInfoRule
-        fields = ['id', 'strategy_name','strategy_id','pattern_type_id','pattern_type_name','pattern','status','latest_time']
+        fields = [
+            'id', 'strategy_name', 'strategy_id', 'pattern_type_id',
+            'pattern_type_name', 'pattern', 'status', 'latest_time'
+        ]
 
     def get_strategy_name(self,obj):
         try:
@@ -89,10 +92,15 @@ def get_url(self, obj):
 
 
 class SensitiveInfoRuleCreateSerializer(serializers.Serializer):
-    strategy_id = serializers.IntegerField(min_value=1, required=True)
-    pattern_type_id = serializers.IntegerField(min_value=1, required=True)
+    strategy_id = serializers.IntegerField(min_value=1,
+                                           max_value=2147483646,
+                                           required=True)
+    pattern_type_id = serializers.IntegerField(min_value=1,
+                                               max_value=2147483646,
+                                               required=True)
     pattern = serializers.CharField(required=True)
-    status = serializers.IntegerField(required=True)
+    status = serializers.ChoiceField(choices=(0, 1), required=True)
+
 
 class _SensitiveInfoArgsSerializer(serializers.Serializer):
     page_size = serializers.IntegerField(default=20,

test/iast/views/test_sensitive_info_rule.py

@@ -0,0 +1,26 @@
+######################################################################
+# @author      : bidaya0 (bidaya0@$HOSTNAME)
+# @file        : test_sensitive_info_rule
+# @created     : 星期一 12月 13, 2021 19:50:46 CST
+#
+# @description :
+######################################################################
+
+
+from rest_framework.test import APITestCase
+from dongtai.models.user import User
+
+class SensitiveInfoRuleTestCase(APITestCase):
+    def setUp(self):
+        self.user = User.objects.filter(pk=1).first()
+        self.client.force_authenticate(user=self.user)
+
+    def test_sensitive_info_rule_create(self):
+        response = self.client.post('/api/v1/sensitive_info_rule',
+                                    data={
+                                        'pattern': '0',
+                                        'pattern_type_id': 1,
+                                        'status': 0,
+                                        'strategy_id': 2147483648
+                                    })
+        assert response.status_code == 200

test/iast/views/test_vul_details.py

@@ -29,8 +29,6 @@ def mockdata(self):
             port=22,
             container=None)
 
-    def tearDown(self):
-        IastServer.objects.filter(pk=self.server.id).delete()
 
     def test_get_server(self):
         obj = VulDetail()

test/iast/views/test_vul_summary.py

@@ -67,5 +67,3 @@ def test_create(self):
         response = self.client.get('/api/v1/vuln/summary')
         assert response.status_code == 200
 
-    def tearDown(self):
-        IastAgent.objects.filter(token='testtoken').delete()