Test protection against picling to/from closed (real) file.

1 files changed, 20 insertions, 0 deletions

diff --git a/Lib/test/test_cpickle.py b/Lib/test/test_cpickle.py
index bedb321..90326f7 100644
--- a/Lib/test/test_cpickle.py
+++ b/Lib/test/test_cpickle.py
@@ -72,4 +72,24 @@ def dotest():
     if x2 == x: print "ok"
     else: print "bad"
 
+    # Test protection against closed files
+    import tempfile, os
+    fn = tempfile.mktemp()
+    f = open(fn, "w")
+    f.close()
+    try:
+        cPickle.dump(123, f)
+    except IOError:
+        pass
+    else:
+        print "dump to closed file should raise IOError"
+    f = open(fn, "r")
+    f.close()
+    try:
+        cPickle.load(f)
+    except IOError:
+        pass
+    else:
+        print "load from closed file should raise IOError"
+
 dotest()