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 """Tests for URL handler registration helper functions.""" |
| 7 |
| 8 import unittest |
| 9 |
| 10 import webapp2 |
| 11 |
| 12 from framework import registerpages_helpers |
| 13 |
| 14 |
| 15 class SendRedirectInScopeTest(unittest.TestCase): |
| 16 |
| 17 def testMakeRedirectInScope_Error(self): |
| 18 self.assertRaises( |
| 19 AssertionError, |
| 20 registerpages_helpers.MakeRedirectInScope, 'no/initial/slash', 'p') |
| 21 self.assertRaises( |
| 22 AssertionError, |
| 23 registerpages_helpers.MakeRedirectInScope, '', 'p') |
| 24 |
| 25 def testMakeRedirectInScope_Normal(self): |
| 26 factory = registerpages_helpers.MakeRedirectInScope('/', 'p') |
| 27 # Non-dasher, normal case |
| 28 request = webapp2.Request.blank( |
| 29 path='/p/foo', headers={'Host': 'example.com'}) |
| 30 response = webapp2.Response() |
| 31 redirector = factory(request, response) |
| 32 redirector.get() |
| 33 self.assertEqual(response.location, '//example.com/p/foo/') |
| 34 self.assertEqual(response.status, '301 Moved Permanently') |
| 35 |
| 36 def testMakeRedirectInScope_Temporary(self): |
| 37 factory = registerpages_helpers.MakeRedirectInScope( |
| 38 '/', 'p', permanent=False) |
| 39 request = webapp2.Request.blank( |
| 40 path='/p/foo', headers={'Host': 'example.com'}) |
| 41 response = webapp2.Response() |
| 42 redirector = factory(request, response) |
| 43 redirector.get() |
| 44 self.assertEqual(response.location, '//example.com/p/foo/') |
| 45 self.assertEqual(response.status, '302 Moved Temporarily') |
| 46 |
| 47 |
| 48 if __name__ == '__main__': |
| 49 unittest.main() |
OLD | NEW |