def test_get_groups(self):
group1 = Group.create_group(name="group1")
group2 = Group.create_group(name="group2")
user1 = User.create_user(username="******", email="*****@*****.**",
password="******")
Member.add_member_by_email(group1.G_Id, "*****@*****.**")
Member.add_member_by_email(group2.G_Id, "*****@*****.**")
test = Member.get_groups(user1[0].U_Id)
self.assertEqual(test.count(), 2)
Member.delete_member(group1.G_Id, user1[0].U_Id)
test = Member.get_groups(user1[0].U_Id)
self.assertEqual(test.count(), 1)
def update_or_delete_member(member_id=None):
from models import Member
if request.method == 'PUT':
req = request.get_json()
return Member.update_member(member_id, **req)
else:
return Member.delete_member(member_id)
def expocr_group_delete_member(request):
if request.method == 'GET':
params = request.GET
elif request.method == 'POST':
params = request.POST
g_id = params.get('g_id')
u_id = params.get('u_id')
result = Member.delete_member(g_id, u_id)
data = {}
data['deleted rows'] = result[0]
data['deleted details'] = result[1]
response = HttpResponse(json.dumps(data), content_type='application/json')
return response
def test_get_members(self):
group1 = Group.create_group(name="group1")
user1 = User.create_user(username="******", email="*****@*****.**",
password="******")
User.create_user(username="******", email="*****@*****.**",
password="******")
User.create_user(username="******", email="*****@*****.**",
password="******")
Member.add_member_by_email(group1.G_Id, "*****@*****.**")
Member.add_member_by_email(group1.G_Id, "*****@*****.**")
Member.add_member_by_email(group1.G_Id, "*****@*****.**")
test = Member.get_members(group1.G_Id)
self.assertEqual(len(test), 3)
test = Member.delete_member(group1.G_Id, user1[0].U_Id)
self.assertEqual(len(test), 2)
