Index: chrome/common/extensions/docs/server2/app_yaml_helper_test.py |
diff --git a/chrome/common/extensions/docs/server2/app_yaml_helper_test.py b/chrome/common/extensions/docs/server2/app_yaml_helper_test.py |
new file mode 100755 |
index 0000000000000000000000000000000000000000..3521637b2c54de63b026f18bae4093f5399953f4 |
--- /dev/null |
+++ b/chrome/common/extensions/docs/server2/app_yaml_helper_test.py |
@@ -0,0 +1,175 @@ |
+#!/usr/bin/env python |
+# Copyright 2013 The Chromium Authors. All rights reserved. |
+# Use of this source code is governed by a BSD-style license that can be |
+# found in the LICENSE file. |
+ |
+import unittest |
+ |
+from app_yaml_helper import AppYamlHelper |
+from file_system import FileNotFoundError |
+from mock_file_system import MockFileSystem |
+from object_store_creator import ObjectStoreCreator |
+from test_file_system import TestFileSystem |
+from test_util import DisableLogging |
+ |
+_ExtractVersion, _IsGreater, _GenerateAppYaml = ( |
+ AppYamlHelper.ExtractVersion, |
+ AppYamlHelper.IsGreater, |
+ AppYamlHelper.GenerateAppYaml) |
+ |
+class AppYamlHelperTest(unittest.TestCase): |
+ def testExtractVersion(self): |
+ def run_test(version): |
+ self.assertEqual(version, _ExtractVersion(_GenerateAppYaml(version))) |
+ run_test('0') |
+ run_test('0-0') |
+ run_test('0-0-0') |
+ run_test('1') |
+ run_test('1-0') |
+ run_test('1-0-0') |
+ run_test('1-0-1') |
+ run_test('1-1-0') |
+ run_test('1-1-1') |
+ run_test('2-0-9') |
+ run_test('2-0-12') |
+ run_test('2-1') |
+ run_test('2-1-0') |
+ run_test('2-11-0') |
+ run_test('3-1-0') |
+ run_test('3-1-3') |
+ run_test('3-12-0') |
+ |
+ def testIsGreater(self): |
+ def assert_is_greater(lhs, rhs): |
+ self.assertTrue(_IsGreater(lhs, rhs), '%s is not > %s' % (lhs, rhs)) |
+ self.assertFalse(_IsGreater(rhs, lhs), |
+ '%s should not be > %s' % (rhs, lhs)) |
+ assert_is_greater('0-0', '0') |
+ assert_is_greater('0-0-0', '0') |
+ assert_is_greater('0-0-0', '0-0') |
+ assert_is_greater('1', '0') |
+ assert_is_greater('1', '0-0') |
+ assert_is_greater('1', '0-0-0') |
+ assert_is_greater('1-0', '0-0') |
+ assert_is_greater('1-0-0-0', '0-0-0') |
+ assert_is_greater('2-0-12', '2-0-9') |
+ assert_is_greater('2-0-12', '2-0-9-0') |
+ assert_is_greater('2-0-12-0', '2-0-9') |
+ assert_is_greater('2-0-12-0', '2-0-9-0') |
+ assert_is_greater('2-1', '2-0-9') |
+ assert_is_greater('2-1', '2-0-12') |
+ assert_is_greater('2-1-0', '2-0-9') |
+ assert_is_greater('2-1-0', '2-0-12') |
+ assert_is_greater('3-1-0', '2-1') |
+ assert_is_greater('3-1-0', '2-1-0') |
+ assert_is_greater('3-1-0', '2-11-0') |
+ assert_is_greater('3-1-3', '3-1-0') |
+ assert_is_greater('3-12-0', '3-1-0') |
+ assert_is_greater('3-12-0', '3-1-3') |
+ assert_is_greater('3-12-0', '3-1-3-0') |
+ |
+ @DisableLogging('warning') |
+ def testInstanceMethods(self): |
+ test_data = { |
+ 'server2': { |
+ 'app.yaml': _GenerateAppYaml('1-0'), |
+ 'app_yaml_helper.py': 'Copyright notice etc' |
+ } |
+ } |
+ |
+ updates = [] |
+ |
+ file_system_at_head = MockFileSystem(TestFileSystem(test_data)) |
+ |
+ def apply_update(update): |
+ file_system_at_head.Update(update) |
+ updates.append(update) |
+ |
+ class TestDelegate(AppYamlHelper.Delegate): |
+ def GetHostFileSystemForRevision(self, revision): |
+ 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.
|
+ assert revision >= 0 |
+ return MockFileSystem.Create(TestFileSystem(test_data), |
+ updates[:revision]) |
+ |
+ helper = AppYamlHelper('server2/app.yaml', |
+ file_system_at_head, |
+ TestDelegate(), |
+ ObjectStoreCreator.ForTest(disable_wrappers=False)) |
+ |
+ def assert_is_up_to_date(version): |
+ self.assertTrue(helper.IsUpToDate(version), |
+ '%s is not up to date' % version) |
+ self.assertRaises(ValueError, |
+ helper.GetFirstRevisionGreaterThan, version) |
+ |
+ self.assertEqual(0, helper.GetFirstRevisionGreaterThan('0-5-0')) |
+ assert_is_up_to_date('1-0-0') |
+ assert_is_up_to_date('1-5-0') |
+ |
+ # Revision 1. |
+ apply_update({'server2': { |
+ 'app.yaml': _GenerateAppYaml('1-5-0') |
+ }}) |
+ |
+ self.assertEqual(0, helper.GetFirstRevisionGreaterThan('0-5-0')) |
+ self.assertEqual(1, helper.GetFirstRevisionGreaterThan('1-0-0')) |
+ assert_is_up_to_date('1-5-0') |
+ assert_is_up_to_date('2-5-0') |
+ |
+ # Revision 2. |
+ apply_update({'server2': { |
+ 'app_yaml_helper.py': 'fixed a bug' |
+ }}) |
+ |
+ self.assertEqual(0, helper.GetFirstRevisionGreaterThan('0-5-0')) |
+ self.assertEqual(1, helper.GetFirstRevisionGreaterThan('1-0-0')) |
+ assert_is_up_to_date('1-5-0') |
+ assert_is_up_to_date('2-5-0') |
+ |
+ # Revision 3. |
+ apply_update({'server2': { |
+ 'app.yaml': _GenerateAppYaml('1-6-0') |
+ }}) |
+ |
+ self.assertEqual(0, helper.GetFirstRevisionGreaterThan('0-5-0')) |
+ self.assertEqual(1, helper.GetFirstRevisionGreaterThan('1-0-0')) |
+ self.assertEqual(3, helper.GetFirstRevisionGreaterThan('1-5-0')) |
+ assert_is_up_to_date('2-5-0') |
+ |
+ # Revision 4. |
+ apply_update({'server2': { |
+ 'app.yaml': _GenerateAppYaml('1-8-0') |
+ }}) |
+ # Revision 5. |
+ apply_update({'server2': { |
+ 'app.yaml': _GenerateAppYaml('2-0-0') |
+ }}) |
+ # Revision 6. |
+ apply_update({'server2': { |
+ 'app.yaml': _GenerateAppYaml('2-2-0') |
+ }}) |
+ # Revision 7. |
+ apply_update({'server2': { |
+ 'app.yaml': _GenerateAppYaml('2-4-0') |
+ }}) |
+ # Revision 8. |
+ apply_update({'server2': { |
+ 'app.yaml': _GenerateAppYaml('2-6-0') |
+ }}) |
+ |
+ self.assertEqual(0, helper.GetFirstRevisionGreaterThan('0-5-0')) |
+ self.assertEqual(1, helper.GetFirstRevisionGreaterThan('1-0-0')) |
+ self.assertEqual(3, helper.GetFirstRevisionGreaterThan('1-5-0')) |
+ self.assertEqual(5, helper.GetFirstRevisionGreaterThan('1-8-0')) |
+ self.assertEqual(6, helper.GetFirstRevisionGreaterThan('2-0-0')) |
+ self.assertEqual(6, helper.GetFirstRevisionGreaterThan('2-1-0')) |
+ self.assertEqual(7, helper.GetFirstRevisionGreaterThan('2-2-0')) |
+ self.assertEqual(7, helper.GetFirstRevisionGreaterThan('2-3-0')) |
+ self.assertEqual(8, helper.GetFirstRevisionGreaterThan('2-4-0')) |
+ self.assertEqual(8, helper.GetFirstRevisionGreaterThan('2-5-0')) |
+ assert_is_up_to_date('2-6-0') |
+ assert_is_up_to_date('2-7-0') |
+ |
+if __name__ == '__main__': |
+ unittest.main() |