add some tests

account/serializers.py

@@ -2,13 +2,13 @@
 from rest_framework import serializers
 
 from utils.serializers import DateTimeTZField
-from .models import User
+from .models import User, AdminType
 
 
 class UserLoginSerializer(serializers.Serializer):
     username = serializers.CharField(max_length=30)
     password = serializers.CharField(max_length=30)
-    tfa_code = serializers.CharField(min_length=6, max_length=6, required=False)
+    tfa_code = serializers.CharField(min_length=6, max_length=6, required=False, allow_null=True)
 
 
 class UserRegisterSerializer(serializers.Serializer):
@@ -27,11 +27,10 @@ class UserChangePasswordSerializer(serializers.Serializer):
 class UserSerializer(serializers.ModelSerializer):
     create_time = DateTimeTZField()
     last_login = DateTimeTZField()
-    admin_extra_permission = serializers.ListField()
 
     class Meta:
         model = User
-        fields = ["id", "username", "real_name", "email", "admin_type", "admin_extra_permission",
+        fields = ["id", "username", "real_name", "email", "admin_type",
                   "create_time", "last_login", "two_factor_auth", "open_api", "is_disabled"]
 
 
@@ -41,9 +40,7 @@ class EditUserSerializer(serializers.Serializer):
     real_name = serializers.CharField(max_length=30)
     password = serializers.CharField(max_length=30, min_length=6, required=False, default=None)
     email = serializers.EmailField(max_length=254)
-    admin_type = serializers.IntegerField(default=0)
+    admin_type = serializers.ChoiceField(choices=(AdminType.REGULAR_USER, AdminType.ADMIN, AdminType.SUPER_ADMIN))
     open_api = serializers.BooleanField()
     two_factor_auth = serializers.BooleanField()
     is_disabled = serializers.BooleanField()
-    admin_extra_permission = serializers.ListField(required=False, default=[],
-                                                   child=serializers.IntegerField())