Index: chrome/common/extensions/docs/server2/template_data_source_test.py |
diff --git a/chrome/common/extensions/docs/server2/template_data_source_test.py b/chrome/common/extensions/docs/server2/template_data_source_test.py |
index 43a687397fe36f4e1940da1adf6748c07716e874..f406def1d288831cf661a561e6b31a9060fcb952 100755 |
--- a/chrome/common/extensions/docs/server2/template_data_source_test.py |
+++ b/chrome/common/extensions/docs/server2/template_data_source_test.py |
@@ -14,6 +14,10 @@ from third_party.handlebar import Handlebar |
class _FakeRequest(object): |
def __init__(self): |
+ self.headers = {} |
+ |
+class _FakeSamplesDataSource(object): |
+ def SetHeaders(self, headers): |
pass |
class TemplateDataSourceTest(unittest.TestCase): |
@@ -21,7 +25,7 @@ class TemplateDataSourceTest(unittest.TestCase): |
self._base_path = os.path.join('test_data', 'template_data_source') |
self._fake_api_data_source = {} |
self._fake_intro_data_source = {} |
- self._fake_samples_data_source = {} |
+ self._fake_samples_data_source = _FakeSamplesDataSource() |
def _ReadLocalFile(self, filename): |
with open(os.path.join(self._base_path, filename), 'r') as f: |