From efb71121b98c4dfded212e8b6fe33873cfd83376 Mon Sep 17 00:00:00 2001 From: Andrey Zelenkov Date: Mon, 2 Apr 2018 17:03:41 +0300 Subject: Tests: added Python test with iterator context. --- test/python/ctx_iter_atexit/wsgi.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 test/python/ctx_iter_atexit/wsgi.py (limited to 'test/python/ctx_iter_atexit') diff --git a/test/python/ctx_iter_atexit/wsgi.py b/test/python/ctx_iter_atexit/wsgi.py new file mode 100644 index 00000000..d0b33daa --- /dev/null +++ b/test/python/ctx_iter_atexit/wsgi.py @@ -0,0 +1,23 @@ +import atexit + +class application: + def __init__(self, environ, start_response): + self.environ = environ + self.start = start_response + + def __iter__(self): + atexit.register(self._atexit) + + content_length = int(self.environ.get('CONTENT_LENGTH', 0)) + body = bytes(self.environ['wsgi.input'].read(content_length)) + + self.start('200', [ + ('Content-Type', self.environ.get('CONTENT_TYPE')), + ('Content-Length', str(len(body))) + ]) + yield body + + def _atexit(self): + self.start('200', [ + ('Content-Length', '0') + ]) -- cgit