chrome/common/extensions/docs/server2/document_renderer_test.py

   1 #!/usr/bin/env python
   2 # Copyright 2013 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import unittest
   7
   8 from document_renderer import DocumentRenderer
   9
  10
  11 class _FakeTableOfContentsRenderer(object):
  12   def __init__(self, string):
  13     self._string = string
  14
  15   def Render(self, _):
  16     return self._string, []
  17
  18
  19 class DocumentRendererUnittest(unittest.TestCase):
  20   def setUp(self):
  21     self._renderer = DocumentRenderer(_FakeTableOfContentsRenderer('t-o-c'))
  22
  23   def testNothingToSubstitute(self):
  24     document = 'hello world'
  25
  26     text, warnings = self._renderer.Render(document)
  27     self.assertEqual(document, text)
  28     self.assertEqual([], warnings)
  29
  30     text, warnings = self._renderer.Render(document, render_title=True)
  31     self.assertEqual(document, text)
  32     self.assertEqual(['Expected a title'], warnings)
  33
  34   def testTitles(self):
  35     document = '<h1>title</h1> then $(title) then another $(title)'
  36
  37     text, warnings = self._renderer.Render(document)
  38     self.assertEqual(document, text)
  39     self.assertEqual(['Found unexpected title "title"'], warnings)
  40
  41     text, warnings = self._renderer.Render(document, render_title=True)
  42     self.assertEqual('<h1>title</h1> then title then another $(title)', text)
  43     self.assertEqual([], warnings)
  44
  45   def testTocs(self):
  46     document = ('here is a toc $(table_of_contents) '
  47                 'and another $(table_of_contents)')
  48     expected_document = 'here is a toc t-o-c and another $(table_of_contents)'
  49
  50     text, warnings = self._renderer.Render(document)
  51     self.assertEqual(expected_document, text)
  52     self.assertEqual([], warnings)
  53
  54     text, warnings = self._renderer.Render(document, render_title=True)
  55     self.assertEqual(expected_document, text)
  56     self.assertEqual(['Expected a title'], warnings)
  57
  58   def testTitleAndToc(self):
  59     document = '<h1>title</h1> $(title) and $(table_of_contents)'
  60
  61     text, warnings = self._renderer.Render(document)
  62     self.assertEqual('<h1>title</h1> $(title) and t-o-c', text)
  63     self.assertEqual(['Found unexpected title "title"'], warnings)
  64
  65     text, warnings = self._renderer.Render(document, render_title=True)
  66     self.assertEqual('<h1>title</h1> title and t-o-c', text)
  67     self.assertEqual([], warnings)
  68
  69
  70 if __name__ == '__main__':
  71   unittest.main()