def test_utils_friend(self): n = utils.get_number_of_friend(self.user1) self.assertEqual(n, 0) user2 = auth_models.User(npm='2', username='******', angkatan=2011, role='dewa2') user2.save() friend = models.Friendship(user1=self.user1, user2=user2) friend.save() serr = utils.serialize_friendship(friend) self.assertTrue('user1' in serr) self.assertTrue('user2' in serr) query = utils.get_user_friends(user=self.user1) self.assertTrue(bool(query)) user3 = auth_models.User(npm='3', username='******', angkatan=1900, role='mahasiswa') user3.save() friendship = utils.insert_new_friend_to_database(self.user1, user3) self.assertEqual(friendship.user1, self.user1) query = utils.get_all_mahasiswa_except(self.user1) self.assertTrue(len(query), 2)
def post(self, request): print('pos method,用户管理添加保存入库处理') """添加用户""" user_name = request.POST.get("user_name") ready_name = request.POST.get("ready_name") passwd = request.POST.get("passwd") role_id = request.POST.get("role") phone = request.POST.get("phone") email = request.POST.get("email") try: # django自带用户信息表 role_obj = auth_db.Role.objects.get(id=role_id) user_obj = auth_db.User(user_name=user_name, email=email, passwd=passwd, ready_name=ready_name, phone=phone) user_obj.save() #添加用户角色(多对多关系) user_obj = auth_db.User.objects.get(user_name=user_name) user_obj.role.add(role_obj) user_obj.save() data = "用户 %s 添加成功,请刷新查看!" % user_name except Exception as e: data = "用户添加失败:\n %s " % e return HttpResponse(data)
def setUp(self): self.username = env('SSO_USERNAME') self.password = env('SSO_PASSWORD') self.user = models.User(npm='1606918055', angkatan='2016', role='Mahasiswa', username='******') self.user.save()
def test_model_friend(self): user2 = auth_models.User(npm='2', username='******', angkatan=2011, role='dewa2') user2.save() friend = models.Friendship(user1=self.user1, user2=user2) friend.save() self.assertEqual(friend.__str__(), self.user1.npm + ' - ' + user2.npm)
def setUp(self): self.user1 = auth_models.User(npm='1', username='******', angkatan=2016, role='dewa') self.user1.save() self.request = HttpRequest() engine = import_module(settings.SESSION_ENGINE) session_key = None self.request.session = engine.SessionStore(session_key) self.request.session['user_login'] = {'npm': self.user1.npm}
def test_views_friend(self): data = {} views.index(self.request, data) self.assertTrue('number_of_friend' in data) self.request.method = 'GET' resp = views.get_all_friend_candidate(self.request) self.assertEqual(200, resp.status_code) user3 = auth_models.User(npm='3', username='******', angkatan=1900, role='mahasiswa') user3.save() self.request.method = 'POST' self.request.POST['npm'] = user3.npm resp = views.post_new_friend(self.request) self.assertEqual(resp.status_code, 200)