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 """Unittests for the Project Creation servlet.""" |
| 7 |
| 8 import unittest |
| 9 |
| 10 import settings |
| 11 from framework import permissions |
| 12 from proto import project_pb2 |
| 13 from proto import site_pb2 |
| 14 from services import service_manager |
| 15 from sitewide import projectcreate |
| 16 from testing import fake |
| 17 from testing import testing_helpers |
| 18 |
| 19 |
| 20 class ProjectCreateTest(unittest.TestCase): |
| 21 |
| 22 def setUp(self): |
| 23 services = service_manager.Services() |
| 24 self.servlet = projectcreate.ProjectCreate('req', 'res', services=services) |
| 25 |
| 26 def CheckAssertBasePermissions( |
| 27 self, restriction, expect_admin_ok, expect_nonadmin_ok): |
| 28 old_project_creation_restriction = settings.project_creation_restriction |
| 29 settings.project_creation_restriction = restriction |
| 30 |
| 31 # Anon users can never do it |
| 32 mr = testing_helpers.MakeMonorailRequest( |
| 33 perms=permissions.GetPermissions(None, {}, None)) |
| 34 self.assertRaises( |
| 35 permissions.PermissionException, |
| 36 self.servlet.AssertBasePermission, mr) |
| 37 |
| 38 mr = testing_helpers.MakeMonorailRequest() |
| 39 if expect_admin_ok: |
| 40 self.servlet.AssertBasePermission(mr) |
| 41 else: |
| 42 self.assertRaises( |
| 43 permissions.PermissionException, |
| 44 self.servlet.AssertBasePermission, mr) |
| 45 |
| 46 mr = testing_helpers.MakeMonorailRequest( |
| 47 perms=permissions.GetPermissions(mr.auth.user_pb, {111L}, None)) |
| 48 if expect_nonadmin_ok: |
| 49 self.servlet.AssertBasePermission(mr) |
| 50 else: |
| 51 self.assertRaises( |
| 52 permissions.PermissionException, |
| 53 self.servlet.AssertBasePermission, mr) |
| 54 |
| 55 settings.project_creation_restriction = old_project_creation_restriction |
| 56 |
| 57 def testAssertBasePermission(self): |
| 58 self.CheckAssertBasePermissions( |
| 59 site_pb2.UserTypeRestriction.ANYONE, True, True) |
| 60 self.CheckAssertBasePermissions( |
| 61 site_pb2.UserTypeRestriction.ADMIN_ONLY, True, False) |
| 62 self.CheckAssertBasePermissions( |
| 63 site_pb2.UserTypeRestriction.NO_ONE, False, False) |
| 64 |
| 65 def testGatherPageData(self): |
| 66 mr = testing_helpers.MakeMonorailRequest() |
| 67 page_data = self.servlet.GatherPageData(mr) |
| 68 self.assertEqual('', page_data['initial_name']) |
| 69 self.assertEqual('', page_data['initial_summary']) |
| 70 self.assertEqual('', page_data['initial_description']) |
| 71 self.assertEqual([], page_data['labels']) |
| 72 |
| 73 def testGatherHelpData(self): |
| 74 project = project_pb2.Project() |
| 75 mr = testing_helpers.MakeMonorailRequest(project=project) |
| 76 |
| 77 # Users not near the lifetime limit see no cue card. |
| 78 help_data = self.servlet.GatherHelpData(mr, {}) |
| 79 self.assertEqual(None, help_data['cue_remaining_projects']) |
| 80 |
| 81 # User who is near the lifetime limit will see a cue card. |
| 82 mr.auth.user_pb.project_creation_limit.lifetime_count = 20 |
| 83 help_data = self.servlet.GatherHelpData(mr, {}) |
| 84 self.assertEqual(5, help_data['cue_remaining_projects']) |
| 85 |
| 86 # User far under custom lifetime limit won't see a cue card. |
| 87 mr.auth.user_pb.project_creation_limit.lifetime_limit = 100 |
| 88 mr.auth.user_pb.project_creation_limit.lifetime_count = 20 |
| 89 help_data = self.servlet.GatherHelpData(mr, {}) |
| 90 self.assertEqual(None, help_data['cue_remaining_projects']) |
| 91 |
| 92 # User near custom lifetime limit will see a cue card. |
| 93 mr.auth.user_pb.project_creation_limit.lifetime_limit = 100 |
| 94 mr.auth.user_pb.project_creation_limit.lifetime_count = 91 |
| 95 help_data = self.servlet.GatherHelpData(mr, {}) |
| 96 self.assertEqual(9, help_data['cue_remaining_projects']) |
| 97 |
| 98 |
| 99 class CheckProjectNameJSONTest(unittest.TestCase): |
| 100 def setUp(self): |
| 101 services = service_manager.Services(project=fake.ProjectService()) |
| 102 self.project = services.project.TestAddProject('proj') |
| 103 self.servlet = projectcreate.CheckProjectNameJSON( |
| 104 'req', 'res', services=services) |
| 105 |
| 106 def testHandleRequestNameTaken(self): |
| 107 mr = testing_helpers.MakeMonorailRequest( |
| 108 project=self.project, user_info={'user_id': 222L}, method='POST', |
| 109 params={'project': self.project.project_name}) |
| 110 json_data = self.servlet.HandleRequest(mr) |
| 111 self.assertEqual('That project name is not available.', |
| 112 json_data['error_message']) |
| 113 |
| 114 def testHandleRequestNameNotTaken(self): |
| 115 mr = testing_helpers.MakeMonorailRequest( |
| 116 project=self.project, user_info={'user_id': 222L}, method='POST', |
| 117 params={'project': 'not-taken'}) |
| 118 json_data = self.servlet.HandleRequest(mr) |
| 119 self.assertEqual('', json_data['error_message']) |
| 120 |
| 121 |
| 122 if __name__ == '__main__': |
| 123 unittest.main() |
OLD | NEW |