src/chrome/common/extensions/docs/server2/link_error_detector_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 file_system import FileNotFoundError
   9 from link_error_detector import LinkErrorDetector
  10 from servlet import Response
  11 from test_file_system import TestFileSystem
  12
  13 file_system = TestFileSystem({
  14   'docs': {
  15     'templates': {
  16       'public': {
  17         'apps': {
  18           '404.html': '404',
  19           'index.html': '''
  20             <h1 id="actual-top">Hello</h1>
  21             <a href="#top">world</p>
  22             <a href="#actual-top">!</a>
  23             <a href="broken.json"></a>
  24             <a href="crx.html"></a>
  25             ''',
  26           'crx.html': '''
  27             <a href="index.html#actual-top">back</a>
  28             <a href="broken.html"></a>
  29             <a href="devtools.events.html">do to underscore translation</a>
  30             ''',
  31           'devtools_events.html': '''
  32             <a href=" http://www.google.com/">leading space in href</a>
  33             <a href=" index.html">home</a>
  34             <a href="index.html#invalid"></a>
  35             <a href="fake.html#invalid"></a>
  36             ''',
  37           'unreachable.html': '''
  38             <p>so lonely</p>
  39             <a href="#aoesu"></a>
  40             <a href="invalid.html"></a>
  41             ''',
  42           'devtools_disconnected.html': ''
  43         }
  44       }
  45     },
  46     'static': {},
  47     'examples': {},
  48   }
  49 })
  50
  51 class LinkErrorDetectorTest(unittest.TestCase):
  52   def render(self, path):
  53     try:
  54       return Response(
  55           content=file_system.ReadSingle('docs/templates/public/' + path).Get(),
  56           status=200)
  57     except FileNotFoundError:
  58       return Response(status=404)
  59
  60   def testGetBrokenLinks(self):
  61     expected_broken_links = set([
  62       (404, 'apps/crx.html', 'apps/broken.html', 'target page not found'),
  63       (404, 'apps/index.html', 'apps/broken.json', 'target page not found'),
  64       (404, 'apps/unreachable.html', 'apps/invalid.html',
  65           'target page not found'),
  66       (404, 'apps/devtools_events.html', 'apps/fake.html#invalid',
  67           'target page not found'),
  68       (200, 'apps/devtools_events.html', 'apps/index.html#invalid',
  69           'target anchor not found'),
  70       (200, 'apps/unreachable.html', '#aoesu', 'target anchor not found')])
  71
  72     link_error_detector = LinkErrorDetector(
  73         file_system, self.render, 'templates/public', ('apps/index.html'))
  74     broken_links = link_error_detector.GetBrokenLinks()
  75
  76     self.assertEqual(expected_broken_links, set(broken_links))
  77
  78   def testGetOrphanedPages(self):
  79     expected_orphaned_pages = set([
  80       'apps/unreachable.html',
  81       'apps/devtools_disconnected.html'])
  82
  83     link_error_detector = LinkErrorDetector(
  84         file_system, self.render, 'templates/public', ('apps/crx.html',))
  85     orphaned_pages = link_error_detector.GetOrphanedPages()
  86
  87     self.assertEqual(expected_orphaned_pages, set(orphaned_pages))
  88
  89 if __name__ == '__main__':
  90   unittest.main()