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 fieldcreate servlet.""" |
| 7 |
| 8 import unittest |
| 9 |
| 10 from framework import permissions |
| 11 from proto import tracker_pb2 |
| 12 from services import service_manager |
| 13 from testing import fake |
| 14 from testing import testing_helpers |
| 15 from tracker import fieldcreate |
| 16 from tracker import tracker_bizobj |
| 17 |
| 18 |
| 19 class FieldCreateTest(unittest.TestCase): |
| 20 |
| 21 def setUp(self): |
| 22 self.services = service_manager.Services( |
| 23 user=fake.UserService(), |
| 24 config=fake.ConfigService(), |
| 25 project=fake.ProjectService()) |
| 26 self.servlet = fieldcreate.FieldCreate( |
| 27 'req', 'res', services=self.services) |
| 28 self.project = self.services.project.TestAddProject('proj') |
| 29 self.mr = testing_helpers.MakeMonorailRequest( |
| 30 project=self.project, perms=permissions.OWNER_ACTIVE_PERMISSIONSET) |
| 31 |
| 32 def testAssertBasePermission(self): |
| 33 # Anon users can never do it |
| 34 self.mr.perms = permissions.READ_ONLY_PERMISSIONSET |
| 35 self.assertRaises( |
| 36 permissions.PermissionException, |
| 37 self.servlet.AssertBasePermission, self.mr) |
| 38 |
| 39 # Project owner can do it. |
| 40 self.mr.perms = permissions.OWNER_ACTIVE_PERMISSIONSET |
| 41 self.servlet.AssertBasePermission(self.mr) |
| 42 |
| 43 # Project member cannot do it |
| 44 self.mr.perms = permissions.COMMITTER_ACTIVE_PERMISSIONSET |
| 45 self.assertRaises( |
| 46 permissions.PermissionException, |
| 47 self.servlet.AssertBasePermission, self.mr) |
| 48 self.mr.perms = permissions.CONTRIBUTOR_ACTIVE_PERMISSIONSET |
| 49 self.assertRaises( |
| 50 permissions.PermissionException, |
| 51 self.servlet.AssertBasePermission, self.mr) |
| 52 |
| 53 def testGatherPageData(self): |
| 54 page_data = self.servlet.GatherPageData(self.mr) |
| 55 self.assertEqual(self.servlet.PROCESS_TAB_LABELS, |
| 56 page_data['admin_tab_mode']) |
| 57 self.assertItemsEqual( |
| 58 ['Defect', 'Enhancement', 'Task', 'Other'], |
| 59 page_data['well_known_issue_types']) |
| 60 |
| 61 def testProcessFormData(self): |
| 62 post_data = fake.PostData( |
| 63 name=['somefield'], |
| 64 field_type=['INT_TYPE'], |
| 65 min_value=['1'], |
| 66 max_value=['99'], |
| 67 notify_on=['any_comment'], |
| 68 is_required=['Yes'], |
| 69 is_multivalued=['Yes'], |
| 70 docstring=['It is just some field'], |
| 71 applicable_type=['Defect'], |
| 72 admin_names=['']) |
| 73 url = self.servlet.ProcessFormData(self.mr, post_data) |
| 74 self.assertTrue('/adminLabels?saved=1&' in url) |
| 75 config = self.services.config.GetProjectConfig( |
| 76 self.mr.cnxn, self.mr.project_id) |
| 77 |
| 78 fd = tracker_bizobj.FindFieldDef('somefield', config) |
| 79 self.assertEqual('somefield', fd.field_name) |
| 80 self.assertEqual(tracker_pb2.FieldTypes.INT_TYPE, fd.field_type) |
| 81 self.assertEqual(1, fd.min_value) |
| 82 self.assertEqual(99, fd.max_value) |
| 83 self.assertEqual(tracker_pb2.NotifyTriggers.ANY_COMMENT, fd.notify_on) |
| 84 self.assertTrue(fd.is_required) |
| 85 self.assertTrue(fd.is_multivalued) |
| 86 self.assertEqual('It is just some field', fd.docstring) |
| 87 self.assertEqual('Defect', fd.applicable_type) |
| 88 self.assertEqual('', fd.applicable_predicate) |
| 89 self.assertEqual([], fd.admin_ids) |
| 90 |
| 91 |
| 92 class CheckFieldNameJSONTest(unittest.TestCase): |
| 93 |
| 94 def setUp(self): |
| 95 self.services = service_manager.Services( |
| 96 user=fake.UserService(), |
| 97 config=fake.ConfigService(), |
| 98 project=fake.ProjectService()) |
| 99 self.servlet = fieldcreate.CheckFieldNameJSON( |
| 100 'req', 'res', services=self.services) |
| 101 self.project = self.services.project.TestAddProject('proj') |
| 102 self.config = self.services.config.GetProjectConfig( |
| 103 'fake cnxn', self.project.project_id) |
| 104 self.services.config.StoreConfig('fake cnxn', self.config) |
| 105 |
| 106 def testHandleRequest_NewField(self): |
| 107 mr = testing_helpers.MakeMonorailRequest( |
| 108 project=self.project, perms=permissions.OWNER_ACTIVE_PERMISSIONSET, |
| 109 path='/p/proj/fields/checkname?field=somefield') |
| 110 page_data = self.servlet.HandleRequest(mr) |
| 111 self.assertItemsEqual( |
| 112 ['error_message', 'choices'], page_data.keys()) |
| 113 self.assertIsNone(page_data['error_message']) |
| 114 self.assertItemsEqual([], page_data['choices']) |
| 115 |
| 116 def testHandleRequest_FieldNameAlreadyUsed(self): |
| 117 fd = tracker_bizobj.MakeFieldDef( |
| 118 123, 789, 'CPU', tracker_pb2.FieldTypes.INT_TYPE, None, |
| 119 '', False, False, None, None, '', False, '', '', |
| 120 tracker_pb2.NotifyTriggers.NEVER, 'doc', False) |
| 121 self.config.field_defs.append(fd) |
| 122 mr = testing_helpers.MakeMonorailRequest( |
| 123 project=self.project, perms=permissions.OWNER_ACTIVE_PERMISSIONSET, |
| 124 path='/p/proj/fields/checkname?field=CPU') |
| 125 page_data = self.servlet.HandleRequest(mr) |
| 126 self.assertItemsEqual( |
| 127 ['error_message', 'choices'], page_data.keys()) |
| 128 self.assertEqual('That name is already in use.', |
| 129 page_data['error_message']) |
| 130 self.assertItemsEqual([], page_data['choices']) |
| 131 |
| 132 def testHandleRequest_ReservedField(self): |
| 133 mr = testing_helpers.MakeMonorailRequest( |
| 134 project=self.project, perms=permissions.OWNER_ACTIVE_PERMISSIONSET, |
| 135 path='/p/proj/fields/checkname?field=summary') |
| 136 page_data = self.servlet.HandleRequest(mr) |
| 137 self.assertItemsEqual( |
| 138 ['error_message', 'choices'], page_data.keys()) |
| 139 self.assertEqual('That name is reserved.', page_data['error_message']) |
| 140 self.assertItemsEqual([], page_data['choices']) |
| 141 |
| 142 def testHandleRequest_LabelsToField(self): |
| 143 mr = testing_helpers.MakeMonorailRequest( |
| 144 project=self.project, perms=permissions.OWNER_ACTIVE_PERMISSIONSET, |
| 145 path='/p/proj/fields/checkname?field=type') |
| 146 page_data = self.servlet.HandleRequest(mr) |
| 147 self.assertItemsEqual( |
| 148 ['error_message', 'choices'], page_data.keys()) |
| 149 self.assertIsNone(page_data['error_message']) |
| 150 self.assertEqual(4, len(page_data['choices'])) |
| 151 |
| 152 |
| 153 class FieldCreateMethodsTest(unittest.TestCase): |
| 154 |
| 155 def setUp(self): |
| 156 self.config = tracker_bizobj.MakeDefaultProjectIssueConfig(789) |
| 157 |
| 158 def testFieldNameErrorMessage_NoConflict(self): |
| 159 self.assertIsNone(fieldcreate.FieldNameErrorMessage( |
| 160 'somefield', self.config)) |
| 161 |
| 162 def testFieldNameErrorMessage_Reserved(self): |
| 163 self.assertEqual( |
| 164 'That name is reserved.', |
| 165 fieldcreate.FieldNameErrorMessage('owner', self.config)) |
| 166 |
| 167 def testFieldNameErrorMessage_AlreadyInUse(self): |
| 168 fd = tracker_bizobj.MakeFieldDef( |
| 169 123, 789, 'CPU', tracker_pb2.FieldTypes.INT_TYPE, None, |
| 170 '', False, False, None, None, '', False, '', '', |
| 171 tracker_pb2.NotifyTriggers.NEVER, 'doc', False) |
| 172 self.config.field_defs.append(fd) |
| 173 self.assertEqual( |
| 174 'That name is already in use.', |
| 175 fieldcreate.FieldNameErrorMessage('CPU', self.config)) |
| 176 |
| 177 def testFieldNameErrorMessage_PrefixOfExisting(self): |
| 178 fd = tracker_bizobj.MakeFieldDef( |
| 179 123, 789, 'sign-off', tracker_pb2.FieldTypes.INT_TYPE, None, |
| 180 '', False, False, None, None, '', False, '', '', |
| 181 tracker_pb2.NotifyTriggers.NEVER, 'doc', False) |
| 182 self.config.field_defs.append(fd) |
| 183 self.assertEqual( |
| 184 'That name is a prefix of an existing field name.', |
| 185 fieldcreate.FieldNameErrorMessage('sign', self.config)) |
| 186 |
| 187 def testFieldNameErrorMessage_IncludesExisting(self): |
| 188 fd = tracker_bizobj.MakeFieldDef( |
| 189 123, 789, 'opt', tracker_pb2.FieldTypes.INT_TYPE, None, |
| 190 '', False, False, None, None, '', False, '', '', |
| 191 tracker_pb2.NotifyTriggers.NEVER, 'doc', False) |
| 192 self.config.field_defs.append(fd) |
| 193 self.assertEqual( |
| 194 'An existing field name is a prefix of that name.', |
| 195 fieldcreate.FieldNameErrorMessage('opt-in', self.config)) |
| 196 |
| 197 def testExistingEnumChoices_NewEnum(self): |
| 198 self.assertItemsEqual( |
| 199 [], |
| 200 fieldcreate.ExistingEnumChoices('Theme', self.config)) |
| 201 |
| 202 def testExistingEnumChoices_ConvertLabelsToEnum(self): |
| 203 label_doc_list = fieldcreate.ExistingEnumChoices('Priority', self.config) |
| 204 self.assertItemsEqual( |
| 205 ['Critical', 'High', 'Medium', 'Low'], |
| 206 [item.name for item in label_doc_list]) |
| 207 |
| 208 |
| 209 if __name__ == '__main__': |
| 210 unittest.main() |
OLD | NEW |