Rename WebKitTools to Tools
[WebKit-https.git] / Tools / Scripts / webkitpy / common / array_stream_unittest.py
1 #!/usr/bin/python
2 # Copyright (C) 2010 Google Inc. All rights reserved.
3 #
4 # Redistribution and use in source and binary forms, with or without
5 # modification, are permitted provided that the following conditions are
6 # met:
7 #
8 #     * Redistributions of source code must retain the above copyright
9 # notice, this list of conditions and the following disclaimer.
10 #     * Redistributions in binary form must reproduce the above
11 # copyright notice, this list of conditions and the following disclaimer
12 # in the documentation and/or other materials provided with the
13 # distribution.
14 #     * Neither the name of Google Inc. nor the names of its
15 # contributors may be used to endorse or promote products derived from
16 # this software without specific prior written permission.
17 #
18 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29
30 """Unit tests for array_stream.py."""
31
32 import pdb
33 import unittest
34
35 from webkitpy.common.array_stream import ArrayStream
36
37
38 class ArrayStreamTest(unittest.TestCase):
39     def assertEmpty(self, a_stream):
40         self.assertTrue(a_stream.empty())
41
42     def assertNotEmpty(self, a_stream):
43         self.assertFalse(a_stream.empty())
44
45     def assertContentsMatch(self, a_stream, contents):
46         self.assertEquals(a_stream.get(), contents)
47
48     def test_basics(self):
49         a = ArrayStream()
50         self.assertEmpty(a)
51         self.assertContentsMatch(a, [])
52
53         a.flush()
54         self.assertEmpty(a)
55         self.assertContentsMatch(a, [])
56
57         a.write("foo")
58         a.write("bar")
59         self.assertNotEmpty(a)
60         self.assertContentsMatch(a, ["foo", "bar"])
61
62         a.flush()
63         self.assertNotEmpty(a)
64         self.assertContentsMatch(a, ["foo", "bar"])
65
66         a.reset()
67         self.assertEmpty(a)
68         self.assertContentsMatch(a, [])
69
70         self.assertEquals(str(a), "<ArrayStream: []>")
71
72         a.write("foo")
73         self.assertNotEmpty(a)
74         self.assertContentsMatch(a, ["foo"])
75         self.assertEquals(str(a), "<ArrayStream: ['foo']>")
76
77 if __name__ == '__main__':
78     unittest.main()