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)