config/tests/unit-LineEndings.py

   1 import unittest
   2
   3 from StringIO import StringIO
   4 import os
   5 import sys
   6 import os.path
   7 sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
   8
   9 from Preprocessor import Preprocessor
  10
  11 class TestLineEndings(unittest.TestCase):
  12   """
  13   Unit tests for the Context class
  14   """
  15
  16   def setUp(self):
  17     self.pp = Preprocessor()
  18     self.pp.out = StringIO()
  19     self.tempnam = os.tempnam('.')
  20
  21   def tearDown(self):
  22     os.remove(self.tempnam)
  23
  24   def createFile(self, lineendings):
  25     f = open(self.tempnam, 'wb')
  26     for line, ending in zip(['a', '#literal b', 'c'], lineendings):
  27       f.write(line+ending)
  28     f.close()
  29
  30   def testMac(self):
  31     self.createFile(['\x0D']*3)
  32     self.pp.do_include(self.tempnam)
  33     self.assertEquals(self.pp.out.getvalue(), 'a\nb\nc\n')
  34
  35   def testUnix(self):
  36     self.createFile(['\x0A']*3)
  37     self.pp.do_include(self.tempnam)
  38     self.assertEquals(self.pp.out.getvalue(), 'a\nb\nc\n')
  39
  40   def testWindows(self):
  41     self.createFile(['\x0D\x0A']*3)
  42     self.pp.do_include(self.tempnam)
  43     self.assertEquals(self.pp.out.getvalue(), 'a\nb\nc\n')
  44
  45 if __name__ == '__main__':
  46   unittest.main()