Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(632)

Side by Side Diff: chrome/common/extensions/docs/server2/cron_servlet_test.py

Issue 139303023: add GCS support to docs server (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: bumped versions Created 6 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 #!/usr/bin/env python 1 #!/usr/bin/env python
2 # Copyright 2013 The Chromium Authors. All rights reserved. 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 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 import unittest 6 import unittest
7 7
8 from appengine_wrappers import GetAppVersion 8 from appengine_wrappers import GetAppVersion
9 from app_yaml_helper import AppYamlHelper 9 from app_yaml_helper import AppYamlHelper
10 from cron_servlet import CronServlet 10 from cron_servlet import CronServlet
11 from empty_dir_file_system import EmptyDirFileSystem 11 from empty_dir_file_system import EmptyDirFileSystem
12 from extensions_paths import ( 12 from extensions_paths import (
13 APP_YAML, CONTENT_PROVIDERS, EXTENSIONS, PUBLIC_TEMPLATES, SERVER2, 13 APP_YAML, CONTENT_PROVIDERS, EXTENSIONS, PUBLIC_TEMPLATES, SERVER2,
14 STATIC_DOCS) 14 STATIC_DOCS)
15 from github_file_system_provider import GithubFileSystemProvider 15 from github_file_system_provider import GithubFileSystemProvider
16 from host_file_system_provider import HostFileSystemProvider 16 from host_file_system_provider import HostFileSystemProvider
17 from local_file_system import LocalFileSystem 17 from local_file_system import LocalFileSystem
18 from mock_file_system import MockFileSystem 18 from mock_file_system import MockFileSystem
19 from servlet import Request 19 from servlet import Request
20 from test_branch_utility import TestBranchUtility 20 from test_branch_utility import TestBranchUtility
21 from test_file_system import MoveTo, TestFileSystem 21 from test_file_system import MoveTo, TestFileSystem
22 from test_util import EnableLogging, ReadFile 22 from test_util import EnableLogging, ReadFile
23 from gcs_file_system_provider import CloudStorageFileSystemProvider
23 24
24 25
25 # NOTE(kalman): The ObjectStore created by the CronServlet is backed onto our 26 # NOTE(kalman): The ObjectStore created by the CronServlet is backed onto our
26 # fake AppEngine memcache/datastore, so the tests aren't isolated. Of course, 27 # fake AppEngine memcache/datastore, so the tests aren't isolated. Of course,
27 # if the host file systems have different identities, they will be, sort of. 28 # if the host file systems have different identities, they will be, sort of.
28 class _TestDelegate(CronServlet.Delegate): 29 class _TestDelegate(CronServlet.Delegate):
29 def __init__(self, create_file_system): 30 def __init__(self, create_file_system):
30 self.file_systems = [] 31 self.file_systems = []
31 # A callback taking a revision and returning a file system. 32 # A callback taking a revision and returning a file system.
32 self._create_file_system = create_file_system 33 self._create_file_system = create_file_system
33 self._app_version = GetAppVersion() 34 self._app_version = GetAppVersion()
34 35
35 def CreateBranchUtility(self, object_store_creator): 36 def CreateBranchUtility(self, object_store_creator):
36 return TestBranchUtility.CreateWithCannedData() 37 return TestBranchUtility.CreateWithCannedData()
37 38
38 def CreateHostFileSystemProvider(self, 39 def CreateHostFileSystemProvider(self,
39 object_store_creator, 40 object_store_creator,
40 max_trunk_revision=None): 41 max_trunk_revision=None):
41 def constructor(branch=None, revision=None): 42 def constructor(branch=None, revision=None):
42 file_system = self._create_file_system(revision) 43 file_system = self._create_file_system(revision)
43 self.file_systems.append(file_system) 44 self.file_systems.append(file_system)
44 return file_system 45 return file_system
45 return HostFileSystemProvider(object_store_creator, 46 return HostFileSystemProvider(object_store_creator,
46 max_trunk_revision=max_trunk_revision, 47 max_trunk_revision=max_trunk_revision,
47 constructor_for_test=constructor) 48 constructor_for_test=constructor)
48 49
49 def CreateGithubFileSystemProvider(self, object_store_creator): 50 def CreateGithubFileSystemProvider(self, object_store_creator):
50 return GithubFileSystemProvider.ForEmpty() 51 return GithubFileSystemProvider.ForEmpty()
51 52
53 def CreateGCSFileSystemProvider(self, object_store_creator):
54 return CloudStorageFileSystemProvider.ForEmpty()
55
52 def GetAppVersion(self): 56 def GetAppVersion(self):
53 return self._app_version 57 return self._app_version
54 58
55 # (non-Delegate method). 59 # (non-Delegate method).
56 def SetAppVersion(self, app_version): 60 def SetAppVersion(self, app_version):
57 self._app_version = app_version 61 self._app_version = app_version
58 62
59 class CronServletTest(unittest.TestCase): 63 class CronServletTest(unittest.TestCase):
60 @EnableLogging('info') 64 @EnableLogging('info')
61 def testEverything(self): 65 def testEverything(self):
(...skipping 166 matching lines...) Expand 10 before | Expand all | Expand 10 after
228 CronServlet(Request.ForTest('trunk'), delegate_for_test=delegate).Get() 232 CronServlet(Request.ForTest('trunk'), delegate_for_test=delegate).Get()
229 self.assertEqual(AppYamlHelper.GenerateAppYaml('2-1-0'), 233 self.assertEqual(AppYamlHelper.GenerateAppYaml('2-1-0'),
230 file_systems[-1].ReadSingle(APP_YAML).Get()) 234 file_systems[-1].ReadSingle(APP_YAML).Get())
231 self.assertEqual('<h1>y</h1> u not update!', 235 self.assertEqual('<h1>y</h1> u not update!',
232 file_systems[-1].ReadSingle(storage_html_path).Get()) 236 file_systems[-1].ReadSingle(storage_html_path).Get())
233 self.assertEqual('important content!', 237 self.assertEqual('important content!',
234 file_systems[-1].ReadSingle(static_txt_path).Get()) 238 file_systems[-1].ReadSingle(static_txt_path).Get())
235 239
236 if __name__ == '__main__': 240 if __name__ == '__main__':
237 unittest.main() 241 unittest.main()
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698