def test_consume(self):
sp = util.StringIOProxy()
self.assertEquals(sp.getvalue(), '')
self.assertEquals(sp.tell(), 0)
sp.consume()
self.assertEquals(sp.getvalue(), '')
self.assertEquals(sp.tell(), 0)
sp = util.StringIOProxy('foobar')
self.assertEquals(sp.getvalue(), 'foobar')
self.assertEquals(sp.tell(), 0)
sp.seek(3)
self.assertEquals(sp.tell(), 3)
sp.consume()
self.assertEquals(sp.getvalue(), 'bar')
self.assertEquals(sp.tell(), 0)
# from ticket 451 - http://pyamf.org/ticket/451
sp = util.StringIOProxy('abcdef')
# move the stream pos to the end
sp.read()
self.assertEquals(len(sp), 6)
sp.consume()
self.assertEquals(len(sp), 0)
def test_close(self):
sp = util.StringIOProxy()
sp.close()
self.assertEquals(len(sp), 0)
self.assertRaises(ValueError, sp.write, 'asdfasdf')
def test_seek(self):
sp = util.StringIOProxy('abcdefghijklmnopqrstuvwxyz')
self.assertEquals(sp.getvalue(), 'abcdefghijklmnopqrstuvwxyz')
self.assertEquals(sp.tell(), 0)
# Relative to the beginning of the stream
sp.seek(0, 0)
self.assertEquals(sp.tell(), 0)
self.assertEquals(sp.getvalue(), 'abcdefghijklmnopqrstuvwxyz')
self.assertEquals(sp.read(1), 'a')
self.assertEquals(len(sp), 26)
sp.seek(10, 0)
self.assertEquals(sp.tell(), 10)
self.assertEquals(sp.getvalue(), 'abcdefghijklmnopqrstuvwxyz')
self.assertEquals(sp.read(1), 'k')
self.assertEquals(len(sp), 26)
sp.seek(-5, 1)
self.assertEquals(sp.tell(), 6)
self.assertEquals(sp.getvalue(), 'abcdefghijklmnopqrstuvwxyz')
self.assertEquals(sp.read(1), 'g')
self.assertEquals(len(sp), 26)
sp.seek(-3, 2)
self.assertEquals(sp.tell(), 23)
self.assertEquals(sp.getvalue(), 'abcdefghijklmnopqrstuvwxyz')
self.assertEquals(sp.read(1), 'x')
self.assertEquals(len(sp), 26)
def test_getvalue(self):
sp = util.StringIOProxy()
sp.write('asdfasdf')
self.assertEquals(sp.getvalue(), 'asdfasdf')
sp.write('spam')
self.assertEquals(sp.getvalue(), 'asdfasdfspam')
def test_read(self):
sp = util.StringIOProxy('this is a test')
self.assertEquals(len(sp), 14)
self.assertEquals(sp.read(1), 't')
self.assertEquals(sp.getvalue(), 'this is a test')
self.assertEquals(len(sp), 14)
self.assertEquals(sp.read(10), 'his is a t')
self.assertEquals(sp.read(), 'est')
def test_truncate(self):
sp = util.StringIOProxy('abcdef')
self.assertEquals(sp.getvalue(), 'abcdef')
self.assertEquals(len(sp), 6)
sp.truncate()
self.assertEquals(sp.getvalue(), '')
self.assertEquals(len(sp), 0)
def test_len(self):
sp = util.StringIOProxy()
self.assertEquals(sp.getvalue(), '')
self.assertEquals(len(sp), 0)
self.assertEquals(sp.tell(), 0)
sp.write('xyz')
self.assertEquals(len(sp), 3)
sp = util.StringIOProxy('foo')
self.assertEquals(len(sp), 3)
sp.seek(0, 2)
sp.write('xyz')
self.assertEquals(len(sp), 6)
def test_readline(self):
sp = util.StringIOProxy("this is a test\nspam and eggs")
self.assertEquals(len(sp), 28)
self.assertEquals(sp.getvalue(), "this is a test\nspam and eggs")
self.assertEquals(sp.readline(), 'this is a test\n')
self.assertEquals(len(sp), 28)
self.assertEquals(sp.getvalue(), "this is a test\nspam and eggs")
self.assertEquals(sp.readline(), 'spam and eggs')
def test_flush(self):
sp = util.StringIOProxy('spameggs')
self.assertEquals(sp.getvalue(), 'spameggs')
self.assertEquals(len(sp), 8)
self.assertEquals(sp.tell(), 0)
sp.flush()
self.assertEquals(sp.getvalue(), 'spameggs')
self.assertEquals(len(sp), 8)
self.assertEquals(sp.tell(), 0)
def test_writelines(self):
lines = ["line 1", "line 2", "line 3", "line 4"]
sp = util.StringIOProxy()
self.assertEquals(sp.getvalue(), '')
self.assertEquals(len(sp), 0)
self.assertEquals(sp.tell(), 0)
sp.writelines(lines)
self.assertEquals(sp.getvalue(), "".join(lines))
self.assertEquals(len(sp), 24)
self.assertEquals(sp.tell(), 24)
def test_write(self):
sp = util.StringIOProxy()
self.assertEquals(sp.getvalue(), '')
self.assertEquals(len(sp), 0)
self.assertEquals(sp.tell(), 0)
sp.write('hello')
self.assertEquals(sp.getvalue(), 'hello')
self.assertEquals(len(sp), 5)
self.assertEquals(sp.tell(), 5)
sp = util.StringIOProxy('xyz')
self.assertEquals(sp.getvalue(), 'xyz')
self.assertEquals(len(sp), 3)
self.assertEquals(sp.tell(), 0)
sp.write('abc')
self.assertEquals(sp.getvalue(), 'abc')
self.assertEquals(len(sp), 3)
self.assertEquals(sp.tell(), 3)
def test_consume(self):
sp = util.StringIOProxy()
self.assertEquals(sp.getvalue(), '')
self.assertEquals(sp.tell(), 0)
sp.consume()
self.assertEquals(sp.getvalue(), '')
self.assertEquals(sp.tell(), 0)
sp = util.StringIOProxy('foobar')
self.assertEquals(sp.getvalue(), 'foobar')
self.assertEquals(sp.tell(), 0)
sp.seek(3)
self.assertEquals(sp.tell(), 3)
sp.consume()
self.assertEquals(sp.getvalue(), 'bar')
self.assertEquals(sp.tell(), 0)
def test_create(self):
sp = util.StringIOProxy()
self.assertEquals(sp._buffer.tell(), 0)
self.assertEquals(sp._buffer.getvalue(), '')
self.assertEquals(len(sp), 0)
self.assertEquals(sp.getvalue(), '')
sp = util.StringIOProxy(None)
self.assertEquals(sp._buffer.tell(), 0)
self.assertEquals(sp._buffer.getvalue(), '')
self.assertEquals(len(sp), 0)
self.assertEquals(sp.getvalue(), '')
sp = util.StringIOProxy('')
self.assertEquals(sp._buffer.tell(), 0)
self.assertEquals(sp._buffer.getvalue(), '')
self.assertEquals(len(sp), 0)
self.assertEquals(sp.getvalue(), '')
sp = util.StringIOProxy('spam')
self.assertEquals(sp._buffer.tell(), 0)
self.assertEquals(sp._buffer.getvalue(), 'spam')
self.assertEquals(len(sp), 4)
self.assertEquals(sp.getvalue(), 'spam')
sp = util.StringIOProxy(StringIO('this is a test'))
self.assertEquals(sp._buffer.tell(), 0)
self.assertEquals(sp._buffer.getvalue(), 'this is a test')
self.assertEquals(len(sp), 14)
self.assertEquals(sp.getvalue(), 'this is a test')
self.assertRaises(TypeError, util.StringIOProxy, self)
def test_tell(self):
sp = util.StringIOProxy('abcdefghijklmnopqrstuvwxyz')
self.assertEquals(sp.getvalue(), 'abcdefghijklmnopqrstuvwxyz')
self.assertEquals(len(sp), 26)
self.assertEquals(sp.tell(), 0)
sp.read(1)
self.assertEquals(sp.tell(), 1)
self.assertEquals(sp.getvalue(), 'abcdefghijklmnopqrstuvwxyz')
self.assertEquals(len(sp), 26)
sp.read(5)
self.assertEquals(sp.tell(), 6)
def test_readlines(self):
sp = util.StringIOProxy("\n".join([
"line 1",
"line 2",
"line 3",
"line 4",
]))
self.assertEquals(len(sp), 27)
self.assertEquals(sp.readlines(), [
"line 1\n",
"line 2\n",
"line 3\n",
"line 4",
])
self.assertEquals(len(sp), 27)
self.assertEquals(sp.getvalue(), "\n".join([
"line 1",
"line 2",
"line 3",
"line 4",
]))
