OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 """Unit tests for rietveld.py.""" | 6 """Unit tests for rietveld.py.""" |
7 | 7 |
8 import logging | 8 import logging |
9 import os | 9 import os |
10 import ssl | 10 import ssl |
(...skipping 29 matching lines...) Expand all Loading... |
40 | 40 |
41 | 41 |
42 class BaseFixture(unittest.TestCase): | 42 class BaseFixture(unittest.TestCase): |
43 # Override. | 43 # Override. |
44 TESTED_CLASS = Exception | 44 TESTED_CLASS = Exception |
45 | 45 |
46 def setUp(self): | 46 def setUp(self): |
47 super(BaseFixture, self).setUp() | 47 super(BaseFixture, self).setUp() |
48 # Access to a protected member XX of a client class | 48 # Access to a protected member XX of a client class |
49 # pylint: disable=W0212 | 49 # pylint: disable=W0212 |
50 self.rietveld = self.TESTED_CLASS('url', 'email', 'password') | 50 self.rietveld = self.TESTED_CLASS('url', None, 'email') |
51 self.rietveld._send = self._rietveld_send | 51 self.rietveld._send = self._rietveld_send |
52 self.requests = [] | 52 self.requests = [] |
53 | 53 |
54 def tearDown(self): | 54 def tearDown(self): |
55 self.assertEqual([], self.requests) | 55 self.assertEqual([], self.requests) |
56 super(BaseFixture, self).tearDown() | 56 super(BaseFixture, self).tearDown() |
57 | 57 |
58 def _rietveld_send(self, url, *args, **kwargs): | 58 def _rietveld_send(self, url, *args, **kwargs): |
59 self.assertTrue(self.requests, url) | 59 self.assertTrue(self.requests, url) |
60 request = self.requests.pop(0) | 60 request = self.requests.pop(0) |
(...skipping 388 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
449 **kwargs): | 449 **kwargs): |
450 """Mock upload.py's Send() to raise SSLError""" | 450 """Mock upload.py's Send() to raise SSLError""" |
451 raise ssl.SSLError('The read operation timed out') | 451 raise ssl.SSLError('The read operation timed out') |
452 | 452 |
453 | 453 |
454 class DefaultTimeoutTest(auto_stub.TestCase): | 454 class DefaultTimeoutTest(auto_stub.TestCase): |
455 TESTED_CLASS = rietveld.Rietveld | 455 TESTED_CLASS = rietveld.Rietveld |
456 | 456 |
457 def setUp(self): | 457 def setUp(self): |
458 super(DefaultTimeoutTest, self).setUp() | 458 super(DefaultTimeoutTest, self).setUp() |
459 self.rietveld = self.TESTED_CLASS('url', 'email', 'password') | 459 self.rietveld = self.TESTED_CLASS('url', None, 'email') |
460 self.mock(self.rietveld.rpc_server, 'Send', MockSend) | 460 self.mock(self.rietveld.rpc_server, 'Send', MockSend) |
461 self.sleep_time = 0 | 461 self.sleep_time = 0 |
462 | 462 |
463 def test_timeout_get(self): | 463 def test_timeout_get(self): |
464 with self.assertRaises(ProbeException) as cm: | 464 with self.assertRaises(ProbeException) as cm: |
465 self.rietveld.get('/api/1234') | 465 self.rietveld.get('/api/1234') |
466 | 466 |
467 self.assertIsNotNone(cm.exception.value, 'Rietveld timeout was not set: %s' | 467 self.assertIsNotNone(cm.exception.value, 'Rietveld timeout was not set: %s' |
468 % traceback.format_exc()) | 468 % traceback.format_exc()) |
469 | 469 |
(...skipping 12 matching lines...) Expand all Loading... |
482 self.mock(time, 'sleep', self.MockSleep) | 482 self.mock(time, 'sleep', self.MockSleep) |
483 self.sleep_time = 0 | 483 self.sleep_time = 0 |
484 with self.assertRaises(ssl.SSLError): | 484 with self.assertRaises(ssl.SSLError): |
485 self.rietveld.post('/api/1234', [('key', 'data')]) | 485 self.rietveld.post('/api/1234', [('key', 'data')]) |
486 self.assertNotEqual(self.sleep_time, 0) | 486 self.assertNotEqual(self.sleep_time, 0) |
487 | 487 |
488 if __name__ == '__main__': | 488 if __name__ == '__main__': |
489 logging.basicConfig(level=[ | 489 logging.basicConfig(level=[ |
490 logging.ERROR, logging.INFO, logging.DEBUG][min(2, sys.argv.count('-v'))]) | 490 logging.ERROR, logging.INFO, logging.DEBUG][min(2, sys.argv.count('-v'))]) |
491 unittest.main() | 491 unittest.main() |
OLD | NEW |