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", ]))