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
@@ skipping 11 matching lines @@
       'dir': 'apples'
     },
     'serveFrom': 'apples-dir',
   },
   'bananas': {
     'serveFrom': '',
     'chromium': {
       'dir': ''
     },
   },
+  'github-provider': {
+    'serveFrom': 'gh',
+    'github': {
+      'owner': 'GoogleChrome',
+      'repo': 'hello-world',
+    },
+  },
+  'github-provider-with-dir': {
+    'serveFrom': 'gh2',
+    'github': {
+      'dir': 'tomatoes/are/a',
+      'owner': 'SomeOwner',
+      'repo': 'some-repo',
+    },
+  },
   '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, file_system):
+    self._file_system = file_system
+    self._calls = []
+
+  def Create(self, owner, repo):
+    self._calls.append((owner, repo))
+    return self._file_system
+
+  def GetAndReset(self):
+    calls = self._calls
+    self._calls = []
+    return calls
+
+
 class ContentProvidersTest(unittest.TestCase):
   def setUp(self):
+    test_file_system = TestFileSystem(_FILE_SYSTEM_DATA)
+    self._github_fs_provider = _MockGithubFileSystemProvider(test_file_system)
     self._content_providers = ContentProviders(
         CompiledFileSystem.Factory(ObjectStoreCreator.ForTest()),
-        TestFileSystem(_FILE_SYSTEM_DATA))
+        test_file_system,
+        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/apples/green/granny smith.txt')
+    self.assertEqual('github-provider', provider.name)
+    self.assertEqual('apples/green/granny smith.txt', path)
+    self.assertEqual([('GoogleChrome', 'hello-world')],
+                     self._github_fs_provider.GetAndReset())
+    self.assertEqual(
+        'granny smith apples',
+        provider.GetContentAndType(_HOST, path).Get().content)
+
+  def testGithubContentProviderWithDir(self):
+    provider, path = self._content_providers.GetByServeFrom(
+        'gh2/fruit/cherry.txt')
+    self.assertEqual('github-provider-with-dir', provider.name)
+    self.assertEqual('fruit/cherry.txt', path)
+    self.assertEqual([('SomeOwner', 'some-repo')],
+                     self._github_fs_provider.GetAndReset())
+    self.assertEqual(
+        'cherry tomatoes',
+        provider.GetContentAndType(_HOST, path).Get().content)
 
 if __name__ == '__main__':
   unittest.main()