Make BIO.BIO class into a context manager.

Also, add some tests.

Fix #81

1 files changed, 12 insertions, 0 deletions

diff --git a/tests/test_bio_membuf.py b/tests/test_bio_membuf.py
index 46931cf..8da6ecc 100644
--- a/tests/test_bio_membuf.py
+++ b/tests/test_bio_membuf.py
@@ -26,12 +26,24 @@ class MemoryBufferTestCase(unittest.TestCase):
         out = mb.read()
         assert out is None
 
+    def test_init_empty_cm(self):
+        with MemoryBuffer() as mb:
+            self.assertEqual(len(mb), 0)
+            out = mb.read()
+            assert out is None
+
     def test_init_something(self):
         mb = MemoryBuffer(self.data)
         self.assertEqual(len(mb), len(self.data))
         out = mb.read()
         self.assertEqual(out, self.data)
 
+    def test_init_something_cm(self):
+        with MemoryBuffer(self.data) as mb:
+            self.assertEqual(len(mb), len(self.data))
+            out = mb.read()
+            self.assertEqual(out, self.data)
+
     def test_read_less_than(self):
         chunk = len(self.data) - 7
         mb = MemoryBuffer(self.data)