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)