From 9a9a2d9125e6b1fbbfff0020ad0f29df74249f44 Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Sat, 12 Mar 2011 16:00:01 +1300
Subject: Unit tests for proxy.read_chunked

---
 test/test_proxy.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/test/test_proxy.py b/test/test_proxy.py
index 6bae46fc..82d2dbec 100644
--- a/test/test_proxy.py
+++ b/test/test_proxy.py
@@ -5,6 +5,22 @@ import email.utils
 import tutils
 
 
+class u_read_chunked(libpry.AutoTree):
+    def test_all(self):
+        s = cStringIO.StringIO("1\r\na\r\n0\r\n")
+        libpry.raises(IOError, proxy.read_chunked, s)
+
+        s = cStringIO.StringIO("1\r\na\r\n0\r\n\r\n")
+        assert proxy.read_chunked(s) == "a"
+
+        s = cStringIO.StringIO("\r\n")
+        libpry.raises(IOError, proxy.read_chunked, s)
+
+        s = cStringIO.StringIO("1\r\nfoo")
+        libpry.raises(IOError, proxy.read_chunked, s)
+
+
+
 class u_parse_request_line(libpry.AutoTree):
     def test_simple(self):
         libpry.raises(proxy.ProxyError, proxy.parse_request_line, "")
@@ -201,4 +217,5 @@ tests = [
     u_parse_url(),
     uError(),
     uClientConnect(),
+    u_read_chunked(),
 ]
-- 
cgit v1.2.3