Fixed problem with large writes in large objects code

1 files changed, 15 insertions, 0 deletions

diff --git a/tests/test_lobject.py b/tests/test_lobject.py
index 799492c..fc0d852 100644
--- a/tests/test_lobject.py
+++ b/tests/test_lobject.py
@@ -98,6 +98,11 @@ class LargeObjectTests(unittest.TestCase):
         lo = self.conn.lobject()
         self.assertEqual(lo.write("some data"), len("some data"))
 
+    def test_write_large(self):
+        lo = self.conn.lobject()
+        data = "data" * 1000000
+        self.assertEqual(lo.write(data), len(data))
+
     def test_read(self):
         lo = self.conn.lobject()
         length = lo.write("some data")
@@ -107,6 +112,16 @@ class LargeObjectTests(unittest.TestCase):
         self.assertEqual(lo.read(4), "some")
         self.assertEqual(lo.read(), " data")
 
+    def test_read_large(self):
+        lo = self.conn.lobject()
+        data = "data" * 1000000
+        length = lo.write("some"+data)
+        lo.close()
+
+        lo = self.conn.lobject(lo.oid)
+        self.assertEqual(lo.read(4), "some")
+        self.assertEqual(lo.read(), data)
+
     def test_seek_tell(self):
         lo = self.conn.lobject()
         length = lo.write("some data")