lib/subunit/python/subunit/tests/test_chunked.py

   1 #
   2 #  subunit: extensions to python unittest to get test results from subprocesses.
   3 #  Copyright (C) 2005  Robert Collins <robertc@robertcollins.net>
   4 #
   5 #  Licensed under either the Apache License, Version 2.0 or the BSD 3-clause
   6 #  license at the users choice. A copy of both licenses are available in the
   7 #  project source as Apache-2.0 and BSD. You may not use this file except in
   8 #  compliance with one of these two licences.
   9 #
  10 #  Unless required by applicable law or agreed to in writing, software
  11 #  distributed under these licenses is distributed on an "AS IS" BASIS, WITHOUT
  12 #  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
  13 #  license you chose for the specific language governing permissions and
  14 #  limitations under that license.
  15 #
  16
  17 from cStringIO import StringIO
  18 import unittest
  19
  20 import subunit.chunked
  21
  22
  23 def test_suite():
  24     loader = subunit.tests.TestUtil.TestLoader()
  25     result = loader.loadTestsFromName(__name__)
  26     return result
  27
  28
  29 class TestDecode(unittest.TestCase):
  30
  31     def setUp(self):
  32         unittest.TestCase.setUp(self)
  33         self.output = StringIO()
  34         self.decoder = subunit.chunked.Decoder(self.output)
  35
  36     def test_close_read_length_short_errors(self):
  37         self.assertRaises(ValueError, self.decoder.close)
  38
  39     def test_close_body_short_errors(self):
  40         self.assertEqual(None, self.decoder.write('2\r\na'))
  41         self.assertRaises(ValueError, self.decoder.close)
  42
  43     def test_close_body_buffered_data_errors(self):
  44         self.assertEqual(None, self.decoder.write('2\r'))
  45         self.assertRaises(ValueError, self.decoder.close)
  46
  47     def test_close_after_finished_stream_safe(self):
  48         self.assertEqual(None, self.decoder.write('2\r\nab'))
  49         self.assertEqual('', self.decoder.write('0\r\n'))
  50         self.decoder.close()
  51
  52     def test_decode_nothing(self):
  53         self.assertEqual('', self.decoder.write('0\r\n'))
  54         self.assertEqual('', self.output.getvalue())
  55
  56     def test_decode_serialised_form(self):
  57         self.assertEqual(None, self.decoder.write("F\r\n"))
  58         self.assertEqual(None, self.decoder.write("serialised\n"))
  59         self.assertEqual('', self.decoder.write("form0\r\n"))
  60
  61     def test_decode_short(self):
  62         self.assertEqual('', self.decoder.write('3\r\nabc0\r\n'))
  63         self.assertEqual('abc', self.output.getvalue())
  64
  65     def test_decode_combines_short(self):
  66         self.assertEqual('', self.decoder.write('6\r\nabcdef0\r\n'))
  67         self.assertEqual('abcdef', self.output.getvalue())
  68
  69     def test_decode_excess_bytes_from_write(self):
  70         self.assertEqual('1234', self.decoder.write('3\r\nabc0\r\n1234'))
  71         self.assertEqual('abc', self.output.getvalue())
  72
  73     def test_decode_write_after_finished_errors(self):
  74         self.assertEqual('1234', self.decoder.write('3\r\nabc0\r\n1234'))
  75         self.assertRaises(ValueError, self.decoder.write, '')
  76
  77     def test_decode_hex(self):
  78         self.assertEqual('', self.decoder.write('A\r\n12345678900\r\n'))
  79         self.assertEqual('1234567890', self.output.getvalue())
  80
  81     def test_decode_long_ranges(self):
  82         self.assertEqual(None, self.decoder.write('10000\r\n'))
  83         self.assertEqual(None, self.decoder.write('1' * 65536))
  84         self.assertEqual(None, self.decoder.write('10000\r\n'))
  85         self.assertEqual(None, self.decoder.write('2' * 65536))
  86         self.assertEqual('', self.decoder.write('0\r\n'))
  87         self.assertEqual('1' * 65536 + '2' * 65536, self.output.getvalue())
  88
  89
  90 class TestEncode(unittest.TestCase):
  91
  92     def setUp(self):
  93         unittest.TestCase.setUp(self)
  94         self.output = StringIO()
  95         self.encoder = subunit.chunked.Encoder(self.output)
  96
  97     def test_encode_nothing(self):
  98         self.encoder.close()
  99         self.assertEqual('0\r\n', self.output.getvalue())
 100
 101     def test_encode_empty(self):
 102         self.encoder.write('')
 103         self.encoder.close()
 104         self.assertEqual('0\r\n', self.output.getvalue())
 105
 106     def test_encode_short(self):
 107         self.encoder.write('abc')
 108         self.encoder.close()
 109         self.assertEqual('3\r\nabc0\r\n', self.output.getvalue())
 110
 111     def test_encode_combines_short(self):
 112         self.encoder.write('abc')
 113         self.encoder.write('def')
 114         self.encoder.close()
 115         self.assertEqual('6\r\nabcdef0\r\n', self.output.getvalue())
 116
 117     def test_encode_over_9_is_in_hex(self):
 118         self.encoder.write('1234567890')
 119         self.encoder.close()
 120         self.assertEqual('A\r\n12345678900\r\n', self.output.getvalue())
 121
 122     def test_encode_long_ranges_not_combined(self):
 123         self.encoder.write('1' * 65536)
 124         self.encoder.write('2' * 65536)
 125         self.encoder.close()
 126         self.assertEqual('10000\r\n' + '1' * 65536 + '10000\r\n' +
 127             '2' * 65536 + '0\r\n', self.output.getvalue())