Docserver: Add support for viewing docs with a codereview patch applied

chrome/common/extensions/docs/server2/caching_rietveld_patcher_test.py

+#!/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 caching_rietveld_patcher import (CachingRietveldPatcher,
+                                      _VERSION_CACHE_MAXAGE)
+from datetime import datetime
+from file_system import FileNotFoundError
+from future import Future
+from object_store_creator import ObjectStoreCreator
+from patcher import Patcher
+from test_patcher import TestPatcher
+
+_TEST_PATCH_VERSION = '1'
+_TEST_PATCH_FILES = (['add.txt'], ['del.txt'], ['modify.txt'])
+_TEST_PATCH_DATA = {
+  'add.txt': 'add',
+  'modify.txt': 'modify',
+}
+
+class FakeDateTime(object):
+  def __init__(self, time=datetime.now()):
+    self.time = time
+
+  def now(self):
+    return self.time
+
+class CachingRietveldPatcherTest(unittest.TestCase):
+  def setUp(self):
+    self._datetime = FakeDateTime()
+    # CachingRietveldPatcher should always call Apply with binary=True.
+    self._test_patcher = TestPatcher(_TEST_PATCH_VERSION,
+                                     _TEST_PATCH_FILES,
+                                     _TEST_PATCH_DATA,
+                                     assert_binary=True)
+    self._patcher = CachingRietveldPatcher(
+        self._test_patcher,
+        ObjectStoreCreator('test', start_empty=False),
+        self._datetime)
+
+  def testGetVersion(self):
+    # Invalidate cache.
+    self._datetime.time += _VERSION_CACHE_MAXAGE
+    # Fill cache.
+    self._patcher.GetVersion()
+    count = self._test_patcher.get_version_count
+    # Should read from cache.
+    self._patcher.GetVersion()
+    self.assertEqual(count, self._test_patcher.get_version_count)
+    # Invalidate cache.
+    self._datetime.time += _VERSION_CACHE_MAXAGE
+    # Should fetch version.
+    self._patcher.GetVersion()
+    self.assertEqual(count + 1, self._test_patcher.get_version_count)
+
+  def testGetPatchedFiles(self):
+    # Fill cache.
+    self._patcher.GetPatchedFiles()
+    count = self._test_patcher.get_patched_files_count
+    # Should read from cache.
+    self._patcher.GetPatchedFiles()
+    self.assertEqual(count, self._test_patcher.get_patched_files_count)
+
+  def testApply(self):
+    # Fill cache.
+    self._patcher.Apply(['add.txt'], None).Get()
+    count = self._test_patcher.apply_count
+    # Should read from cache even though it's reading another file.
+    self._patcher.Apply(['modify.txt'], None).Get()
+    self.assertEqual(count, self._test_patcher.apply_count)
+
+    # Make sure RietveldPatcher handles |binary| correctly.
+    self.assertTrue(isinstance(self._patcher.Apply(['add.txt'], None, True).
+        Get()['add.txt'], str),
+        'Expected result is binary. It was text.')
+    self.assertTrue(isinstance(self._patcher.Apply(['add.txt'], None).
+        Get()['add.txt'], unicode),
+        'Expected result is text. It was binary.')
+
+if __name__ == '__main__':
+  unittest.main()