test/test_dependencies.py

   1 #! /usr/bin/env python
   2
   3 # $Id$
   4 # Author: Lea Wiemann <LeWiemann@gmail.com>
   5 # Copyright: This module has been placed in the public domain.
   6
   7 """
   8 Test module for the --record-dependencies option.
   9 """
  10
  11 import os.path
  12 import unittest
  13 import sys
  14 import DocutilsTestSupport              # must be imported before docutils
  15 import docutils.core
  16 import docutils.utils
  17
  18
  19 class RecordDependenciesTests(unittest.TestCase):
  20
  21     # docutils.utils.DependencyList records relative URLs, not platform paths,
  22     # so use "/" as a path separator even on Windows (not os.path.join).
  23
  24     def get_record(self, **settings):
  25         recordfile = 'record.txt'
  26         settings.setdefault('source_path',
  27                             os.path.join('data', 'dependencies.txt'))
  28         settings.setdefault('settings_overrides', {})
  29         settings['settings_overrides'] = settings['settings_overrides'].copy()
  30         settings['settings_overrides']['_disable_config'] = 1
  31         if 'record_dependencies' not in settings['settings_overrides']:
  32             settings['settings_overrides']['record_dependencies'] = \
  33                 docutils.utils.DependencyList(recordfile)
  34         docutils.core.publish_file(destination=DocutilsTestSupport.DevNull(),
  35                                    **settings)
  36         settings['settings_overrides']['record_dependencies'].close()
  37         return open(recordfile).read().splitlines()
  38
  39     def test_dependencies(self):
  40         self.assertEqual(self.get_record(),
  41                          ['data/include.txt', 'data/raw.txt'])
  42         self.assertEqual(self.get_record(writer_name='latex'),
  43                          ['data/include.txt',
  44                           'data/raw.txt',
  45                           # this is a URL, not a path:
  46                           'some_image.png'])
  47
  48     def test_csv_dependencies(self):
  49         try:
  50             import csv
  51             self.assertEqual(
  52                 self.get_record(source_path=os.path.join('data',
  53                                                          'csv_dep.txt')),
  54                 ['data/csv_data.txt'])
  55         except ImportError:
  56             pass
  57
  58     def test_stylesheet_dependencies(self):
  59         # Parameters to publish_file.
  60         s = {'settings_overrides': {}}
  61         so = s['settings_overrides']
  62         so['embed_stylesheet'] = 0
  63         so['stylesheet_path'] = os.path.join('data', 'stylesheet.txt')
  64         so['stylesheet'] = None
  65         s['writer_name'] = 'html'
  66         self.assert_(os.path.join('data', 'stylesheet.txt') not in
  67                      self.get_record(**s))
  68         so['embed_stylesheet'] = 1
  69         self.assert_('data/stylesheet.txt' in self.get_record(**s))
  70         del so['embed_stylesheet']
  71         s['writer_name'] = 'latex'
  72         self.assert_('data/stylesheet.txt' in self.get_record(**s))
  73
  74
  75 if __name__ == '__main__':
  76     unittest.main()