OLD | NEW |
---|---|
(Empty) | |
1 #!/usr/bin/env python | |
2 # Copyright 2013 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 import unittest | |
7 | |
8 from app_yaml_helper import AppYamlHelper | |
9 from file_system import FileNotFoundError | |
10 from mock_file_system import MockFileSystem | |
11 from object_store_creator import ObjectStoreCreator | |
12 from test_file_system import TestFileSystem | |
13 from test_util import DisableLogging | |
14 | |
15 _ExtractVersion, _IsGreater, _GenerateAppYaml = ( | |
16 AppYamlHelper.ExtractVersion, | |
17 AppYamlHelper.IsGreater, | |
18 AppYamlHelper.GenerateAppYaml) | |
19 | |
20 class AppYamlHelperTest(unittest.TestCase): | |
21 def testExtractVersion(self): | |
22 def run_test(version): | |
23 self.assertEqual(version, _ExtractVersion(_GenerateAppYaml(version))) | |
24 run_test('0') | |
25 run_test('0-0') | |
26 run_test('0-0-0') | |
27 run_test('1') | |
28 run_test('1-0') | |
29 run_test('1-0-0') | |
30 run_test('1-0-1') | |
31 run_test('1-1-0') | |
32 run_test('1-1-1') | |
33 run_test('2-0-9') | |
34 run_test('2-0-12') | |
35 run_test('2-1') | |
36 run_test('2-1-0') | |
37 run_test('2-11-0') | |
38 run_test('3-1-0') | |
39 run_test('3-1-3') | |
40 run_test('3-12-0') | |
41 | |
42 def testIsGreater(self): | |
43 def assert_is_greater(lhs, rhs): | |
44 self.assertTrue(_IsGreater(lhs, rhs), '%s is not > %s' % (lhs, rhs)) | |
45 self.assertFalse(_IsGreater(rhs, lhs), | |
46 '%s should not be > %s' % (rhs, lhs)) | |
47 assert_is_greater('0-0', '0') | |
48 assert_is_greater('0-0-0', '0') | |
49 assert_is_greater('0-0-0', '0-0') | |
50 assert_is_greater('1', '0') | |
51 assert_is_greater('1', '0-0') | |
52 assert_is_greater('1', '0-0-0') | |
53 assert_is_greater('1-0', '0-0') | |
54 assert_is_greater('1-0-0-0', '0-0-0') | |
55 assert_is_greater('2-0-12', '2-0-9') | |
56 assert_is_greater('2-0-12', '2-0-9-0') | |
57 assert_is_greater('2-0-12-0', '2-0-9') | |
58 assert_is_greater('2-0-12-0', '2-0-9-0') | |
59 assert_is_greater('2-1', '2-0-9') | |
60 assert_is_greater('2-1', '2-0-12') | |
61 assert_is_greater('2-1-0', '2-0-9') | |
62 assert_is_greater('2-1-0', '2-0-12') | |
63 assert_is_greater('3-1-0', '2-1') | |
64 assert_is_greater('3-1-0', '2-1-0') | |
65 assert_is_greater('3-1-0', '2-11-0') | |
66 assert_is_greater('3-1-3', '3-1-0') | |
67 assert_is_greater('3-12-0', '3-1-0') | |
68 assert_is_greater('3-12-0', '3-1-3') | |
69 assert_is_greater('3-12-0', '3-1-3-0') | |
70 | |
71 @DisableLogging('warning') | |
72 def testInstanceMethods(self): | |
73 test_data = { | |
74 'server2': { | |
75 'app.yaml': _GenerateAppYaml('1-0'), | |
76 'app_yaml_helper.py': 'Copyright notice etc' | |
77 } | |
78 } | |
79 | |
80 updates = [] | |
81 | |
82 file_system_at_head = MockFileSystem(TestFileSystem(test_data)) | |
83 | |
84 def apply_update(update): | |
85 file_system_at_head.Update(update) | |
86 updates.append(update) | |
87 | |
88 class TestDelegate(AppYamlHelper.Delegate): | |
89 def GetHostFileSystemForRevision(self, revision): | |
90 assert revision is not None | |
cduvall
2013/05/10 06:45:19
use self.assert* here?
not at google - send to devlin
2013/05/10 17:49:58
Done.
| |
91 assert revision >= 0 | |
92 return MockFileSystem.Create(TestFileSystem(test_data), | |
93 updates[:revision]) | |
94 | |
95 helper = AppYamlHelper('server2/app.yaml', | |
96 file_system_at_head, | |
97 TestDelegate(), | |
98 ObjectStoreCreator.ForTest(disable_wrappers=False)) | |
99 | |
100 def assert_is_up_to_date(version): | |
101 self.assertTrue(helper.IsUpToDate(version), | |
102 '%s is not up to date' % version) | |
103 self.assertRaises(ValueError, | |
104 helper.GetFirstRevisionGreaterThan, version) | |
105 | |
106 self.assertEqual(0, helper.GetFirstRevisionGreaterThan('0-5-0')) | |
107 assert_is_up_to_date('1-0-0') | |
108 assert_is_up_to_date('1-5-0') | |
109 | |
110 # Revision 1. | |
111 apply_update({'server2': { | |
112 'app.yaml': _GenerateAppYaml('1-5-0') | |
113 }}) | |
114 | |
115 self.assertEqual(0, helper.GetFirstRevisionGreaterThan('0-5-0')) | |
116 self.assertEqual(1, helper.GetFirstRevisionGreaterThan('1-0-0')) | |
117 assert_is_up_to_date('1-5-0') | |
118 assert_is_up_to_date('2-5-0') | |
119 | |
120 # Revision 2. | |
121 apply_update({'server2': { | |
122 'app_yaml_helper.py': 'fixed a bug' | |
123 }}) | |
124 | |
125 self.assertEqual(0, helper.GetFirstRevisionGreaterThan('0-5-0')) | |
126 self.assertEqual(1, helper.GetFirstRevisionGreaterThan('1-0-0')) | |
127 assert_is_up_to_date('1-5-0') | |
128 assert_is_up_to_date('2-5-0') | |
129 | |
130 # Revision 3. | |
131 apply_update({'server2': { | |
132 'app.yaml': _GenerateAppYaml('1-6-0') | |
133 }}) | |
134 | |
135 self.assertEqual(0, helper.GetFirstRevisionGreaterThan('0-5-0')) | |
136 self.assertEqual(1, helper.GetFirstRevisionGreaterThan('1-0-0')) | |
137 self.assertEqual(3, helper.GetFirstRevisionGreaterThan('1-5-0')) | |
138 assert_is_up_to_date('2-5-0') | |
139 | |
140 # Revision 4. | |
141 apply_update({'server2': { | |
142 'app.yaml': _GenerateAppYaml('1-8-0') | |
143 }}) | |
144 # Revision 5. | |
145 apply_update({'server2': { | |
146 'app.yaml': _GenerateAppYaml('2-0-0') | |
147 }}) | |
148 # Revision 6. | |
149 apply_update({'server2': { | |
150 'app.yaml': _GenerateAppYaml('2-2-0') | |
151 }}) | |
152 # Revision 7. | |
153 apply_update({'server2': { | |
154 'app.yaml': _GenerateAppYaml('2-4-0') | |
155 }}) | |
156 # Revision 8. | |
157 apply_update({'server2': { | |
158 'app.yaml': _GenerateAppYaml('2-6-0') | |
159 }}) | |
160 | |
161 self.assertEqual(0, helper.GetFirstRevisionGreaterThan('0-5-0')) | |
162 self.assertEqual(1, helper.GetFirstRevisionGreaterThan('1-0-0')) | |
163 self.assertEqual(3, helper.GetFirstRevisionGreaterThan('1-5-0')) | |
164 self.assertEqual(5, helper.GetFirstRevisionGreaterThan('1-8-0')) | |
165 self.assertEqual(6, helper.GetFirstRevisionGreaterThan('2-0-0')) | |
166 self.assertEqual(6, helper.GetFirstRevisionGreaterThan('2-1-0')) | |
167 self.assertEqual(7, helper.GetFirstRevisionGreaterThan('2-2-0')) | |
168 self.assertEqual(7, helper.GetFirstRevisionGreaterThan('2-3-0')) | |
169 self.assertEqual(8, helper.GetFirstRevisionGreaterThan('2-4-0')) | |
170 self.assertEqual(8, helper.GetFirstRevisionGreaterThan('2-5-0')) | |
171 assert_is_up_to_date('2-6-0') | |
172 assert_is_up_to_date('2-7-0') | |
173 | |
174 if __name__ == '__main__': | |
175 unittest.main() | |
OLD | NEW |