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()