blob: 9f838a547d0abb37ceac13e53de0840c58f13878 [file] [log] [blame]
# 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)