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=[ |