Index: chrome/common/extensions/docs/server2/template_data_source_test.py |
=================================================================== |
--- chrome/common/extensions/docs/server2/template_data_source_test.py (revision 202781) |
+++ chrome/common/extensions/docs/server2/template_data_source_test.py (working copy) |
@@ -28,6 +28,23 @@ |
def Create(self, *args, **optargs): |
return self._input_dict |
+class _FakeSidenavDataSource(object): |
+ def __init__(self, input_dict): |
+ self._input_dict = input_dict |
+ |
+ def get(self, key): |
+ return self._input_dict.get(key) |
+ |
+ def SetPath(self, path): |
+ pass |
+ |
+ class Factory(object): |
+ def __init__(self, input_dict={}): |
+ self._input_dict = input_dict |
+ |
+ def Create(self, *args, **optargs): |
+ return _FakeSidenavDataSource(self._input_dict) |
+ |
class TemplateDataSourceTest(unittest.TestCase): |
def setUp(self): |
self._base_path = os.path.join(sys.path[0], |
@@ -36,7 +53,7 @@ |
self._fake_api_list_data_source_factory = _FakeFactory() |
self._fake_intro_data_source_factory = _FakeFactory() |
self._fake_samples_data_source_factory = _FakeFactory() |
- self._fake_sidenav_data_source_factory = _FakeFactory() |
+ self._fake_sidenav_data_source_factory = _FakeSidenavDataSource.Factory() |
def _ReadLocalFile(self, filename): |
with open(os.path.join(self._base_path, filename), 'r') as f: |
@@ -73,7 +90,7 @@ |
reference_resolver_factory, |
'.', |
'.', |
- '/static')) |
+ '')) |
def testSimple(self): |
self._base_path = os.path.join(self._base_path, 'simple') |