From 923772f4ce272b9b9ef93f58eaecf5be64fdccc0 Mon Sep 17 00:00:00 2001
From: Fabien MARTY <fabien.marty@gmail.com>
Date: Wed, 4 Mar 2015 15:40:57 +0100
Subject: [PATCH] add a magic __len__ method and the corresponding test

---
 tests/test_writebuffer.py | 2 ++
 tornadis/write_buffer.py  | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/tests/test_writebuffer.py b/tests/test_writebuffer.py
index 7954837..82b9421 100644
--- a/tests/test_writebuffer.py
+++ b/tests/test_writebuffer.py
@@ -29,10 +29,12 @@ def test_empty_write_buffer(self):
         self.assertEquals(s, b"")
         c = x.get_chunk(4096)
         self.assertEquals(c, b"")
+        self.assertEquals(len(x), 0)
 
     def test_write_buffer1(self):
         b = self._make_test_buffer()
         s = bytes(b)
+        self.assertEquals(len(b), 9)
         self.assertEquals(s, b"123456789")
         self.assertFalse(b.is_empty())
         self.assertEquals(b._total_length, 9)
diff --git a/tornadis/write_buffer.py b/tornadis/write_buffer.py
index b4c8e84..c82e880 100644
--- a/tornadis/write_buffer.py
+++ b/tornadis/write_buffer.py
@@ -48,6 +48,9 @@ def __str__(self):
     def __bytes__(self):
         return self._tobytes()
 
+    def __len__(self):
+        return self._total_length
+
     def _tobytes(self):
         """Serializes the write buffer into a single string (bytes).