Import StringIO so it can be used.stringio

1 files changed, 11 insertions, 0 deletions

diff --git a/tests/test_response.py b/tests/test_response.py
new file mode 100644
index 0000000..d4d4b65
--- /dev/null
+++ b/tests/test_response.py
@@ -0,0 +1,11 @@
+from paste.response import *
+
+def test_replace_header():
+    h = [('content-type', 'text/plain'),
+         ('x-blah', 'foobar')]
+    replace_header(h, 'content-length', '10')
+    assert h[-1] == ('content-length', '10')
+    replace_header(h, 'Content-Type', 'text/html')
+    assert ('content-type', 'text/html') in h
+    assert ('content-type', 'text/plain') not in h
+