sitewide/test/groupadmin_test.py

# 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)