新增修改用户密码和对应的API测试功能
This commit is contained in:
@@ -16,3 +16,9 @@ class UserRegisterSerializer(serializers.Serializer):
|
||||
real_name = serializers.CharField(max_length=30)
|
||||
password = serializers.CharField(max_length=30, min_length=6)
|
||||
|
||||
|
||||
class UserChangePasswordSerializer(serializers.Serializer):
|
||||
username = serializers.CharField(max_length=30)
|
||||
old_password = serializers.CharField(max_length=30, min_length=6)
|
||||
new_password = serializers.CharField(max_length=30, min_length=6)
|
||||
|
||||
|
||||
@@ -78,3 +78,24 @@ class UserRegisterAPITest(APITestCase):
|
||||
response = self.client.post(self.url, data=data)
|
||||
self.assertEqual(response.data, {"code": 1, "data": u"用户名已存在"})
|
||||
|
||||
|
||||
class UserChangePasswordAPITest(APITestCase):
|
||||
def setUp(self):
|
||||
self.client = APIClient()
|
||||
self.url = reverse("user_change_password_api")
|
||||
User.objects.create(username="test", password="aaabbb")
|
||||
|
||||
def test_error_old_password(self):
|
||||
data = {"username": "test", "old_password": "aaaccc", "new_password": "aaaddd"}
|
||||
response = self.client.post(self.url, data=data)
|
||||
self.assertEqual(response.data, {"code": 1, "data": u"密码不正确,请重新修改!"})
|
||||
|
||||
def test_invalid_data_format(self):
|
||||
data = {"username": "test", "old_password": "aaa", "new_password": "aaaddd"}
|
||||
response = self.client.post(self.url, data=data)
|
||||
self.assertEqual(response.data["code"], 1)
|
||||
|
||||
def test_username_does_not_exist(self):
|
||||
data = {"username": "test1", "old_password": "aaabbb", "new_password": "aaaddd"}
|
||||
response = self.client.post(self.url, data=data)
|
||||
self.assertEqual(response.data["code"], 1)
|
||||
@@ -6,7 +6,8 @@ from rest_framework.views import APIView
|
||||
from utils.shortcuts import serializer_invalid_response, error_response, success_response
|
||||
|
||||
from .models import User
|
||||
from .serializers import UserLoginSerializer, UsernameCheckSerializer, UserRegisterSerializer
|
||||
from .serializers import UserLoginSerializer, UsernameCheckSerializer, UserRegisterSerializer, \
|
||||
UserChangePasswordSerializer
|
||||
|
||||
|
||||
class UserLoginAPIView(APIView):
|
||||
@@ -52,12 +53,25 @@ class UserRegisterAPIView(APIView):
|
||||
return serializer_invalid_response(serializer)
|
||||
|
||||
|
||||
class UserChangePasswordView(APIView):
|
||||
def get(self, request):
|
||||
pass
|
||||
|
||||
class UserChangePasswordAPIView(APIView):
|
||||
def post(self, request):
|
||||
pass
|
||||
"""
|
||||
用户修改密码json api接口
|
||||
---
|
||||
request_serializer: UserChangePasswordSerializer
|
||||
"""
|
||||
serializer = UserChangePasswordSerializer(data=request.DATA)
|
||||
if serializer.is_valid():
|
||||
data = serializer.data
|
||||
user = auth.authenticate(username=data["username"], password=data["old_password"])
|
||||
if user:
|
||||
user.set_password(data["new_password"])
|
||||
user.save()
|
||||
return success_response(u"用户密码修改成功!")
|
||||
else:
|
||||
return error_response(u"密码不正确,请重新修改!")
|
||||
else:
|
||||
return serializer_invalid_response(serializer)
|
||||
|
||||
|
||||
class UsernameCheckAPIView(APIView):
|
||||
|
||||
Reference in New Issue
Block a user