| # 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. |
| |
| """Tests for the project_svc module.""" |
| from __future__ import print_function |
| from __future__ import division |
| from __future__ import absolute_import |
| |
| import six |
| import time |
| import unittest |
| |
| try: |
| from mox3 import mox |
| except ImportError: |
| import mox |
| import mock |
| |
| from google.appengine.ext import testbed |
| |
| from framework import framework_constants |
| from framework import sql |
| from mrproto import project_pb2 |
| from mrproto import user_pb2 |
| from services import config_svc |
| from services import project_svc |
| from testing import fake |
| |
| NOW = 12345678 |
| |
| |
| def MakeProjectService(cache_manager, my_mox): |
| project_service = project_svc.ProjectService(cache_manager) |
| project_service.project_tbl = my_mox.CreateMock(sql.SQLTableManager) |
| project_service.user2project_tbl = my_mox.CreateMock(sql.SQLTableManager) |
| project_service.extraperm_tbl = my_mox.CreateMock(sql.SQLTableManager) |
| project_service.membernotes_tbl = my_mox.CreateMock(sql.SQLTableManager) |
| project_service.usergroupprojects_tbl = my_mox.CreateMock( |
| sql.SQLTableManager) |
| project_service.acexclusion_tbl = my_mox.CreateMock( |
| sql.SQLTableManager) |
| return project_service |
| |
| |
| class ProjectTwoLevelCacheTest(unittest.TestCase): |
| |
| def setUp(self): |
| self.testbed = testbed.Testbed() |
| self.testbed.activate() |
| self.testbed.init_memcache_stub() |
| |
| self.mox = mox.Mox() |
| self.cnxn = self.mox.CreateMock(sql.MonorailConnection) |
| self.cache_manager = fake.CacheManager() |
| self.project_service = MakeProjectService(self.cache_manager, self.mox) |
| |
| def tearDown(self): |
| self.testbed.deactivate() |
| |
| def testDeserializeProjects(self): |
| project_rows = [ |
| ( |
| 123, 'proj1', 'test proj 1', 'test project', 'live', 'anyone', '', |
| '', None, '', 0, 50 * 1024 * 1024, NOW, NOW, None, True, False, |
| False, None, None, None, None, None, None, False), |
| ( |
| 234, 'proj2', 'test proj 2', 'test project', 'live', 'anyone', '', |
| '', None, '', 0, 50 * 1024 * 1024, NOW, NOW, None, True, False, |
| False, None, None, None, None, None, None, True) |
| ] |
| role_rows = [ |
| (123, 111, 'owner'), (123, 444, 'owner'), |
| (123, 222, 'committer'), |
| (123, 333, 'contributor'), |
| (234, 111, 'owner')] |
| extraperm_rows = [] |
| |
| project_dict = self.project_service.project_2lc._DeserializeProjects( |
| project_rows, role_rows, extraperm_rows) |
| |
| six.assertCountEqual(self, [123, 234], list(project_dict.keys())) |
| self.assertEqual(123, project_dict[123].project_id) |
| self.assertEqual('proj1', project_dict[123].project_name) |
| self.assertEqual(NOW, project_dict[123].recent_activity) |
| six.assertCountEqual(self, [111, 444], project_dict[123].owner_ids) |
| six.assertCountEqual(self, [222], project_dict[123].committer_ids) |
| six.assertCountEqual(self, [333], project_dict[123].contributor_ids) |
| self.assertEqual(234, project_dict[234].project_id) |
| six.assertCountEqual(self, [111], project_dict[234].owner_ids) |
| self.assertEqual(False, project_dict[123].issue_notify_always_detailed) |
| self.assertEqual(True, project_dict[234].issue_notify_always_detailed) |
| |
| |
| class UserToProjectIdTwoLevelCacheTest(unittest.TestCase): |
| |
| def setUp(self): |
| self.testbed = testbed.Testbed() |
| self.testbed.activate() |
| self.testbed.init_memcache_stub() |
| |
| self.cnxn = fake.MonorailConnection() |
| self.cache_manager = fake.CacheManager() |
| self.mox = mox.Mox() |
| self.project_service = MakeProjectService(self.cache_manager, self.mox) |
| self.user_to_project_2lc = self.project_service.user_to_project_2lc |
| |
| # Set up DB query mocks. |
| self.cached_user_ids = [100, 101] |
| self.from_db_user_ids = [102, 103] |
| test_table = [ |
| (900, self.cached_user_ids[0]), # Project 900, User 100 |
| (900, self.cached_user_ids[1]), # Project 900, User 101 |
| (901, self.cached_user_ids[0]), # Project 901, User 101 |
| (902, self.from_db_user_ids[0]), # Project 902, User 102 |
| (902, self.from_db_user_ids[1]), # Project 902, User 103 |
| (903, self.from_db_user_ids[0]), # Project 903, User 102 |
| ] |
| self.project_service.user2project_tbl.Select = mock.Mock( |
| return_value=test_table) |
| |
| def tearDown(self): |
| # memcache.flush_all() |
| self.testbed.deactivate() |
| self.mox.UnsetStubs() |
| self.mox.ResetAll() |
| |
| def testGetAll(self): |
| # Cache user 100 and 101. |
| self.user_to_project_2lc.CacheItem(self.cached_user_ids[0], set([900, 901])) |
| self.user_to_project_2lc.CacheItem(self.cached_user_ids[1], set([900])) |
| # Test that other project_ids and user_ids get returned by DB queries. |
| first_hit, first_misses = self.user_to_project_2lc.GetAll( |
| self.cnxn, self.cached_user_ids + self.from_db_user_ids) |
| |
| self.project_service.user2project_tbl.Select.assert_called_once_with( |
| self.cnxn, cols=['project_id', 'user_id']) |
| |
| self.assertEqual( |
| first_hit, { |
| 100: set([900, 901]), |
| 101: set([900]), |
| 102: set([902, 903]), |
| 103: set([902]), |
| }) |
| self.assertEqual([], first_misses) |
| |
| def testGetAllRateLimit(self): |
| test_now = time.time() |
| # Initial request that queries table. |
| self.user_to_project_2lc._GetCurrentTime = mock.Mock( |
| return_value=test_now + 60) |
| self.user_to_project_2lc.GetAll( |
| self.cnxn, self.cached_user_ids + self.from_db_user_ids) |
| |
| # Request a user with no projects right after the last request. |
| self.user_to_project_2lc._GetCurrentTime = mock.Mock( |
| return_value=test_now + 61) |
| second_hit, second_misses = self.user_to_project_2lc.GetAll( |
| self.cnxn, [104]) |
| |
| # Request one more user without project that should make a DB request |
| # because the required rate limit time has passed. |
| self.user_to_project_2lc._GetCurrentTime = mock.Mock( |
| return_value=test_now + 121) |
| third_hit, third_misses = self.user_to_project_2lc.GetAll(self.cnxn, [105]) |
| |
| # Queried only twice because the second request was rate limited. |
| self.assertEqual(self.project_service.user2project_tbl.Select.call_count, 2) |
| |
| # Rate limited response will not return the full table. |
| self.assertEqual(second_hit, { |
| 104: set([]), |
| }) |
| self.assertEqual([], second_misses) |
| self.assertEqual( |
| third_hit, { |
| 100: set([900, 901]), |
| 101: set([900]), |
| 102: set([902, 903]), |
| 103: set([902]), |
| 105: set([]), |
| }) |
| self.assertEqual([], third_misses) |
| |
| |
| class ProjectServiceTest(unittest.TestCase): |
| |
| def setUp(self): |
| self.testbed = testbed.Testbed() |
| self.testbed.activate() |
| self.testbed.init_memcache_stub() |
| |
| self.mox = mox.Mox() |
| self.cnxn = self.mox.CreateMock(sql.MonorailConnection) |
| self.cache_manager = fake.CacheManager() |
| self.config_service = self.mox.CreateMock(config_svc.ConfigService) |
| self.project_service = MakeProjectService(self.cache_manager, self.mox) |
| |
| self.proj1 = fake.Project(project_name='proj1', project_id=123) |
| self.proj2 = fake.Project(project_name='proj2', project_id=234) |
| |
| def tearDown(self): |
| self.testbed.deactivate() |
| self.mox.UnsetStubs() |
| self.mox.ResetAll() |
| |
| def SetUpCreateProject(self): |
| # Check for existing project: there should be none. |
| self.project_service.project_tbl.Select( |
| self.cnxn, cols=['project_name', 'project_id'], |
| project_name=['proj1']).AndReturn([]) |
| |
| # Inserting the project gives the project ID. |
| self.project_service.project_tbl.InsertRow( |
| self.cnxn, project_name='proj1', |
| summary='Test project summary', description='Test project description', |
| home_page=None, docs_url=None, source_url=None, |
| logo_file_name=None, logo_gcs_id=None, |
| state='LIVE', access='ANYONE').AndReturn(123) |
| |
| # Insert the users. There are none. |
| self.project_service.user2project_tbl.InsertRows( |
| self.cnxn, ['project_id', 'user_id', 'role_name'], []) |
| |
| def testCreateProject(self): |
| self.SetUpCreateProject() |
| self.mox.ReplayAll() |
| self.project_service.CreateProject( |
| self.cnxn, 'proj1', owner_ids=[], committer_ids=[], contributor_ids=[], |
| summary='Test project summary', description='Test project description') |
| self.mox.VerifyAll() |
| |
| def SetUpLookupProjectIDs(self): |
| self.project_service.project_tbl.Select( |
| self.cnxn, cols=['project_name', 'project_id'], |
| project_name=['proj2']).AndReturn([('proj2', 234)]) |
| |
| def testLookupProjectIDs(self): |
| self.SetUpLookupProjectIDs() |
| self.project_service.project_names_to_ids.CacheItem('proj1', 123) |
| self.mox.ReplayAll() |
| id_dict = self.project_service.LookupProjectIDs( |
| self.cnxn, ['proj1', 'proj2']) |
| self.mox.VerifyAll() |
| self.assertEqual({'proj1': 123, 'proj2': 234}, id_dict) |
| |
| def testLookupProjectNames(self): |
| self.SetUpGetProjects() # Same as testGetProjects() |
| self.project_service.project_2lc.CacheItem(123, self.proj1) |
| self.mox.ReplayAll() |
| name_dict = self.project_service.LookupProjectNames( |
| self.cnxn, [123, 234]) |
| self.mox.VerifyAll() |
| self.assertEqual({123: 'proj1', 234: 'proj2'}, name_dict) |
| |
| def SetUpGetProjects(self, roles=None, extra_perms=None): |
| project_rows = [ |
| ( |
| 234, 'proj2', 'test proj 2', 'test project', 'live', 'anyone', '', |
| '', None, '', 0, 50 * 1024 * 1024, NOW, NOW, None, True, False, |
| False, None, None, None, None, None, None, False) |
| ] |
| self.project_service.project_tbl.Select( |
| self.cnxn, cols=project_svc.PROJECT_COLS, |
| project_id=[234]).AndReturn(project_rows) |
| self.project_service.user2project_tbl.Select( |
| self.cnxn, cols=['project_id', 'user_id', 'role_name'], |
| project_id=[234]).AndReturn(roles or []) |
| self.project_service.extraperm_tbl.Select( |
| self.cnxn, cols=project_svc.EXTRAPERM_COLS, |
| project_id=[234]).AndReturn(extra_perms or []) |
| |
| def testGetProjects(self): |
| self.project_service.project_2lc.CacheItem(123, self.proj1) |
| self.SetUpGetProjects() |
| self.mox.ReplayAll() |
| project_dict = self.project_service.GetProjects( |
| self.cnxn, [123, 234]) |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [123, 234], list(project_dict.keys())) |
| self.assertEqual('proj1', project_dict[123].project_name) |
| self.assertEqual('proj2', project_dict[234].project_name) |
| |
| def testGetProjects_ExtraPerms(self): |
| self.SetUpGetProjects(extra_perms=[(234, 222, 'BarPerm'), |
| (234, 111, 'FooPerm')]) |
| self.mox.ReplayAll() |
| project_dict = self.project_service.GetProjects(self.cnxn, [234]) |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [234], list(project_dict.keys())) |
| self.assertEqual( |
| [project_pb2.Project.ExtraPerms( |
| member_id=111, perms=['FooPerm']), |
| project_pb2.Project.ExtraPerms( |
| member_id=222, perms=['BarPerm'])], |
| project_dict[234].extra_perms) |
| |
| |
| def testGetVisibleProjects_AnyoneAccessWithUser(self): |
| project_rows = [ |
| ( |
| 234, 'proj2', 'test proj 2', 'test project', 'live', 'anyone', '', |
| '', None, '', 0, 50 * 1024 * 1024, NOW, NOW, None, True, False, |
| False, None, None, None, False) |
| ] |
| |
| self.project_service.project_tbl.Select( |
| self.cnxn, cols=['project_id'], |
| state=project_pb2.ProjectState.LIVE).AndReturn(project_rows) |
| self.SetUpGetProjects() |
| self.mox.ReplayAll() |
| user_a = user_pb2.User(email='a@example.com') |
| project_ids = self.project_service.GetVisibleProjects( |
| self.cnxn, user_a, set([111])) |
| |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [234], project_ids) |
| |
| def testGetVisibleProjects_AnyoneAccessWithAnon(self): |
| project_rows = [ |
| ( |
| 234, 'proj2', 'test proj 2', 'test project', 'live', 'anyone', '', |
| '', None, '', 0, 50 * 1024 * 1024, NOW, NOW, None, True, False, |
| False, None, None, None, None, None, None, False) |
| ] |
| |
| self.project_service.project_tbl.Select( |
| self.cnxn, cols=['project_id'], |
| state=project_pb2.ProjectState.LIVE).AndReturn(project_rows) |
| self.SetUpGetProjects() |
| self.mox.ReplayAll() |
| project_ids = self.project_service.GetVisibleProjects(self.cnxn, None, None) |
| |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [234], project_ids) |
| |
| def testGetVisibleProjects_RestrictedAccessWithMember(self): |
| project_rows = [ |
| ( |
| 234, 'proj2', 'test proj 2', 'test project', 'live', 'members_only', |
| '', '', None, '', 0, 50 * 1024 * 1024, NOW, NOW, None, True, False, |
| False, False, None, None, None, None, None, None, False) |
| ] |
| self.proj2.access = project_pb2.ProjectAccess.MEMBERS_ONLY |
| self.proj2.contributor_ids.append(111) |
| self.project_service.project_2lc.CacheItem(234, self.proj2) |
| |
| self.project_service.project_tbl.Select( |
| self.cnxn, cols=['project_id'], |
| state=project_pb2.ProjectState.LIVE).AndReturn(project_rows) |
| self.mox.ReplayAll() |
| user_a = user_pb2.User(email='a@example.com') |
| project_ids = self.project_service.GetVisibleProjects( |
| self.cnxn, user_a, set([111])) |
| |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [234], project_ids) |
| |
| def testGetVisibleProjects_RestrictedAccessWithNonMember(self): |
| project_rows = [ |
| ( |
| 234, 'proj2', 'test proj 2', 'test project', 'live', 'members_only', |
| '', '', None, '', 0, 50 * 1024 * 1024, NOW, NOW, None, True, False, |
| False, None, None, None, None, None, None, False) |
| ] |
| self.proj2.access = project_pb2.ProjectAccess.MEMBERS_ONLY |
| self.project_service.project_2lc.CacheItem(234, self.proj2) |
| |
| self.project_service.project_tbl.Select( |
| self.cnxn, cols=['project_id'], |
| state=project_pb2.ProjectState.LIVE).AndReturn(project_rows) |
| self.mox.ReplayAll() |
| user_a = user_pb2.User(email='a@example.com') |
| project_ids = self.project_service.GetVisibleProjects( |
| self.cnxn, user_a, set([111])) |
| |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [], project_ids) |
| |
| def testGetVisibleProjects_RestrictedAccessWithAnon(self): |
| project_rows = [ |
| ( |
| 234, 'proj2', 'test proj 2', 'test project', 'live', 'members_only', |
| '', '', None, '', 0, 50 * 1024 * 1024, NOW, NOW, None, True, False, |
| False, None, None, None, None, None, None, False) |
| ] |
| self.proj2.access = project_pb2.ProjectAccess.MEMBERS_ONLY |
| self.project_service.project_2lc.CacheItem(234, self.proj2) |
| |
| self.project_service.project_tbl.Select( |
| self.cnxn, cols=['project_id'], |
| state=project_pb2.ProjectState.LIVE).AndReturn(project_rows) |
| self.mox.ReplayAll() |
| project_ids = self.project_service.GetVisibleProjects(self.cnxn, None, None) |
| |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [], project_ids) |
| |
| def testGetVisibleProjects_RestrictedAccessWithSiteAdmin(self): |
| project_rows = [ |
| ( |
| 234, 'proj2', 'test proj 2', 'test project', 'live', 'members_only', |
| '', '', None, '', 0, 50 * 1024 * 1024, NOW, NOW, None, True, False, |
| False, None, None, None, None, None, None, False) |
| ] |
| self.proj2.access = project_pb2.ProjectAccess.MEMBERS_ONLY |
| self.project_service.project_2lc.CacheItem(234, self.proj2) |
| |
| self.project_service.project_tbl.Select( |
| self.cnxn, cols=['project_id'], |
| state=project_pb2.ProjectState.LIVE).AndReturn(project_rows) |
| self.mox.ReplayAll() |
| user_a = user_pb2.User(email='a@example.com') |
| user_a.is_site_admin = True |
| project_ids = self.project_service.GetVisibleProjects( |
| self.cnxn, user_a, set([111])) |
| |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [234], project_ids) |
| |
| def testGetVisibleProjects_ArchivedProject(self): |
| project_rows = [ |
| ( |
| 234, 'proj2', 'test proj 2', 'test project', 'archived', 'anyone', |
| '', '', None, '', 0, 50 * 1024 * 1024, NOW, NOW, None, True, False, |
| False, None, None, None, None, None, None, False) |
| ] |
| self.proj2.state = project_pb2.ProjectState.ARCHIVED |
| self.project_service.project_2lc.CacheItem(234, self.proj2) |
| |
| self.project_service.project_tbl.Select( |
| self.cnxn, cols=['project_id'], |
| state=project_pb2.ProjectState.LIVE).AndReturn(project_rows) |
| self.mox.ReplayAll() |
| user_a = user_pb2.User(email='a@example.com') |
| project_ids = self.project_service.GetVisibleProjects( |
| self.cnxn, user_a, set([111])) |
| |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [234], project_ids) |
| |
| def testGetProjectsByName(self): |
| self.project_service.project_names_to_ids.CacheItem('proj1', 123) |
| self.project_service.project_2lc.CacheItem(123, self.proj1) |
| self.SetUpLookupProjectIDs() |
| self.SetUpGetProjects() |
| self.mox.ReplayAll() |
| project_dict = self.project_service.GetProjectsByName( |
| self.cnxn, ['proj1', 'proj2']) |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, ['proj1', 'proj2'], list(project_dict.keys())) |
| self.assertEqual(123, project_dict['proj1'].project_id) |
| self.assertEqual(234, project_dict['proj2'].project_id) |
| |
| def SetUpExpungeProject(self): |
| self.project_service.user2project_tbl.Delete( |
| self.cnxn, project_id=234) |
| self.project_service.usergroupprojects_tbl.Delete( |
| self.cnxn, project_id=234) |
| self.project_service.extraperm_tbl.Delete( |
| self.cnxn, project_id=234) |
| self.project_service.membernotes_tbl.Delete( |
| self.cnxn, project_id=234) |
| self.project_service.acexclusion_tbl.Delete( |
| self.cnxn, project_id=234) |
| self.project_service.project_tbl.Delete( |
| self.cnxn, project_id=234) |
| |
| def testExpungeProject(self): |
| self.SetUpExpungeProject() |
| self.mox.ReplayAll() |
| self.project_service.ExpungeProject(self.cnxn, 234) |
| self.mox.VerifyAll() |
| |
| def SetUpUpdateProject(self, project_id, delta): |
| self.project_service.project_tbl.SelectValue( |
| self.cnxn, 'project_name', project_id=project_id).AndReturn('projN') |
| self.project_service.project_tbl.Update( |
| self.cnxn, delta, project_id=project_id, commit=False) |
| self.cnxn.Commit() |
| |
| def testUpdateProject(self): |
| delta = {'summary': 'An even better one-line summary'} |
| self.SetUpUpdateProject(234, delta) |
| self.mox.ReplayAll() |
| self.project_service.UpdateProject( |
| self.cnxn, 234, summary='An even better one-line summary') |
| self.mox.VerifyAll() |
| |
| def testUpdateProject_NotifyAlwaysDetailed(self): |
| delta = {'issue_notify_always_detailed': True} |
| self.SetUpUpdateProject(234, delta) |
| self.mox.ReplayAll() |
| self.project_service.UpdateProject( |
| self.cnxn, 234, issue_notify_always_detailed=True) |
| self.mox.VerifyAll() |
| |
| def SetUpUpdateProjectRoles( |
| self, project_id, owner_ids, committer_ids, contributor_ids): |
| self.project_service.project_tbl.SelectValue( |
| self.cnxn, 'project_name', project_id=project_id).AndReturn('projN') |
| self.project_service.project_tbl.Update( |
| self.cnxn, {'cached_content_timestamp': NOW}, project_id=project_id, |
| commit=False) |
| |
| self.project_service.user2project_tbl.Delete( |
| self.cnxn, project_id=project_id, role_name='owner', commit=False) |
| self.project_service.user2project_tbl.Delete( |
| self.cnxn, project_id=project_id, role_name='committer', commit=False) |
| self.project_service.user2project_tbl.Delete( |
| self.cnxn, project_id=project_id, role_name='contributor', |
| commit=False) |
| |
| self.project_service.user2project_tbl.InsertRows( |
| self.cnxn, ['project_id', 'user_id', 'role_name'], |
| [(project_id, user_id, 'owner') for user_id in owner_ids], |
| commit=False) |
| self.project_service.user2project_tbl.InsertRows( |
| self.cnxn, ['project_id', 'user_id', 'role_name'], |
| [(project_id, user_id, 'committer') for user_id in committer_ids], |
| commit=False) |
| self.project_service.user2project_tbl.InsertRows( |
| self.cnxn, ['project_id', 'user_id', 'role_name'], |
| [(project_id, user_id, 'contributor') for user_id in contributor_ids], |
| commit=False) |
| |
| self.cnxn.Commit() |
| |
| def testUpdateProjectRoles(self): |
| self.SetUpUpdateProjectRoles(234, [111, 222], [333], []) |
| self.mox.ReplayAll() |
| self.project_service.UpdateProjectRoles( |
| self.cnxn, 234, [111, 222], [333], [], now=NOW) |
| self.mox.VerifyAll() |
| |
| def SetUpMarkProjectDeletable(self): |
| delta = { |
| 'project_name': 'DELETABLE_123', |
| 'state': 'deletable', |
| } |
| self.project_service.project_tbl.Update(self.cnxn, delta, project_id=123) |
| self.config_service.InvalidateMemcacheForEntireProject(123) |
| |
| def testMarkProjectDeletable(self): |
| self.SetUpMarkProjectDeletable() |
| self.mox.ReplayAll() |
| self.project_service.MarkProjectDeletable( |
| self.cnxn, 123, self.config_service) |
| self.mox.VerifyAll() |
| |
| def testUpdateRecentActivity_SignificantlyLaterActivity(self): |
| activity_time = NOW + framework_constants.SECS_PER_HOUR * 3 |
| delta = {'recent_activity_timestamp': activity_time} |
| self.SetUpGetProjects() |
| self.SetUpUpdateProject(234, delta) |
| self.mox.ReplayAll() |
| self.project_service.UpdateRecentActivity(self.cnxn, 234, now=activity_time) |
| self.mox.VerifyAll() |
| |
| def testUpdateRecentActivity_NotSignificant(self): |
| activity_time = NOW + 123 |
| self.SetUpGetProjects() |
| # ProjectUpdate is not called. |
| self.mox.ReplayAll() |
| self.project_service.UpdateRecentActivity(self.cnxn, 234, now=activity_time) |
| self.mox.VerifyAll() |
| |
| def SetUpGetUserRolesInAllProjects(self): |
| rows = [ |
| (123, 'committer'), |
| (234, 'owner'), |
| ] |
| self.project_service.user2project_tbl.Select( |
| self.cnxn, cols=['project_id', 'role_name'], |
| user_id={111, 888}).AndReturn(rows) |
| |
| def testGetUserRolesInAllProjects(self): |
| self.SetUpGetUserRolesInAllProjects() |
| self.mox.ReplayAll() |
| actual = self.project_service.GetUserRolesInAllProjects( |
| self.cnxn, {111, 888}) |
| owned_project_ids, membered_project_ids, contrib_project_ids = actual |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [234], owned_project_ids) |
| six.assertCountEqual(self, [123], membered_project_ids) |
| six.assertCountEqual(self, [], contrib_project_ids) |
| |
| def testGetUserRolesInAllProjectsWithoutEffectiveIds(self): |
| self.mox.ReplayAll() |
| actual = self.project_service.GetUserRolesInAllProjects(self.cnxn, {}) |
| owned_project_ids, membered_project_ids, contrib_project_ids = actual |
| self.mox.VerifyAll() |
| six.assertCountEqual(self, [], owned_project_ids) |
| six.assertCountEqual(self, [], membered_project_ids) |
| six.assertCountEqual(self, [], contrib_project_ids) |
| |
| def SetUpUpdateExtraPerms(self): |
| self.project_service.extraperm_tbl.Delete( |
| self.cnxn, project_id=234, user_id=111, commit=False) |
| self.project_service.extraperm_tbl.InsertRows( |
| self.cnxn, project_svc.EXTRAPERM_COLS, |
| [(234, 111, 'SecurityTeam')], commit=False) |
| self.project_service.project_tbl.Update( |
| self.cnxn, {'cached_content_timestamp': NOW}, |
| project_id=234, commit=False) |
| self.cnxn.Commit() |
| |
| def testUpdateExtraPerms(self): |
| self.SetUpGetProjects(roles=[(234, 111, 'owner')]) |
| self.SetUpUpdateExtraPerms() |
| self.mox.ReplayAll() |
| self.project_service.UpdateExtraPerms( |
| self.cnxn, 234, 111, ['SecurityTeam'], now=NOW) |
| self.mox.VerifyAll() |
| |
| def testExpungeUsersInProjects(self): |
| self.project_service.extraperm_tbl.Delete = mock.Mock() |
| self.project_service.acexclusion_tbl.Delete = mock.Mock() |
| self.project_service.membernotes_tbl.Delete = mock.Mock() |
| self.project_service.user2project_tbl.Delete = mock.Mock() |
| |
| user_ids = [111, 222] |
| limit= 16 |
| self.project_service.ExpungeUsersInProjects( |
| self.cnxn, user_ids, limit=limit) |
| |
| call = [mock.call(self.cnxn, user_id=user_ids, limit=limit, commit=False)] |
| self.project_service.extraperm_tbl.Delete.assert_has_calls(call) |
| self.project_service.acexclusion_tbl.Delete.assert_has_calls(call) |
| self.project_service.membernotes_tbl.Delete.assert_has_calls(call) |
| self.project_service.user2project_tbl.Delete.assert_has_calls(call) |