Home | History | Annotate | Download | only in test 
      1 #!/usr/bin/env python
      2 #
      3 # Copyright 2011, Google Inc.
      4 # All rights reserved.
      5 #
      6 # Redistribution and use in source and binary forms, with or without
      7 # modification, are permitted provided that the following conditions are
      8 # met:
      9 #
     10 #     * Redistributions of source code must retain the above copyright
     11 # notice, this list of conditions and the following disclaimer.
     12 #     * Redistributions in binary form must reproduce the above
     13 # copyright notice, this list of conditions and the following disclaimer
     14 # in the documentation and/or other materials provided with the
     15 # distribution.
     16 #     * Neither the name of Google Inc. nor the names of its
     17 # contributors may be used to endorse or promote products derived from
     18 # this software without specific prior written permission.
     19 #
     20 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     21 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     22 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
     23 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
     24 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     25 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
     26 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     27 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     28 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     29 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     30 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     31 
     32 
     33 """Tests for memorizingfile module."""
     34 
     35 
     36 import StringIO
     37 import unittest
     38 
     39 import set_sys_path  # Update sys.path to locate mod_pywebsocket module.
     40 
     41 from mod_pywebsocket import memorizingfile
     42 
     43 
     44 class UtilTest(unittest.TestCase):
     45     """A unittest for memorizingfile module."""
     46 
     47     def check(self, memorizing_file, num_read, expected_list):
     48         for unused in range(num_read):
     49             memorizing_file.readline()
     50         actual_list = memorizing_file.get_memorized_lines()
     51         self.assertEqual(len(expected_list), len(actual_list))
     52         for expected, actual in zip(expected_list, actual_list):
     53             self.assertEqual(expected, actual)
     54 
     55     def check_with_size(self, memorizing_file, read_size, expected_list):
     56         read_list = []
     57         read_line = ''
     58         while True:
     59             line = memorizing_file.readline(read_size)
     60             line_length = len(line)
     61             self.assertTrue(line_length <= read_size)
     62             if line_length == 0:
     63                 if read_line != '':
     64                     read_list.append(read_line)
     65                 break
     66             read_line += line
     67             if line[line_length - 1] == '\n':
     68                 read_list.append(read_line)
     69                 read_line = ''
     70         actual_list = memorizing_file.get_memorized_lines()
     71         self.assertEqual(len(expected_list), len(actual_list))
     72         self.assertEqual(len(expected_list), len(read_list))
     73         for expected, actual, read in zip(expected_list, actual_list,
     74                                           read_list):
     75             self.assertEqual(expected, actual)
     76             self.assertEqual(expected, read)
     77 
     78     def test_get_memorized_lines(self):
     79         memorizing_file = memorizingfile.MemorizingFile(StringIO.StringIO(
     80                 'Hello\nWorld\nWelcome'))
     81         self.check(memorizing_file, 3, ['Hello\n', 'World\n', 'Welcome'])
     82 
     83     def test_get_memorized_lines_limit_memorized_lines(self):
     84         memorizing_file = memorizingfile.MemorizingFile(StringIO.StringIO(
     85                 'Hello\nWorld\nWelcome'), 2)
     86         self.check(memorizing_file, 3, ['Hello\n', 'World\n'])
     87 
     88     def test_get_memorized_lines_empty_file(self):
     89         memorizing_file = memorizingfile.MemorizingFile(StringIO.StringIO(
     90                 ''))
     91         self.check(memorizing_file, 10, [])
     92 
     93     def test_get_memorized_lines_with_size(self):
     94         for size in range(1, 10):
     95             memorizing_file = memorizingfile.MemorizingFile(StringIO.StringIO(
     96                 'Hello\nWorld\nWelcome'))
     97             self.check_with_size(memorizing_file, size,
     98                                  ['Hello\n', 'World\n', 'Welcome'])
     99 
    100 if __name__ == '__main__':
    101     unittest.main()
    102 
    103 
    104 # vi:sts=4 sw=4 et
    105