[pypy-svn] r74719 - pypy/trunk/pypy/lib/test2

[wildchild at codespeak.net]

Author: wildchild
Date: Mon May 24 19:52:23 2010
New Revision: 74719

Modified:
   pypy/trunk/pypy/lib/test2/test_stackless.py
Log:
Test to show the problem resolved in r74718 about pickling tasklets halted on stackless.schedule().


Modified: pypy/trunk/pypy/lib/test2/test_stackless.py
==============================================================================
--- pypy/trunk/pypy/lib/test2/test_stackless.py	(original)
+++ pypy/trunk/pypy/lib/test2/test_stackless.py	Mon May 24 19:52:23 2010
@@ -54,3 +54,36 @@
 ''' in mod.__dict__
         finally:
             del sys.modules['mod']
+    
+    def test_pickle2(self):
+        # To test a bug where too much stuff gets pickled when
+        # a tasklet halted on stackless.schedule() is pickled.
+        import new, sys
+
+        mod = new.module('mod')
+        sys.modules['mod'] = mod
+        mod.can_unpickle = self.can_unpickle
+        mod.skip = skip
+        try:
+            exec '''
+import pickle, sys
+import stackless
+import socket
+
+def task_should_be_picklable():
+    stackless.schedule()
+
+def task_socket():
+    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    stackless.schedule()
+
+def task_pickle(ref_task):
+    p = pickle.dumps(ref_task)
+    
+ref_task = stackless.tasklet(task_should_be_picklable)()
+stackless.tasklet(task_socket)()
+stackless.tasklet(task_pickle)(ref_task)
+stackless.run()
+''' in mod.__dict__
+        finally:
+            del sys.modules['mod']