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 moved project notification page servlet.""" |
| 7 |
| 8 import unittest |
| 9 |
| 10 import webapp2 |
| 11 |
| 12 from services import service_manager |
| 13 from sitewide import moved |
| 14 from testing import fake |
| 15 from testing import testing_helpers |
| 16 |
| 17 |
| 18 class MovedTest(unittest.TestCase): |
| 19 |
| 20 def setUp(self): |
| 21 self.services = service_manager.Services( |
| 22 project=fake.ProjectService()) |
| 23 self.servlet = moved.ProjectMoved('req', 'res', services=self.services) |
| 24 |
| 25 def testGatherPageData(self): |
| 26 project_name = 'my-project' |
| 27 moved_to = 'http://we-are-outta-here.com/' |
| 28 _request, mr = testing_helpers.GetRequestObjects( |
| 29 path='/hosting/moved?project=my-project') |
| 30 |
| 31 try: |
| 32 self.servlet.GatherPageData(mr) |
| 33 self.fail() |
| 34 except webapp2.HTTPException as e: |
| 35 self.assertEquals(404, e.code) |
| 36 |
| 37 project = self.services.project.TestAddProject(project_name) |
| 38 # Project exists but has not been moved, so 400 BAD_REQUEST. |
| 39 try: |
| 40 self.servlet.GatherPageData(mr) |
| 41 self.fail() |
| 42 except webapp2.HTTPException as e: |
| 43 self.assertEquals(400, e.code) |
| 44 |
| 45 # Display the moved_to url if it is valid. |
| 46 project.moved_to = moved_to |
| 47 page_data = self.servlet.GatherPageData(mr) |
| 48 self.assertItemsEqual( |
| 49 ['project_name', 'moved_to_url'], |
| 50 page_data.keys()) |
| 51 self.assertEqual(project_name, page_data['project_name']) |
| 52 self.assertEqual(moved_to, page_data['moved_to_url']) |
| 53 |
| 54 # We only display URLs that start with 'http'. |
| 55 project.moved_to = 'javascript:alert(1)' |
| 56 page_data = self.servlet.GatherPageData(mr) |
| 57 self.assertItemsEqual( |
| 58 ['project_name', 'moved_to_url'], |
| 59 page_data.keys()) |
| 60 self.assertEqual(project_name, page_data['project_name']) |
| 61 self.assertEqual('#invalid-destination-url', page_data['moved_to_url']) |
| 62 |
| 63 |
| 64 if __name__ == '__main__': |
| 65 unittest.main() |
OLD | NEW |