2015-08-10 10:07:08 +00:00
|
|
|
# coding=utf-8
|
2015-08-11 06:50:33 +00:00
|
|
|
import json
|
2015-08-09 04:57:59 +00:00
|
|
|
|
2015-08-10 10:07:08 +00:00
|
|
|
from django.core.urlresolvers import reverse
|
|
|
|
|
|
|
|
from rest_framework.test import APITestCase, APIClient
|
|
|
|
|
2015-08-11 06:50:33 +00:00
|
|
|
from account.models import User, REGULAR_USER, ADMIN, SUPER_ADMIN
|
|
|
|
from group.models import Group, UserGroupRelation, JoinGroupRequest
|
|
|
|
|
|
|
|
|
|
|
|
class GroupAPITest(APITestCase):
|
|
|
|
pass
|
2015-08-10 10:07:08 +00:00
|
|
|
|
|
|
|
|
|
|
|
class GroupAdminAPITest(APITestCase):
|
2015-08-11 06:50:33 +00:00
|
|
|
def _create_group(self, name, join_group_setting):
|
|
|
|
group = Group.objects.create(name=name, description="des0",
|
|
|
|
join_group_setting=join_group_setting, visible=True,
|
|
|
|
admin=self.user)
|
|
|
|
return group
|
|
|
|
|
2015-08-10 10:07:08 +00:00
|
|
|
def setUp(self):
|
|
|
|
self.client = APIClient()
|
|
|
|
self.url = reverse('group_admin_api')
|
2015-08-11 06:50:33 +00:00
|
|
|
self.user = User.objects.create(username="test", admin_type=SUPER_ADMIN)
|
|
|
|
self.user.set_password("testaa")
|
|
|
|
self.user.save()
|
|
|
|
self.group = self._create_group("group1", 0)
|
|
|
|
self.client.login(username="test", password="testaa")
|
2015-08-10 10:07:08 +00:00
|
|
|
|
2015-08-11 06:50:33 +00:00
|
|
|
# 以下是创建小组的测试
|
2015-08-10 10:07:08 +00:00
|
|
|
def test_invalid_format(self):
|
|
|
|
data = {"name": "group1"}
|
|
|
|
response = self.client.post(self.url, data=data)
|
|
|
|
self.assertEqual(response.data["code"], 1)
|
|
|
|
|
2015-08-11 06:50:33 +00:00
|
|
|
def test_create_group_successfully(self):
|
|
|
|
data = {"name": "group0", "description": "des0", "join_group_setting": "1"}
|
2015-08-10 10:07:08 +00:00
|
|
|
response = self.client.post(self.url, data=data)
|
|
|
|
self.assertEqual(response.data["code"], 0)
|
|
|
|
|
2015-08-11 06:50:33 +00:00
|
|
|
def test_group_already_exists(self):
|
|
|
|
data = {"name": "group1", "description": "des0", "join_group_setting": "1"}
|
|
|
|
response = self.client.post(self.url, data=data)
|
|
|
|
self.assertEqual(response.data, {"code": 1, "data": u"小组名已经存在"})
|
|
|
|
|
2015-08-10 10:07:08 +00:00
|
|
|
# 以下是修改小组的测试
|
|
|
|
def test_put_invalid_data(self):
|
|
|
|
data = {"name": "group1"}
|
|
|
|
response = self.client.put(self.url, data=data)
|
|
|
|
self.assertEqual(response.data["code"], 1)
|
|
|
|
|
|
|
|
def test_edit_group_does_not_exist(self):
|
2015-08-11 06:50:33 +00:00
|
|
|
data = {"group_id": self.group.id + 1, "name": "group0", "description": "des0",
|
2015-08-10 10:07:08 +00:00
|
|
|
"join_group_setting": 2}
|
|
|
|
response = self.client.put(self.url, data=data)
|
|
|
|
self.assertEqual(response.data, {"code": 1, "data": u"小组不存在"})
|
|
|
|
|
2015-08-11 06:50:33 +00:00
|
|
|
def test_edit_group_successfully(self):
|
|
|
|
data = {"group_id": self.group.id, "name": "group0", "description": "des0",
|
2015-08-10 10:07:08 +00:00
|
|
|
"join_group_setting": 2}
|
|
|
|
response = self.client.put(self.url, data=data)
|
|
|
|
self.assertEqual(response.data["code"], 0)
|
2015-08-11 06:50:33 +00:00
|
|
|
self.assertEqual(response.data["data"]["name"], "group0")
|
|
|
|
self.assertEqual(response.data["data"]["join_group_setting"], 2)
|
|
|
|
|
|
|
|
def test_edit_group_exists(self):
|
|
|
|
group = self._create_group("group2", 1)
|
|
|
|
data = {"group_id": group.id, "name": "group1", "description": "des0",
|
|
|
|
"join_group_setting": 0}
|
|
|
|
response = self.client.put(self.url, data=data)
|
|
|
|
self.assertEqual(response.data, {"code": 1, "data": u"小组名已经存在"})
|
2015-08-10 10:07:08 +00:00
|
|
|
|
|
|
|
# 以下是查询小组列表或者是单个小组时的测试
|
|
|
|
def test_select_group_does_not_exist(self):
|
2015-08-11 06:50:33 +00:00
|
|
|
data = {"group_id": self.group.id + 1}
|
2015-08-10 10:07:08 +00:00
|
|
|
response = self.client.get(self.url, data=data)
|
|
|
|
self.assertEqual(response.data, {"code": 1, "data": u"小组不存在"})
|
|
|
|
|
2015-08-11 06:50:33 +00:00
|
|
|
def test_select_group_successfully(self):
|
|
|
|
data = {"group_id": self.group.id}
|
|
|
|
response = self.client.get(self.url, data=data)
|
|
|
|
self.assertEqual(response.data["code"], 0)
|
|
|
|
|
|
|
|
def test_success_get_all_groups(self):
|
|
|
|
self.assertEqual(self.client.get(self.url).data["code"], 0)
|
|
|
|
|
|
|
|
|
|
|
|
class GroupMemberAdminAPITest(APITestCase):
|
|
|
|
def setUp(self):
|
|
|
|
self.client = APIClient()
|
|
|
|
self.url = reverse('group_member_admin_api')
|
|
|
|
self.user = User.objects.create(username="test", admin_type=SUPER_ADMIN)
|
|
|
|
self.user.set_password("testaa")
|
|
|
|
self.user.save()
|
|
|
|
self.user1 = User.objects.create(username="member1", admin_type=REGULAR_USER)
|
|
|
|
self.user1.set_password("testxx")
|
|
|
|
self.user1.save()
|
2015-08-10 10:07:08 +00:00
|
|
|
self.client.login(username="test", password="testaa")
|
2015-08-11 06:50:33 +00:00
|
|
|
self.group = Group.objects.create(name="group1", description="des1",
|
|
|
|
join_group_setting="1", visible="True",
|
|
|
|
admin=self.user)
|
|
|
|
UserGroupRelation.objects.create(group=self.group, user=self.user1)
|
|
|
|
|
|
|
|
# 以下是查询小组成员的测试
|
|
|
|
def test_missing_parameter(self):
|
|
|
|
self.assertEqual(self.client.get(self.url).data, {"code": 1, "data": u"参数错误"})
|
|
|
|
|
|
|
|
def test_group_does_not_exist(self):
|
|
|
|
data = {"group_id": self.group.id + 1}
|
|
|
|
response = self.client.get(self.url, data=data)
|
|
|
|
self.assertEqual(response.data, {"code": 1, "data": u"小组不存在"})
|
|
|
|
|
|
|
|
def test_get_member_list_successfully(self):
|
|
|
|
data = {"group_id": self.group.id}
|
2015-08-10 10:07:08 +00:00
|
|
|
response = self.client.get(self.url, data=data)
|
|
|
|
self.assertEqual(response.data["code"], 0)
|
|
|
|
|
2015-08-11 06:50:33 +00:00
|
|
|
# 以下是删除小组成员的测试
|
|
|
|
def test_invalid_format(self):
|
|
|
|
data = {"members": [self.user1.id]}
|
|
|
|
response = self.client.put(self.url, data=json.dumps(data), content_type="application/json")
|
|
|
|
self.assertEqual(response.data["code"], 1)
|
|
|
|
|
|
|
|
def test_del_group_does_not_exist(self):
|
|
|
|
data = {"group_id": self.group.id + 1, "members": [self.user1.id]}
|
|
|
|
response = self.client.put(self.url, data=json.dumps(data), content_type="application/json")
|
|
|
|
self.assertEqual(response.data, {"code": 1, "data": u"小组不存在"})
|
|
|
|
|
|
|
|
def test_success_del_members(self):
|
|
|
|
data = {"group_id": self.group.id, "members": [self.user1.id]}
|
|
|
|
response = self.client.put(self.url, data=json.dumps(data), content_type="application/json")
|
|
|
|
self.assertEqual(response.data, {"code": 0, "data": u"删除成功"})
|
|
|
|
try:
|
|
|
|
UserGroupRelation.objects.get(group=self.group, user=self.user1)
|
|
|
|
raise AssertionError()
|
|
|
|
except UserGroupRelation.DoesNotExist:
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
class JoinGroupAPITest(APITestCase):
|
|
|
|
def _create_group(self, name, join_group_setting):
|
|
|
|
group = Group.objects.create(name=name, description="des0",
|
|
|
|
join_group_setting=join_group_setting, visible="True",
|
|
|
|
admin=self.user)
|
|
|
|
return group
|
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
self.client = APIClient()
|
|
|
|
self.url = reverse('group_join_admin_api')
|
|
|
|
self.user = User.objects.create(username="test", admin_type=SUPER_ADMIN)
|
|
|
|
self.user.set_password("testaa")
|
|
|
|
self.user.save()
|
2015-08-10 10:07:08 +00:00
|
|
|
self.client.login(username="test", password="testaa")
|
2015-08-11 06:50:33 +00:00
|
|
|
self.group = self._create_group("group0", 0)
|
2015-08-10 10:07:08 +00:00
|
|
|
|
2015-08-11 06:50:33 +00:00
|
|
|
# 以下是用户要加入某个小组的测试
|
|
|
|
def test_invalid_format(self):
|
|
|
|
data = {"message": "message1"}
|
|
|
|
response = self.client.post(self.url, data=data)
|
|
|
|
self.assertEqual(response.data["code"], 1)
|
|
|
|
|
|
|
|
def test_group_does_not_exist(self):
|
|
|
|
data = {"group_id": self.group.id + 1, "message": "message1"}
|
|
|
|
response = self.client.post(self.url, data=data)
|
|
|
|
self.assertEqual(response.data, {"code": 1, "data": u"小组不存在"})
|
|
|
|
|
|
|
|
def test_join0(self):
|
|
|
|
data = {"group_id": self.group.id, "message": "message0"}
|
|
|
|
response = self.client.post(self.url, data=data)
|
|
|
|
self.assertEqual(response.data, {"code": 0, "data": u"你已经成功的加入该小组"})
|
|
|
|
|
|
|
|
# 再加入一遍 已经是小组成员了
|
|
|
|
data = {"group_id": self.group.id, "message": "message0"}
|
|
|
|
response = self.client.post(self.url, data=data)
|
|
|
|
self.assertEqual(response.data, {"code": 1, "data": u"你已经是小组成员了"})
|
|
|
|
|
|
|
|
def test_success_join1(self):
|
|
|
|
group = self._create_group("group1", 1)
|
|
|
|
data = {"group_id": group.id, "message": "message1"}
|
|
|
|
response = self.client.post(self.url, data=data)
|
|
|
|
self.assertEqual(response.data, {"code": 0, "data": u"申请提交成功,请等待审核"})
|
|
|
|
try:
|
|
|
|
JoinGroupRequest.objects.get(user=self.user, group=group, status=False)
|
|
|
|
except JoinGroupRequest.DoesNotExist:
|
|
|
|
raise AssertionError()
|
2015-08-10 10:07:08 +00:00
|
|
|
|
2015-08-11 06:50:33 +00:00
|
|
|
def test_success_join2(self):
|
|
|
|
group = self._create_group("group2", 2)
|
|
|
|
data = {"group_id": group.id, "message": "message2"}
|
|
|
|
response = self.client.post(self.url, data=data)
|
|
|
|
self.assertEqual(response.data, {"code": 1, "data": u"该小组不允许任何人加入"})
|
|
|
|
|
|
|
|
# 以下是搜索小组的测试
|
|
|
|
def test_error_get_data(self):
|
|
|
|
self.assertEqual(self.client.get(self.url).data["code"], 1)
|
|
|
|
|
|
|
|
def test_query_by_keyword(self):
|
|
|
|
response = self.client.get(self.url + "?keyword=group0")
|
|
|
|
self.assertEqual(response.data["code"], 0)
|