Open Source Monorail

appengine/monorail/sitewide/test/moved_test.py

+# Copyright 2016 The Chromium Authors. All rights reserved.
+# Use of this source code is govered by a BSD-style
+# license that can be found in the LICENSE file or at
+# https://developers.google.com/open-source/licenses/bsd
+
+"""Unit tests for the moved project notification page servlet."""
+
+import unittest
+
+import webapp2
+
+from services import service_manager
+from sitewide import moved
+from testing import fake
+from testing import testing_helpers
+
+
+class MovedTest(unittest.TestCase):
+
+  def setUp(self):
+    self.services = service_manager.Services(
+        project=fake.ProjectService())
+    self.servlet = moved.ProjectMoved('req', 'res', services=self.services)
+
+  def testGatherPageData(self):
+    project_name = 'my-project'
+    moved_to = 'http://we-are-outta-here.com/'
+    _request, mr = testing_helpers.GetRequestObjects(
+        path='/hosting/moved?project=my-project')
+
+    try:
+      self.servlet.GatherPageData(mr)
+      self.fail()
+    except webapp2.HTTPException as e:
+      self.assertEquals(404, e.code)
+
+    project = self.services.project.TestAddProject(project_name)
+    # Project exists but has not been moved, so 400 BAD_REQUEST.
+    try:
+      self.servlet.GatherPageData(mr)
+      self.fail()
+    except webapp2.HTTPException as e:
+      self.assertEquals(400, e.code)
+
+    # Display the moved_to url if it is valid.
+    project.moved_to = moved_to
+    page_data = self.servlet.GatherPageData(mr)
+    self.assertItemsEqual(
+        ['project_name', 'moved_to_url'],
+        page_data.keys())
+    self.assertEqual(project_name, page_data['project_name'])
+    self.assertEqual(moved_to, page_data['moved_to_url'])
+
+    # We only display URLs that start with 'http'.
+    project.moved_to = 'javascript:alert(1)'
+    page_data = self.servlet.GatherPageData(mr)
+    self.assertItemsEqual(
+        ['project_name', 'moved_to_url'],
+        page_data.keys())
+    self.assertEqual(project_name, page_data['project_name'])
+    self.assertEqual('#invalid-destination-url', page_data['moved_to_url'])
+
+
+if __name__ == '__main__':
+  unittest.main()