def get_auth_by_pwd(self, user_pwd):
user_pwd_db = self.conn_read.get(self.user_auth_key)
if not user_pwd_db:
user_pwd_db = self._get_user_pwd_from_db()
if not user_pwd_db:
return False
return user_pwd_db == get_pwd_md5(self.user_name, user_pwd)
def _get_auth(self, fn, kwargs=None):
user = self.get_user_model()
if not user:
return False, 404
if get_pwd_md5(self.user_name, self.user_pwd) != user.user_pwd:
return False, 403
if kwargs:
fn(**kwargs)
else:
fn()
return True, None
import django
from django.http.request import HttpRequest
from django.test.client import Client
from informations.models import UsersModel
from informations.views import UserView, UserPwdView
from tools.utils import get_pwd_md5
django.setup()
user_name = 'test_case_1'
user_pwd = 'password'
user_pwd_db = get_pwd_md5(user_name, user_pwd)
def clean_test_data():
UsersModel.objects.filter(user_name=user_name).delete()
def set_test_data():
UsersModel.objects.create(user_name=user_name, user_pwd=user_pwd_db)
class UserViewTest(unittest.TestCase):
def setUp(self):
unittest.TestCase.setUp(self)
clean_test_data()
def _update_user_pwd(self, user_pwd_new):
UsersModel.objects.filter(user_name=self.user_name).update(
user_pwd=get_pwd_md5(self.user_name, user_pwd_new))
def create_user(self, kwargs):
user = UsersModel.objects.create(
user_name=self.user_name,
user_pwd=get_pwd_md5(self.user_name, self.user_pwd),
**kwargs)
return True, user.get_json_simple()
