1d0825bca7fe65beaee391d30da42e937db621564Steve Block# Copyright (c) 2010 Google Inc. All rights reserved. 2d0825bca7fe65beaee391d30da42e937db621564Steve Block# 3d0825bca7fe65beaee391d30da42e937db621564Steve Block# Redistribution and use in source and binary forms, with or without 4d0825bca7fe65beaee391d30da42e937db621564Steve Block# modification, are permitted provided that the following conditions are 5d0825bca7fe65beaee391d30da42e937db621564Steve Block# met: 6d0825bca7fe65beaee391d30da42e937db621564Steve Block# 7d0825bca7fe65beaee391d30da42e937db621564Steve Block# * Redistributions of source code must retain the above copyright 8d0825bca7fe65beaee391d30da42e937db621564Steve Block# notice, this list of conditions and the following disclaimer. 9d0825bca7fe65beaee391d30da42e937db621564Steve Block# * Redistributions in binary form must reproduce the above 10d0825bca7fe65beaee391d30da42e937db621564Steve Block# copyright notice, this list of conditions and the following disclaimer 11d0825bca7fe65beaee391d30da42e937db621564Steve Block# in the documentation and/or other materials provided with the 12d0825bca7fe65beaee391d30da42e937db621564Steve Block# distribution. 13d0825bca7fe65beaee391d30da42e937db621564Steve Block# * Neither the name of Google Inc. nor the names of its 14d0825bca7fe65beaee391d30da42e937db621564Steve Block# contributors may be used to endorse or promote products derived from 15d0825bca7fe65beaee391d30da42e937db621564Steve Block# this software without specific prior written permission. 16d0825bca7fe65beaee391d30da42e937db621564Steve Block# 17d0825bca7fe65beaee391d30da42e937db621564Steve Block# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18d0825bca7fe65beaee391d30da42e937db621564Steve Block# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19d0825bca7fe65beaee391d30da42e937db621564Steve Block# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20d0825bca7fe65beaee391d30da42e937db621564Steve Block# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21d0825bca7fe65beaee391d30da42e937db621564Steve Block# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22d0825bca7fe65beaee391d30da42e937db621564Steve Block# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23d0825bca7fe65beaee391d30da42e937db621564Steve Block# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24d0825bca7fe65beaee391d30da42e937db621564Steve Block# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25d0825bca7fe65beaee391d30da42e937db621564Steve Block# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26d0825bca7fe65beaee391d30da42e937db621564Steve Block# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27d0825bca7fe65beaee391d30da42e937db621564Steve Block# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28d0825bca7fe65beaee391d30da42e937db621564Steve Block 29d0825bca7fe65beaee391d30da42e937db621564Steve Blockimport unittest 30d0825bca7fe65beaee391d30da42e937db621564Steve Block 31dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockfrom webkitpy.common.net.networktransaction import NetworkTransaction, NetworkTimeout 32dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockfrom webkitpy.common.system.logtesting import LoggingTestCase 33dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockfrom webkitpy.thirdparty.autoinstalled.mechanize import HTTPError 34d0825bca7fe65beaee391d30da42e937db621564Steve Block 35dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 36dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Blockclass NetworkTransactionTest(LoggingTestCase): 37d0825bca7fe65beaee391d30da42e937db621564Steve Block exception = Exception("Test exception") 38d0825bca7fe65beaee391d30da42e937db621564Steve Block 39d0825bca7fe65beaee391d30da42e937db621564Steve Block def test_success(self): 40d0825bca7fe65beaee391d30da42e937db621564Steve Block transaction = NetworkTransaction() 41d0825bca7fe65beaee391d30da42e937db621564Steve Block self.assertEqual(transaction.run(lambda: 42), 42) 42d0825bca7fe65beaee391d30da42e937db621564Steve Block 43d0825bca7fe65beaee391d30da42e937db621564Steve Block def _raise_exception(self): 44d0825bca7fe65beaee391d30da42e937db621564Steve Block raise self.exception 45d0825bca7fe65beaee391d30da42e937db621564Steve Block 46d0825bca7fe65beaee391d30da42e937db621564Steve Block def test_exception(self): 47d0825bca7fe65beaee391d30da42e937db621564Steve Block transaction = NetworkTransaction() 48d0825bca7fe65beaee391d30da42e937db621564Steve Block did_process_exception = False 49d0825bca7fe65beaee391d30da42e937db621564Steve Block did_throw_exception = True 50d0825bca7fe65beaee391d30da42e937db621564Steve Block try: 51d0825bca7fe65beaee391d30da42e937db621564Steve Block transaction.run(lambda: self._raise_exception()) 52d0825bca7fe65beaee391d30da42e937db621564Steve Block did_throw_exception = False 53d0825bca7fe65beaee391d30da42e937db621564Steve Block except Exception, e: 54d0825bca7fe65beaee391d30da42e937db621564Steve Block did_process_exception = True 55d0825bca7fe65beaee391d30da42e937db621564Steve Block self.assertEqual(e, self.exception) 56d0825bca7fe65beaee391d30da42e937db621564Steve Block self.assertTrue(did_throw_exception) 57d0825bca7fe65beaee391d30da42e937db621564Steve Block self.assertTrue(did_process_exception) 58d0825bca7fe65beaee391d30da42e937db621564Steve Block 59a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch def _raise_500_error(self): 60d0825bca7fe65beaee391d30da42e937db621564Steve Block self._run_count += 1 61d0825bca7fe65beaee391d30da42e937db621564Steve Block if self._run_count < 3: 62a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch raise HTTPError("http://example.com/", 500, "internal server error", None, None) 63d0825bca7fe65beaee391d30da42e937db621564Steve Block return 42 64d0825bca7fe65beaee391d30da42e937db621564Steve Block 65a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch def _raise_404_error(self): 66a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch raise HTTPError("http://foo.com/", 404, "not found", None, None) 67a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch 68d0825bca7fe65beaee391d30da42e937db621564Steve Block def test_retry(self): 69d0825bca7fe65beaee391d30da42e937db621564Steve Block self._run_count = 0 70d0825bca7fe65beaee391d30da42e937db621564Steve Block transaction = NetworkTransaction(initial_backoff_seconds=0) 71a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch self.assertEqual(transaction.run(lambda: self._raise_500_error()), 42) 72d0825bca7fe65beaee391d30da42e937db621564Steve Block self.assertEqual(self._run_count, 3) 73cad810f21b803229eb11403f9209855525a25d57Steve Block self.assertLog(['WARNING: Received HTTP status 500 loading "http://example.com/". ' 74dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 'Retrying in 0 seconds...\n', 75cad810f21b803229eb11403f9209855525a25d57Steve Block 'WARNING: Received HTTP status 500 loading "http://example.com/". ' 76dcc8cf2e65d1aa555cce12431a16547e66b469eeSteve Block 'Retrying in 0.0 seconds...\n']) 77d0825bca7fe65beaee391d30da42e937db621564Steve Block 78a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch def test_convert_404_to_None(self): 79a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch transaction = NetworkTransaction(convert_404_to_None=True) 80a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch self.assertEqual(transaction.run(lambda: self._raise_404_error()), None) 81a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch 82d0825bca7fe65beaee391d30da42e937db621564Steve Block def test_timeout(self): 83d0825bca7fe65beaee391d30da42e937db621564Steve Block self._run_count = 0 84d0825bca7fe65beaee391d30da42e937db621564Steve Block transaction = NetworkTransaction(initial_backoff_seconds=60*60, timeout_seconds=60) 85d0825bca7fe65beaee391d30da42e937db621564Steve Block did_process_exception = False 86d0825bca7fe65beaee391d30da42e937db621564Steve Block did_throw_exception = True 87d0825bca7fe65beaee391d30da42e937db621564Steve Block try: 88a94275402997c11dd2e778633dacf4b7e630a35dBen Murdoch transaction.run(lambda: self._raise_500_error()) 89d0825bca7fe65beaee391d30da42e937db621564Steve Block did_throw_exception = False 90d0825bca7fe65beaee391d30da42e937db621564Steve Block except NetworkTimeout, e: 91d0825bca7fe65beaee391d30da42e937db621564Steve Block did_process_exception = True 92d0825bca7fe65beaee391d30da42e937db621564Steve Block self.assertTrue(did_throw_exception) 93d0825bca7fe65beaee391d30da42e937db621564Steve Block self.assertTrue(did_process_exception) 94