| 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())
|
|
|