OLD | NEW |
(Empty) | |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is govered by a BSD-style |
| 3 # license that can be found in the LICENSE file or at |
| 4 # https://developers.google.com/open-source/licenses/bsd |
| 5 |
| 6 """Unit tests for the componentcreate servlet.""" |
| 7 |
| 8 import unittest |
| 9 |
| 10 from framework import permissions |
| 11 from services import service_manager |
| 12 from testing import fake |
| 13 from testing import testing_helpers |
| 14 from tracker import componentcreate |
| 15 from tracker import tracker_bizobj |
| 16 |
| 17 import webapp2 |
| 18 |
| 19 |
| 20 class ComponentCreateTest(unittest.TestCase): |
| 21 |
| 22 def setUp(self): |
| 23 self.services = service_manager.Services( |
| 24 user=fake.UserService(), |
| 25 config=fake.ConfigService(), |
| 26 project=fake.ProjectService()) |
| 27 self.servlet = componentcreate.ComponentCreate( |
| 28 'req', 'res', services=self.services) |
| 29 self.project = self.services.project.TestAddProject('proj') |
| 30 self.mr = testing_helpers.MakeMonorailRequest( |
| 31 project=self.project, perms=permissions.OWNER_ACTIVE_PERMISSIONSET) |
| 32 self.mr.auth.email = 'b@example.com' |
| 33 self.config = self.services.config.GetProjectConfig( |
| 34 'fake cnxn', self.project.project_id) |
| 35 self.services.config.StoreConfig('fake cnxn', self.config) |
| 36 self.cd = tracker_bizobj.MakeComponentDef( |
| 37 1, self.project.project_id, 'BackEnd', 'doc', False, [], [111L], 0, |
| 38 122L) |
| 39 self.config.component_defs = [self.cd] |
| 40 self.services.user.TestAddUser('a@example.com', 111L) |
| 41 self.services.user.TestAddUser('b@example.com', 122L) |
| 42 |
| 43 def testAssertBasePermission(self): |
| 44 # Anon users can never do it |
| 45 self.mr.perms = permissions.READ_ONLY_PERMISSIONSET |
| 46 self.assertRaises( |
| 47 permissions.PermissionException, |
| 48 self.servlet.AssertBasePermission, self.mr) |
| 49 |
| 50 # Project owner can do it. |
| 51 self.mr.perms = permissions.OWNER_ACTIVE_PERMISSIONSET |
| 52 self.servlet.AssertBasePermission(self.mr) |
| 53 |
| 54 # Project member cannot do it |
| 55 self.mr.perms = permissions.COMMITTER_ACTIVE_PERMISSIONSET |
| 56 self.assertRaises( |
| 57 permissions.PermissionException, |
| 58 self.servlet.AssertBasePermission, self.mr) |
| 59 self.mr.perms = permissions.CONTRIBUTOR_ACTIVE_PERMISSIONSET |
| 60 self.assertRaises( |
| 61 permissions.PermissionException, |
| 62 self.servlet.AssertBasePermission, self.mr) |
| 63 |
| 64 def testGatherPageData_CreatingAtTopLevel(self): |
| 65 page_data = self.servlet.GatherPageData(self.mr) |
| 66 self.assertEqual(self.servlet.PROCESS_TAB_COMPONENTS, |
| 67 page_data['admin_tab_mode']) |
| 68 self.assertIsNone(page_data['parent_path']) |
| 69 |
| 70 def testGatherPageData_CreatingASubComponent(self): |
| 71 self.mr.component_path = 'BackEnd' |
| 72 page_data = self.servlet.GatherPageData(self.mr) |
| 73 self.assertEqual(self.servlet.PROCESS_TAB_COMPONENTS, |
| 74 page_data['admin_tab_mode']) |
| 75 self.assertEqual('BackEnd', page_data['parent_path']) |
| 76 |
| 77 def testProcessFormData_NotFound(self): |
| 78 post_data = fake.PostData( |
| 79 parent_path=['Monitoring'], |
| 80 leaf_name=['Rules'], |
| 81 docstring=['Detecting outages'], |
| 82 deprecated=[False], |
| 83 admins=[''], |
| 84 cc=['']) |
| 85 self.assertRaises( |
| 86 webapp2.HTTPException, |
| 87 self.servlet.ProcessFormData, self.mr, post_data) |
| 88 |
| 89 def testProcessFormData_Normal(self): |
| 90 post_data = fake.PostData( |
| 91 parent_path=['BackEnd'], |
| 92 leaf_name=['DB'], |
| 93 docstring=['A database'], |
| 94 deprecated=[False], |
| 95 admins=[''], |
| 96 cc=['']) |
| 97 url = self.servlet.ProcessFormData(self.mr, post_data) |
| 98 self.assertTrue('/adminComponents?saved=1&' in url) |
| 99 config = self.services.config.GetProjectConfig( |
| 100 self.mr.cnxn, self.mr.project_id) |
| 101 |
| 102 cd = tracker_bizobj.FindComponentDef('BackEnd>DB', config) |
| 103 self.assertEqual('BackEnd>DB', cd.path) |
| 104 self.assertEqual('A database', cd.docstring) |
| 105 self.assertEqual([], cd.admin_ids) |
| 106 self.assertEqual([], cd.cc_ids) |
| 107 self.assertTrue(cd.created > 0) |
| 108 self.assertEqual(122L, cd.creator_id) |
| 109 |
| 110 |
| 111 class CheckComponentNameJSONTest(unittest.TestCase): |
| 112 |
| 113 def setUp(self): |
| 114 self.services = service_manager.Services( |
| 115 config=fake.ConfigService(), |
| 116 project=fake.ProjectService()) |
| 117 self.servlet = componentcreate.CheckComponentNameJSON( |
| 118 'req', 'res', services=self.services) |
| 119 self.project = self.services.project.TestAddProject('proj') |
| 120 self.config = self.services.config.GetProjectConfig( |
| 121 'fake cnxn', self.project.project_id) |
| 122 self.cd = tracker_bizobj.MakeComponentDef( |
| 123 1, self.project.project_id, 'BackEnd', 'doc', False, [], [111L], 0, |
| 124 122L) |
| 125 self.config.component_defs = [self.cd] |
| 126 self.services.config.StoreConfig('fake cnxn', self.config) |
| 127 |
| 128 def testHandleRequest_NewComponent(self): |
| 129 mr = testing_helpers.MakeMonorailRequest( |
| 130 project=self.project, perms=permissions.OWNER_ACTIVE_PERMISSIONSET, |
| 131 path='/p/proj/components/checkname?leaf_name=DB') |
| 132 page_data = self.servlet.HandleRequest(mr) |
| 133 self.assertItemsEqual(['error_message'], page_data.keys()) |
| 134 self.assertIsNone(page_data['error_message']) |
| 135 |
| 136 def testHandleRequest_NameAlreadyUsed(self): |
| 137 mr = testing_helpers.MakeMonorailRequest( |
| 138 project=self.project, perms=permissions.OWNER_ACTIVE_PERMISSIONSET, |
| 139 path='/p/proj/components/checkname?leaf_name=BackEnd') |
| 140 page_data = self.servlet.HandleRequest(mr) |
| 141 self.assertItemsEqual(['error_message'], page_data.keys()) |
| 142 self.assertEqual('That name is already in use.', |
| 143 page_data['error_message']) |
| 144 |
| 145 |
| 146 class ComponentCreateMethodsTest(unittest.TestCase): |
| 147 |
| 148 def setUp(self): |
| 149 self.config = tracker_bizobj.MakeDefaultProjectIssueConfig(789) |
| 150 cd1 = tracker_bizobj.MakeComponentDef( |
| 151 1, 789, 'BackEnd', 'doc', False, [], [111L], 0, 122L) |
| 152 cd2 = tracker_bizobj.MakeComponentDef( |
| 153 2, 789, 'BackEnd>DB', 'doc', True, [], [111L], 0, 122L) |
| 154 self.config.component_defs = [cd1, cd2] |
| 155 |
| 156 def testLeafNameErrorMessage_Invalid(self): |
| 157 self.assertEqual( |
| 158 'Invalid component name', |
| 159 componentcreate.LeafNameErrorMessage('', 'bad name', self.config)) |
| 160 |
| 161 def testLeafNameErrorMessage_AlreadyInUse(self): |
| 162 self.assertEqual( |
| 163 'That name is already in use.', |
| 164 componentcreate.LeafNameErrorMessage('', 'BackEnd', self.config)) |
| 165 self.assertEqual( |
| 166 'That name is already in use.', |
| 167 componentcreate.LeafNameErrorMessage('BackEnd', 'DB', self.config)) |
| 168 |
| 169 def testLeafNameErrorMessage_OK(self): |
| 170 self.assertIsNone( |
| 171 componentcreate.LeafNameErrorMessage('', 'FrontEnd', self.config)) |
| 172 self.assertIsNone( |
| 173 componentcreate.LeafNameErrorMessage('BackEnd', 'Search', self.config)) |
| 174 |
| 175 |
| 176 if __name__ == '__main__': |
| 177 unittest.main() |
OLD | NEW |