Index: chrome/common/extensions/docs/server2/content_providers_test.py |
diff --git a/chrome/common/extensions/docs/server2/content_providers_test.py b/chrome/common/extensions/docs/server2/content_providers_test.py |
index 1dac2f03932a8d59410b4ff916ac0438215e85a4..48461afa946c6a3e922f7335c2b6f5e303469628 100755 |
--- a/chrome/common/extensions/docs/server2/content_providers_test.py |
+++ b/chrome/common/extensions/docs/server2/content_providers_test.py |
@@ -101,12 +101,14 @@ class _MockGithubFileSystemProvider(object): |
class ContentProvidersTest(unittest.TestCase): |
def setUp(self): |
+ object_store_creator = ObjectStoreCreator.ForTest() |
test_file_system = TestFileSystem(_FILE_SYSTEM_DATA, relative_to=EXTENSIONS) |
self._github_fs_provider = _MockGithubFileSystemProvider(test_file_system) |
object_store_creator = ObjectStoreCreator.ForTest() |
# TODO(mangini): create tests for GCS |
self._gcs_fs_provider = CloudStorageFileSystemProvider(object_store_creator) |
self._content_providers = ContentProviders( |
+ object_store_creator, |
CompiledFileSystem.Factory(object_store_creator), |
test_file_system, |
self._github_fs_provider, |
@@ -137,30 +139,38 @@ class ContentProvidersTest(unittest.TestCase): |
provider.GetContentAndType('apples/gala.txt').Get().content) |
def testSimpleServlet(self): |
- provider, path = self._content_providers.GetByServeFrom('apples-dir') |
+ provider, serve_from, path = self._content_providers.GetByServeFrom( |
+ 'apples-dir') |
self.assertEqual('apples', provider.name) |
+ self.assertEqual('apples-dir', serve_from) |
self.assertEqual('', path) |
- provider, path = self._content_providers.GetByServeFrom( |
+ provider, serve_from, path = self._content_providers.GetByServeFrom( |
'apples-dir/are/forever') |
self.assertEqual('apples', provider.name) |
+ self.assertEqual('apples-dir', serve_from) |
self.assertEqual('are/forever', path) |
def testComplexServlet(self): |
- provider, path = self._content_providers.GetByServeFrom( |
+ provider, serve_from, path = self._content_providers.GetByServeFrom( |
'tomatoes-dir/are/a') |
self.assertEqual('tomatoes', provider.name) |
+ self.assertEqual('tomatoes-dir/are/a', serve_from) |
self.assertEqual('', path) |
- provider, path = self._content_providers.GetByServeFrom( |
+ provider, serve_from, path = self._content_providers.GetByServeFrom( |
'tomatoes-dir/are/a/fruit/they/are') |
self.assertEqual('tomatoes', provider.name) |
+ self.assertEqual('tomatoes-dir/are/a', serve_from) |
self.assertEqual('fruit/they/are', path) |
def testEmptyStringServlet(self): |
- provider, path = self._content_providers.GetByServeFrom('tomatoes-dir/are') |
+ provider, serve_from, path = self._content_providers.GetByServeFrom( |
+ 'tomatoes-dir/are') |
self.assertEqual('bananas', provider.name) |
+ self.assertEqual('', serve_from) |
self.assertEqual('tomatoes-dir/are', path) |
- provider, path = self._content_providers.GetByServeFrom('') |
+ provider, serve_from, path = self._content_providers.GetByServeFrom('') |
self.assertEqual('bananas', provider.name) |
+ self.assertEqual('', serve_from) |
self.assertEqual('', path) |
@DisableLogging('error') |
@@ -168,9 +178,10 @@ class ContentProvidersTest(unittest.TestCase): |
self.assertEqual(None, self._content_providers.GetByName('cabbages')) |
def testGithubContentProvider(self): |
- provider, path = self._content_providers.GetByServeFrom( |
+ provider, serve_from, path = self._content_providers.GetByServeFrom( |
'gh/apples/green/granny smith.txt') |
self.assertEqual('github-provider', provider.name) |
+ self.assertEqual('gh', serve_from) |
self.assertEqual('apples/green/granny smith.txt', path) |
self.assertEqual([('GoogleChrome', 'hello-world')], |
self._github_fs_provider.GetAndReset()) |
@@ -179,9 +190,10 @@ class ContentProvidersTest(unittest.TestCase): |
provider.GetContentAndType(path).Get().content) |
def testGithubContentProviderWithDir(self): |
- provider, path = self._content_providers.GetByServeFrom( |
+ provider, serve_from, path = self._content_providers.GetByServeFrom( |
'gh2/fruit/cherry.txt') |
self.assertEqual('github-provider-with-dir', provider.name) |
+ self.assertEqual('gh2', serve_from) |
self.assertEqual('fruit/cherry.txt', path) |
self.assertEqual([('SomeOwner', 'some-repo')], |
self._github_fs_provider.GetAndReset()) |