blob: 643b1a6216327db7f7e81efd9c163a52ab07fb61 [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.
"""Unittest for Hotlist People servlet."""
from __future__ import print_function
from __future__ import division
from __future__ import absolute_import
try:
from mox3 import mox
except ImportError:
import mox
import unittest
import logging
import ezt
from testing import fake
from features import hotlistpeople
from framework import permissions
from services import service_manager
from testing import testing_helpers
class HotlistPeopleListTest(unittest.TestCase):
def setUp(self):
self.services = service_manager.Services(
features=fake.FeaturesService(),
project=fake.ProjectService(),
user=fake.UserService(),
usergroup=fake.UserGroupService())
self.owner_user = self.services.user.TestAddUser('buzbuz@gmail.com', 111)
self.editor_user = self.services.user.TestAddUser('monica@gmail.com', 222)
self.non_member_user = self.services.user.TestAddUser(
'who-dis@gmail.com', 333)
self.private_hotlist = self.services.features.TestAddHotlist(
'PrivateHotlist', 'owner only', [111], [222], is_private=True)
self.public_hotlist = self.services.features.TestAddHotlist(
'PublicHotlist', 'everyone', [111], [222], is_private=False)
self.servlet = hotlistpeople.HotlistPeopleList(services=self.services)
self.mox = mox.Mox()
def tearDown(self):
self.mox.UnsetStubs()
self.mox.ResetAll()
def testAssertBasePermission(self):
# owner can view people in private hotlist
mr = testing_helpers.MakeMonorailRequest(
hotlist=self.private_hotlist, perms=permissions.EMPTY_PERMISSIONSET)
mr.auth.effective_ids = {111, 444}
self.servlet.AssertBasePermission(mr)
# editor can view people in private hotlist
mr.auth.effective_ids = {222, 333}
self.servlet.AssertBasePermission(mr)
# non-members cannot view people in private hotlist
mr.auth.effective_ids = {444, 333}
self.assertRaises(permissions.PermissionException,
self.servlet.AssertBasePermission, mr)
# owner can view people in public hotlist
mr = testing_helpers.MakeMonorailRequest(hotlist=self.public_hotlist)
mr.auth.effective_ids = {111, 444}
self.servlet.AssertBasePermission(mr)
# editor can view people in public hotlist
mr.auth.effective_ids = {222, 333}
self.servlet.AssertBasePermission(mr)
# non-members cannot view people in public hotlist
mr.auth.effective_ids = {444, 333}
self.servlet.AssertBasePermission(mr)
def testGatherPageData(self):
mr = testing_helpers.MakeMonorailRequest(
hotlist=self.public_hotlist, perms=permissions.EMPTY_PERMISSIONSET)
mr.auth.user_id = 111
mr.auth.effective_ids = {111}
mr.cnxn = 'fake cnxn'
page_data = self.servlet.GatherPageData(mr)
self.assertEqual(ezt.boolean(True), page_data['offer_membership_editing'])
self.assertEqual(ezt.boolean(False), page_data['offer_remove_self'])
self.assertEqual(page_data['total_num_owners'], 1)
self.assertEqual(page_data['newly_added_views'], [])
self.assertEqual(len(page_data['pagination'].visible_results), 2)
# non-owners cannot edit people list
mr.auth.user_id = 222
mr.auth.effective_ids = {222}
page_data = self.servlet.GatherPageData(mr)
self.assertEqual(ezt.boolean(False), page_data['offer_membership_editing'])
self.assertEqual(ezt.boolean(True), page_data['offer_remove_self'])
mr.auth.user_id = 333
mr.auth.effective_ids = {333}
page_data = self.servlet.GatherPageData(mr)
self.assertEqual(ezt.boolean(False), page_data['offer_membership_editing'])
self.assertEqual(ezt.boolean(False), page_data['offer_remove_self'])
def testProcessFormData_Permission(self):
"""Only owner can change member of hotlist."""
mr = testing_helpers.MakeMonorailRequest(
path='/u/buzbuz@gmail.com/hotlists/PrivateHotlist/people',
hotlist=self.private_hotlist, perms=permissions.EMPTY_PERMISSIONSET)
mr.auth.effective_ids = {111, 444}
self.servlet.ProcessFormData(mr, {})
mr.auth.effective_ids = {222, 444}
self.assertRaises(permissions.PermissionException,
self.servlet.ProcessFormData, mr, {})
def testProcessRemoveMembers(self):
hotlist = self.servlet.services.features.TestAddHotlist(
'HotlistName', 'removing 222, monica', [111], [222])
mr = testing_helpers.MakeMonorailRequest(
path='/u/buzbuz@gmail.com/hotlists/HotlistName/people',
hotlist=hotlist)
mr.hotlist_id = hotlist.hotlist_id
post_data = fake.PostData(
remove = ['monica@gmail.com'])
url = self.servlet.ProcessRemoveMembers(
mr, post_data, '/u/111/hotlists/HotlistName')
self.assertTrue('/u/111/hotlists/HotlistName/people' in url)
self.assertEqual(hotlist.editor_ids, [])
def testProcessRemoveSelf(self):
hotlist = self.servlet.services.features.TestAddHotlist(
'HotlistName', 'self removing 222, monica', [111], [222])
mr = testing_helpers.MakeMonorailRequest(
path='/u/buzbuz@gmail.com/hotlists/HotlistName/people',
hotlist=hotlist)
mr.hotlist_id = hotlist.hotlist_id
mr.cnxn = 'fake cnxn'
# The owner cannot be removed using ProcessRemoveSelf(); this is enforced
# by permission in ProcessFormData, not in the function itself;
# nor may a random user...
mr.auth.user_id = 333
mr.auth.effective_ids = {333}
url = self.servlet.ProcessRemoveSelf(mr, '/u/111/hotlists/HotlistName')
self.assertTrue('/u/111/hotlists/HotlistName/people' in url)
self.assertEqual(hotlist.owner_ids, [111])
self.assertEqual(hotlist.editor_ids, [222])
# ...but an editor can.
mr.auth.user_id = 222
mr.auth.effective_ids = {222}
url = self.servlet.ProcessRemoveSelf(mr, '/u/111/hotlists/HotlistName')
self.assertTrue('/u/111/hotlists/HotlistName/people' in url)
self.assertEqual(hotlist.owner_ids, [111])
self.assertEqual(hotlist.editor_ids, [])
def testProcessAddMembers(self):
hotlist = self.servlet.services.features.TestAddHotlist(
'HotlistName', 'adding 333, who-dis', [111], [222])
mr = testing_helpers.MakeMonorailRequest(
path='/u/buzbuz@gmail.com/hotlists/HotlistName/people',
hotlist=hotlist)
mr.hotlist_id = hotlist.hotlist_id
post_data = fake.PostData(
addmembers = ['who-dis@gmail.com'],
role = ['editor'])
url = self.servlet.ProcessAddMembers(
mr, post_data, '/u/111/hotlists/HotlistName')
self.assertTrue('/u/111/hotlists/HotlistName/people' in url)
self.assertEqual(hotlist.editor_ids, [222, 333])
def testProcessAddMembers_OwnerToEditor(self):
hotlist = self.servlet.services.features.TestAddHotlist(
'HotlistName', 'adding owner 111, buzbuz as editor', [111], [222])
mr = testing_helpers.MakeMonorailRequest(
path='/u/buzbuz@gmail.com/hotlists/HotlistName/people',
hotlist=hotlist)
mr.hotlist_id = hotlist.hotlist_id
addmembers_input = 'buzbuz@gmail.com'
post_data = fake.PostData(
addmembers = [addmembers_input],
role = ['editor'])
self.mox.StubOutWithMock(self.servlet, 'PleaseCorrect')
self.servlet.PleaseCorrect(
mr, initial_add_members=addmembers_input, initially_expand_form=True)
self.mox.ReplayAll()
url = self.servlet.ProcessAddMembers(
mr, post_data, '/u/111/hotlists/HotlistName')
self.mox.VerifyAll()
self.assertEqual(
'Cannot have a hotlist without an owner; please leave at least one.',
mr.errors.addmembers)
self.assertIsNone(url)
# Verify that no changes have actually occurred.
self.assertEqual(hotlist.owner_ids, [111])
self.assertEqual(hotlist.editor_ids, [222])
def testProcessChangeOwnership(self):
hotlist = self.servlet.services.features.TestAddHotlist(
'HotlistName', 'new owner 333, who-dis', [111], [222])
mr = testing_helpers.MakeMonorailRequest(
path='/u/buzbuz@gmail.com/hotlists/HotlistName/people',
hotlist=hotlist)
mr.hotlist_id = hotlist.hotlist_id
post_data = fake.PostData(
changeowners = ['who-dis@gmail.com'],
becomeeditor = ['on'])
url = self.servlet.ProcessChangeOwnership(mr, post_data)
self.assertTrue('/u/333/hotlists/HotlistName/people' in url)
self.assertEqual(hotlist.owner_ids, [333])
self.assertEqual(hotlist.editor_ids, [222, 111])
def testProcessChangeOwnership_UnownedHotlist(self):
hotlist = self.services.features.TestAddHotlist(
'unowned', 'new owner 333, who-dis', [], [222])
mr = testing_helpers.MakeMonorailRequest(
path='/whatever',
hotlist=hotlist)
mr.hotlist_id = hotlist.hotlist_id
post_data = fake.PostData(
changeowners = ['who-dis@gmail.com'],
becomeeditor = ['on'])
self.servlet.ProcessChangeOwnership(mr, post_data)
self.assertEqual([333], mr.hotlist.owner_ids)
def testProcessChangeOwnership_BadEmail(self):
hotlist = self.servlet.services.features.TestAddHotlist(
'HotlistName', 'new owner 333, who-dis', [111], [222])
mr = testing_helpers.MakeMonorailRequest(
path='/u/buzbuz@gmail.com/hotlists/HotlistName/people',
hotlist=hotlist)
mr.hotlist_id = hotlist.hotlist_id
changeowners_input = 'who-dis@gmail.com, extra-email@gmail.com'
post_data = fake.PostData(
changeowners = [changeowners_input],
becomeeditor = ['on'])
self.mox.StubOutWithMock(self.servlet, 'PleaseCorrect')
self.servlet.PleaseCorrect(
mr, initial_new_owner_username=changeowners_input, open_dialog='yes')
self.mox.ReplayAll()
url = self.servlet.ProcessChangeOwnership(mr, post_data)
self.mox.VerifyAll()
self.assertEqual(
'Please add one valid user email.', mr.errors.transfer_ownership)
self.assertIsNone(url)
def testProcessChangeOwnership_DuplicateName(self):
# other_hotlist = self.servlet.services.features.TestAddHotlist(
# 'HotlistName', 'hotlist with same name', [333], [])
# hotlist = self.servlet.services.features.TestAddHotlist(
# 'HotlistName', 'new owner 333, who-dis', [111], [222])
# in the test_hotlists dict of features_service in testing/fake
# 'other_hotlist' is overwritten by 'hotlist'
# TODO(jojwang): edit the fake features_service to allow hotlists
# with the same name but different owners
pass