| OLD | NEW |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is govered by a BSD-style | 2 # Use of this source code is govered by a BSD-style |
| 3 # license that can be found in the LICENSE file or at | 3 # license that can be found in the LICENSE file or at |
| 4 # https://developers.google.com/open-source/licenses/bsd | 4 # https://developers.google.com/open-source/licenses/bsd |
| 5 | 5 |
| 6 """Unit tests for projectadminadvanced module.""" | 6 """Unit tests for projectadminadvanced module.""" |
| 7 | 7 |
| 8 import time | 8 import time |
| 9 import unittest | 9 import unittest |
| 10 | 10 |
| 11 from framework import permissions | 11 from framework import permissions |
| 12 from project import projectadminadvanced | 12 from project import projectadminadvanced |
| 13 from proto import project_pb2 | 13 from proto import project_pb2 |
| 14 from services import service_manager | 14 from services import service_manager |
| 15 from testing import fake | 15 from testing import fake |
| 16 from testing import testing_helpers | 16 from testing import testing_helpers |
| 17 | 17 |
| 18 NOW = 1277762224 |
| 19 |
| 18 | 20 |
| 19 class ProjectAdminAdvancedTest(unittest.TestCase): | 21 class ProjectAdminAdvancedTest(unittest.TestCase): |
| 20 """Unit tests for the ProjectAdminAdvanced servlet class.""" | 22 """Unit tests for the ProjectAdminAdvanced servlet class.""" |
| 21 | 23 |
| 22 def setUp(self): | 24 def setUp(self): |
| 23 services = service_manager.Services( | 25 services = service_manager.Services( |
| 24 project=fake.ProjectService()) | 26 project=fake.ProjectService()) |
| 25 self.servlet = projectadminadvanced.ProjectAdminAdvanced( | 27 self.servlet = projectadminadvanced.ProjectAdminAdvanced( |
| 26 'req', 'res', services=services) | 28 'req', 'res', services=services) |
| 27 self.project = services.project.TestAddProject('proj') | 29 self.project = services.project.TestAddProject('proj') |
| 28 self.mr = testing_helpers.MakeMonorailRequest( | 30 self.mr = testing_helpers.MakeMonorailRequest( |
| 29 project=self.project, perms=permissions.OWNER_ACTIVE_PERMISSIONSET) | 31 project=self.project, perms=permissions.OWNER_ACTIVE_PERMISSIONSET) |
| 32 self.orig_time_function = time.time |
| 33 time.time = lambda: NOW |
| 34 |
| 35 def tearDown(self): |
| 36 time.time = self.orig_time_function |
| 30 | 37 |
| 31 def testAssertBasePermission(self): | 38 def testAssertBasePermission(self): |
| 32 # Signed-out users cannot edit the project | 39 # Signed-out users cannot edit the project |
| 33 self.mr.perms = permissions.READ_ONLY_PERMISSIONSET | 40 self.mr.perms = permissions.READ_ONLY_PERMISSIONSET |
| 34 self.assertRaises(permissions.PermissionException, | 41 self.assertRaises(permissions.PermissionException, |
| 35 self.servlet.AssertBasePermission, self.mr) | 42 self.servlet.AssertBasePermission, self.mr) |
| 36 | 43 |
| 37 # Non-member users cannot edit the project | 44 # Non-member users cannot edit the project |
| 38 self.mr.perms = permissions.USER_PERMISSIONSET | 45 self.mr.perms = permissions.USER_PERMISSIONSET |
| 39 self.assertRaises(permissions.PermissionException, | 46 self.assertRaises(permissions.PermissionException, |
| (...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 99 def testBuildComponentQuota(self): | 106 def testBuildComponentQuota(self): |
| 100 ezt_item = self.servlet._BuildComponentQuota( | 107 ezt_item = self.servlet._BuildComponentQuota( |
| 101 5000, 10000, 'attachments') | 108 5000, 10000, 'attachments') |
| 102 self.assertEqual(50, ezt_item.used_percent) | 109 self.assertEqual(50, ezt_item.used_percent) |
| 103 self.assertEqual('attachments', ezt_item.field_name) | 110 self.assertEqual('attachments', ezt_item.field_name) |
| 104 | 111 |
| 105 def testProcessFormData_NotDeleted(self): | 112 def testProcessFormData_NotDeleted(self): |
| 106 self.mr.project_name = 'proj' | 113 self.mr.project_name = 'proj' |
| 107 post_data = fake.PostData() | 114 post_data = fake.PostData() |
| 108 next_url = self.servlet.ProcessFormData(self.mr, post_data) | 115 next_url = self.servlet.ProcessFormData(self.mr, post_data) |
| 109 now = int(time.time()) | |
| 110 self.assertEqual( | 116 self.assertEqual( |
| 111 'http://127.0.0.1/p/proj/adminAdvanced?saved=1&ts=%s' % now, | 117 'http://127.0.0.1/p/proj/adminAdvanced?saved=1&ts=%s' % NOW, |
| 112 next_url) | 118 next_url) |
| 113 | 119 |
| 114 def testProcessFormData_AfterDeletion(self): | 120 def testProcessFormData_AfterDeletion(self): |
| 115 self.mr.project_name = 'proj' | 121 self.mr.project_name = 'proj' |
| 116 self.project.state = project_pb2.ProjectState.ARCHIVED | 122 self.project.state = project_pb2.ProjectState.ARCHIVED |
| 117 post_data = fake.PostData(deletebtn='1') | 123 post_data = fake.PostData(deletebtn='1') |
| 118 next_url = self.servlet.ProcessFormData(self.mr, post_data) | 124 next_url = self.servlet.ProcessFormData(self.mr, post_data) |
| 119 self.assertEqual('http://127.0.0.1/hosting/', next_url) | 125 self.assertEqual('http://127.0.0.1/hosting/', next_url) |
| 120 | 126 |
| 121 | 127 |
| 122 if __name__ == '__main__': | 128 if __name__ == '__main__': |
| 123 unittest.main() | 129 unittest.main() |
| OLD | NEW |