Docserver: Enable GitHub content providers.

chrome/common/extensions/docs/server2/content_providers_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 json
 import unittest
 
 from compiled_file_system import CompiledFileSystem
 from content_providers import ContentProviders
+from empty_dir_file_system import EmptyDirFileSystem
+from github_file_system_provider import GithubFileSystemProvider

[Jeffrey Yasskin, 2013/11/06 04:00:19]

I don't think you use this.

[not at google - send to devlin, 2013/11/06 23:00:13]

Done.

 from object_store_creator import ObjectStoreCreator
 from test_file_system import TestFileSystem
 from test_util import DisableLogging
 
 
 _HOST = 'https://developer.chrome.com'
 
 
 _CONTENT_PROVIDERS = {
   'apples': {
     'chromium': {
       'dir': 'apples'
     },
     'serveFrom': 'apples-dir',
   },
   'bananas': {
     'serveFrom': '',
     'chromium': {
       'dir': ''
     },
   },
+  'github-provider': {
+    'serveFrom': 'gh',
+    'github': {
+      'owner': 'GoogleChrome',
+      'repo': 'hello-world',
+    },
+  },
   'tomatoes': {
     'serveFrom': 'tomatoes-dir/are/a',
     'chromium': {
       'dir': 'tomatoes/are/a'
     },
   },
 }
 
 
 _FILE_SYSTEM_DATA = {
@@ skipping 16 matching lines @@
         'vegetable.txt': 'no they aren\'t',
         'fruit': {
           'cherry.txt': 'cherry tomatoes',
         },
       },
     },
   },
 }
 
 
+class _MockGithubFileSystemProvider(object):
+  '''A GithubFileSystemProvider imitation which records every call to Create
+  and returns them from GetAndReset.
+  '''
+
+  def __init__(self):
+    self._calls = []
+
+  def Create(self, owner, repo):
+    self._calls.append((owner, repo))
+    return EmptyDirFileSystem()
+
+  def GetAndReset(self):
+    calls = self._calls
+    self._calls = []
+    return calls
+
+
 class ContentProvidersTest(unittest.TestCase):
   def setUp(self):
+    self._github_fs_provider = _MockGithubFileSystemProvider()
     self._content_providers = ContentProviders(
         CompiledFileSystem.Factory(ObjectStoreCreator.ForTest()),
-        TestFileSystem(_FILE_SYSTEM_DATA))
+        TestFileSystem(_FILE_SYSTEM_DATA),
+        self._github_fs_provider)
 
   def testSimpleRootPath(self):
     provider = self._content_providers.GetByName('apples')
     self.assertEqual(
         'gala apples',
         provider.GetContentAndType(_HOST, 'gala.txt').Get().content)
     self.assertEqual(
         'granny smith apples',
         provider.GetContentAndType(_HOST, 'green/granny smith.txt').Get()
             .content)
@@ skipping 37 matching lines @@
     self.assertEqual('bananas', provider.name)
     self.assertEqual('tomatoes-dir/are', path)
     provider, path = self._content_providers.GetByServeFrom('')
     self.assertEqual('bananas', provider.name)
     self.assertEqual('', path)
 
   @DisableLogging('error')
   def testProviderNotFound(self):
     self.assertEqual(None, self._content_providers.GetByName('cabbages'))
 
+  def testGithubContentProvider(self):
+    provider, path = self._content_providers.GetByServeFrom('gh/some/path')
+    self.assertEqual('github-provider', provider.name)
+    self.assertEqual('some/path', path)
+    self.assertEqual([('GoogleChrome', 'hello-world')],
+                     self._github_fs_provider.GetAndReset())
+
 
 if __name__ == '__main__':
   unittest.main()