blob: 1322a0fd5a0b8653f591053d0197a12e5b9e71ec [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 admin servlet."""
from __future__ import print_function
from __future__ import division
from __future__ import absolute_import
import unittest
from framework import permissions
from mrproto import usergroup_pb2
from services import service_manager
from sitewide import groupadmin
from testing import fake
from testing import testing_helpers
class GrouAdminTest(unittest.TestCase):
"""Tests for the GroupAdmin servlet."""
def setUp(self):
self.services = service_manager.Services(
user=fake.UserService(),
usergroup=fake.UserGroupService(),
project=fake.ProjectService())
self.services.user.TestAddUser('a@example.com', 111)
self.services.user.TestAddUser('b@example.com', 222)
self.services.user.TestAddUser('c@example.com', 333)
self.services.user.TestAddUser('group@example.com', 888)
self.services.user.TestAddUser('importgroup@example.com', 999)
self.services.usergroup.TestAddGroupSettings(888, 'group@example.com')
self.services.usergroup.TestAddGroupSettings(
999, 'importgroup@example.com', external_group_type='mdb')
self.servlet = groupadmin.GroupAdmin(services=self.services)
self.mr = testing_helpers.MakeMonorailRequest()
self.mr.viewed_username = 'group@example.com'
self.mr.viewed_user_auth.user_id = 888
def testAssertBasePermission(self):
mr = testing_helpers.MakeMonorailRequest(
perms=permissions.GetPermissions(None, {}, None))
mr.viewed_user_auth.user_id = 888
self.assertRaises(
permissions.PermissionException,
self.servlet.AssertBasePermission, mr)
self.services.usergroup.TestAddMembers(888, [111], 'owner')
self.servlet.AssertBasePermission(self.mr)
def testGatherPageData_Normal(self):
page_data = self.servlet.GatherPageData(self.mr)
self.assertEqual('group@example.com', page_data['groupname'])
self.assertEqual('Group Members', page_data['initial_visibility'].name)
self.assertEqual(3, len(page_data['visibility_levels']))
def testGatherPageData_Import(self):
mr = testing_helpers.MakeMonorailRequest()
mr.viewed_username = 'importgroup@example.com'
mr.viewed_user_auth.user_id = 999
page_data = self.servlet.GatherPageData(mr)
self.assertEqual('importgroup@example.com', page_data['groupname'])
self.assertTrue(page_data['import_group'])
self.assertEqual('MDB', page_data['initial_group_type'].name)
def testProcessFormData_Normal(self):
post_data = fake.PostData(visibility='0')
url = self.servlet.ProcessFormData(self.mr, post_data)
self.assertIn('/g/group@example.com/groupadmin', url)
group_settings = self.services.usergroup.GetGroupSettings(None, 888)
self.assertEqual(usergroup_pb2.MemberVisibility.OWNERS,
group_settings.who_can_view_members)
def testProcessFormData_Import(self):
post_data = fake.PostData(
group_type='1', import_group=['on'])
url = self.servlet.ProcessFormData(self.mr, post_data)
self.assertIn('/g/group@example.com/groupadmin', url)
group_settings = self.services.usergroup.GetGroupSettings(None, 888)
self.assertEqual(usergroup_pb2.MemberVisibility.OWNERS,
group_settings.who_can_view_members)
self.assertEqual(usergroup_pb2.GroupType.MDB,
group_settings.ext_group_type)