| # Copyright 2016 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Unit test for User Group List servlet.""" |
| from __future__ import print_function |
| from __future__ import division |
| from __future__ import absolute_import |
| |
| import unittest |
| |
| from google.appengine.ext import testbed |
| |
| from framework import permissions |
| from services import service_manager |
| from sitewide import grouplist |
| from testing import fake |
| from testing import testing_helpers |
| |
| |
| class GroupListTest(unittest.TestCase): |
| """Tests for the GroupList servlet.""" |
| |
| def setUp(self): |
| self.services = service_manager.Services( |
| usergroup=fake.UserGroupService()) |
| self.servlet = grouplist.GroupList(services=self.services) |
| self.mr = testing_helpers.MakeMonorailRequest() |
| self.testbed = testbed.Testbed() |
| self.testbed.activate() |
| self.testbed.init_memcache_stub() |
| self.testbed.init_datastore_v3_stub() |
| |
| def tearDown(self): |
| self.testbed.deactivate() |
| |
| def testAssertBasePermission_Anon(self): |
| self.mr.perms = permissions.READ_ONLY_PERMISSIONSET |
| with self.assertRaises(permissions.PermissionException): |
| self.servlet.AssertBasePermission(self.mr) |
| |
| def testAssertBasePermission_RegularUsers(self): |
| self.mr.perms = permissions.COMMITTER_ACTIVE_PERMISSIONSET |
| with self.assertRaises(permissions.PermissionException): |
| self.servlet.AssertBasePermission(self.mr) |
| |
| def testAssertBasePermission_SiteAdmin(self): |
| self.mr.perms = permissions.ADMIN_PERMISSIONSET |
| self.servlet.AssertBasePermission(self.mr) |
| |
| def testGatherPagData_ZeroGroups(self): |
| page_data = self.servlet.GatherPageData(self.mr) |
| self.assertEqual([], page_data['groups']) |
| |
| def testGatherPagData_NonzeroGroups(self): |
| self.services.usergroup.TestAddGroupSettings(777, 'group_a@example.com') |
| self.services.usergroup.TestAddGroupSettings(888, 'group_b@example.com') |
| self.services.usergroup.TestAddMembers(888, [111, 222, 333]) |
| page_data = self.servlet.GatherPageData(self.mr) |
| group_view_a, group_view_b = page_data['groups'] |
| self.assertEqual('group_a@example.com', group_view_a.name) |
| self.assertEqual(0, group_view_a.num_members) |
| self.assertEqual('group_b@example.com', group_view_b.name) |
| self.assertEqual(3, group_view_b.num_members) |
| |
| def testProcessFormData_NoPermission(self): |
| mr = testing_helpers.MakeMonorailRequest( |
| perms=permissions.USER_PERMISSIONSET) |
| post_data = fake.PostData( |
| removebtn=[1]) |
| self.assertRaises(permissions.PermissionException, |
| self.servlet.ProcessFormData, mr, post_data) |
| |
| def testProcessFormData_Normal(self): |
| self.services.usergroup.TestAddGroupSettings( |
| 888, 'group_b@example.com', friend_projects=[789]) |
| self.services.usergroup.TestAddMembers(888, [111, 222, 333]) |
| |
| post_data = fake.PostData( |
| remove=[888], |
| removebtn=[1]) |
| self.servlet.ProcessFormData(self.mr, post_data) |
| self.assertNotIn(888, self.services.usergroup.group_settings) |