| Index: tests/rietveld_test.py
 | 
| diff --git a/tests/rietveld_test.py b/tests/rietveld_test.py
 | 
| index 17627a5fd1b2a69903e809612d5e0d8e80d6862b..3ac2444881686ae66d864d5947c9a81780b9b130 100755
 | 
| --- a/tests/rietveld_test.py
 | 
| +++ b/tests/rietveld_test.py
 | 
| @@ -8,11 +8,13 @@
 | 
|  import logging
 | 
|  import os
 | 
|  import sys
 | 
| +import traceback
 | 
|  import unittest
 | 
|  
 | 
|  sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 | 
|  
 | 
|  from testing_support.patches_data import GIT, RAW
 | 
| +from testing_support import auto_stub
 | 
|  
 | 
|  import patch
 | 
|  import rietveld
 | 
| @@ -421,6 +423,46 @@ class CachingRietveldTest(BaseFixture):
 | 
|      self.assertEqual(expected, self.rietveld.get_patchset_properties(1, 2))
 | 
|  
 | 
|  
 | 
| +class ProbeException(Exception):
 | 
| +  """Deep-probe a value."""
 | 
| +  value = None
 | 
| +
 | 
| +  def __init__(self, value):
 | 
| +    super(ProbeException, self).__init__()
 | 
| +    self.value = value
 | 
| +
 | 
| +
 | 
| +def MockSend(request_path, payload=None,
 | 
| +             content_type="application/octet-stream",
 | 
| +             timeout=None,
 | 
| +             extra_headers=None,
 | 
| +             **kwargs):
 | 
| +  """Mock upload.py's Send() to probe the timeout value"""
 | 
| +  raise ProbeException(timeout)
 | 
| +
 | 
| +
 | 
| +class DefaultTimeoutTest(auto_stub.TestCase):
 | 
| +  TESTED_CLASS = rietveld.Rietveld
 | 
| +
 | 
| +  def setUp(self):
 | 
| +    super(DefaultTimeoutTest, self).setUp()
 | 
| +    self.rietveld = self.TESTED_CLASS('url', 'email', 'password')
 | 
| +    self.mock(self.rietveld.rpc_server, 'Send', MockSend)
 | 
| +
 | 
| +  def test_timeout_get(self):
 | 
| +    with self.assertRaises(ProbeException) as cm:
 | 
| +      self.rietveld.get('/api/1234')
 | 
| +
 | 
| +    self.assertIsNotNone(cm.exception.value, 'Rietveld timeout was not set: %s'
 | 
| +                         % traceback.format_exc())
 | 
| +
 | 
| +  def test_timeout_post(self):
 | 
| +    with self.assertRaises(ProbeException) as cm:
 | 
| +      self.rietveld.post('/api/1234', [('key', 'data')])
 | 
| +
 | 
| +    self.assertIsNotNone(cm.exception.value, 'Rietveld timeout was not set: %s'
 | 
| +                         % traceback.format_exc())
 | 
| +
 | 
|  
 | 
|  if __name__ == '__main__':
 | 
|    logging.basicConfig(level=[
 | 
| 
 |