| OLD | NEW |
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 # Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
| 3 # Use of this source code is governed by a BSD-style license that can be | |
| 4 # found in the LICENSE file. | |
| 5 | |
| 6 """Unit tests for verification/tree_status.py.""" | |
| 7 | |
| 8 import calendar | |
| 9 import json | |
| 10 import logging | |
| 11 import os | |
| 12 import StringIO | |
| 13 import sys | |
| 14 import unittest | |
| 15 import urllib2 | |
| 16 | |
| 17 ROOT_DIR = os.path.dirname(os.path.abspath(__file__)) | |
| 18 sys.path.insert(0, os.path.join(ROOT_DIR, '..')) | |
| 19 | |
| 20 # From tests/ | |
| 21 import mocks | |
| 22 | |
| 23 from verification import tree_status | |
| 24 | |
| 25 | |
| 26 class TreeStatusTest(mocks.TestCase): | |
| 27 def setUp(self): | |
| 28 super(TreeStatusTest, self).setUp() | |
| 29 reference = calendar.timegm((2010, 1, 1, 12, 0, 0, 0, 0, -1)) | |
| 30 self.mock(tree_status.time, 'time', lambda: reference) | |
| 31 self.urlrequests = [] | |
| 32 self.mock(urllib2, 'urlopen', self._urlopen) | |
| 33 | |
| 34 def tearDown(self): | |
| 35 super(TreeStatusTest, self).setUp() | |
| 36 | |
| 37 def _urlopen(self, _): | |
| 38 return StringIO.StringIO(json.dumps(self.urlrequests.pop(0))) | |
| 39 | |
| 40 def test_fail(self): | |
| 41 self.urlrequests = [ | |
| 42 [ | |
| 43 { | |
| 44 'date': '2010-01-01 11:56:00.0', | |
| 45 'general_state': 'open', | |
| 46 'message': 'Foo', | |
| 47 }, | |
| 48 { | |
| 49 'date': '2010-01-01 11:57:00.0', | |
| 50 'general_state': 'closed', | |
| 51 'message': 'Bar', | |
| 52 }, | |
| 53 { | |
| 54 'date': '2010-01-01 11:58:00.0', | |
| 55 'general_state': 'open', | |
| 56 'message': 'Baz', | |
| 57 }, | |
| 58 ], | |
| 59 ] | |
| 60 obj = tree_status.TreeStatus(tree_status_url='foo') | |
| 61 self.assertEqual(True, obj.postpone()) | |
| 62 self.assertEqual(u'Tree is currently not open: Bar', obj.why_not()) | |
| 63 self.assertEqual([], self.urlrequests) | |
| 64 | |
| 65 def test_pass(self): | |
| 66 self.urlrequests = [ | |
| 67 [ | |
| 68 { | |
| 69 'date': '2010-01-01 11:54:00.0', | |
| 70 'general_state': 'open', | |
| 71 'message': 'Foo', | |
| 72 }, | |
| 73 { | |
| 74 'date': '2010-01-01 11:57:00.0', | |
| 75 'general_state': 'open', | |
| 76 'message': 'Bar', | |
| 77 }, | |
| 78 { | |
| 79 'date': '2010-01-01 11:53:00.0', | |
| 80 'general_state': 'closed', | |
| 81 'message': 'Baz', | |
| 82 }, | |
| 83 ], | |
| 84 ] | |
| 85 obj = tree_status.TreeStatus(tree_status_url='foo') | |
| 86 self.assertEqual(False, obj.postpone()) | |
| 87 self.assertEqual(None, obj.why_not()) | |
| 88 self.assertEqual([], self.urlrequests) | |
| 89 | |
| 90 def test_skip_tree_check(self): | |
| 91 self.urlrequests = [ | |
| 92 [ | |
| 93 { | |
| 94 'date': '2010-01-01 11:56:00.0', | |
| 95 'general_state': 'open', | |
| 96 'message': 'Foo', | |
| 97 }, | |
| 98 { | |
| 99 'date': '2010-01-01 11:57:00.0', | |
| 100 'general_state': 'closed', | |
| 101 'message': 'Bar', | |
| 102 }, | |
| 103 { | |
| 104 'date': '2010-01-01 11:58:00.0', | |
| 105 'general_state': 'open', | |
| 106 'message': 'Baz', | |
| 107 }, | |
| 108 ], | |
| 109 ] | |
| 110 # Create a dummy pending obj to pass to the verifier. | |
| 111 class dummy_pending(object): | |
| 112 issue = 123 | |
| 113 description = 'foobarbaz\nNOTREECHECKS=true\nfoobarbaz' | |
| 114 verifications = {} | |
| 115 def __init__(self): | |
| 116 pass | |
| 117 | |
| 118 verifier = tree_status.TreeStatusVerifier('dummy_status_url') | |
| 119 verifier.verify(dummy_pending()) | |
| 120 | |
| 121 obj = dummy_pending.verifications['tree status'] | |
| 122 | |
| 123 self.assertEquals(tree_status.AlwaysOpenTreeStatus, type(obj)) | |
| 124 self.assertEqual(False, obj.postpone()) | |
| 125 self.assertEqual(None, obj.why_not()) | |
| 126 # None of the urls should have been opened since it is a skip request. | |
| 127 self.assertTrue(len(self.urlrequests[0]) == 3) | |
| 128 | |
| 129 def test_state(self): | |
| 130 t = tree_status.TreeStatus(tree_status_url='foo') | |
| 131 self.assertEqual(tree_status.base.SUCCEEDED, t.get_state()) | |
| 132 self.assertEqual([], self.urlrequests) | |
| 133 | |
| 134 | |
| 135 if __name__ == '__main__': | |
| 136 logging.basicConfig( | |
| 137 level=[logging.ERROR, logging.WARNING, logging.INFO, logging.DEBUG][ | |
| 138 min(sys.argv.count('-v'), 3)], | |
| 139 format='%(levelname)5s %(module)15s(%(lineno)3d): %(message)s') | |
| 140 unittest.main() | |
| OLD | NEW |