Home | History | Annotate | Download | only in gather 
      1 #!/usr/bin/env python
      2 # Copyright (c) 2012 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 '''Unit tests for TxtFile gatherer'''
      7 
      8 
      9 import os
     10 import sys
     11 if __name__ == '__main__':
     12   sys.path.append(os.path.join(os.path.dirname(__file__), '../..'))
     13 
     14 
     15 import StringIO
     16 import unittest
     17 
     18 from grit.gather import txt
     19 
     20 
     21 class TxtUnittest(unittest.TestCase):
     22   def testGather(self):
     23     input = StringIO.StringIO('Hello there\nHow are you?')
     24     gatherer = txt.TxtFile(input)
     25     gatherer.Parse()
     26     self.failUnless(gatherer.GetText() == input.getvalue())
     27     self.failUnless(len(gatherer.GetCliques()) == 1)
     28     self.failUnless(gatherer.GetCliques()[0].GetMessage().GetRealContent() ==
     29                     input.getvalue())
     30 
     31 
     32 if __name__ == '__main__':
     33   unittest.main()
     34 
     35