From ee909e265b40743479ca2f3e8b518f76c91f83f8 Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Sat, 28 Apr 2012 22:54:45 +1200
Subject: Add a shortcut for setting Location header.

---
 test/test_rparse.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'test/test_rparse.py')

diff --git a/test/test_rparse.py b/test/test_rparse.py
index 5a3cb53c..c854b0c6 100644
--- a/test/test_rparse.py
+++ b/test/test_rparse.py
@@ -177,9 +177,10 @@ class uDisconnects(libpry.AutoTree):
         assert isinstance(v, rparse.DisconnectRandom)
 
 
-class uShortcutContentType(libpry.AutoTree):
+class uShortcuts(libpry.AutoTree):
     def test_parse(self):
-        assert len(rparse.parse({}, "400:c'foo'").headers) == 1
+        assert rparse.parse({}, "400:c'foo'").headers[0][0][:] == "Content-Type"
+        assert rparse.parse({}, "400:l'foo'").headers[0][0][:] == "Location"
 
 
 class uPauses(libpry.AutoTree):
@@ -312,5 +313,5 @@ tests = [
     uDisconnects(),
     uMisc(),
     uparse(),
-    uShortcutContentType()
+    uShortcuts()
 ]
-- 
cgit v1.2.3