File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 15
15
from logging import StreamHandler
16
16
from flask .testsuite import FlaskTestCase , catch_warnings , catch_stderr
17
17
from werkzeug .http import parse_cache_control_header , parse_options_header
18
- from flask ._compat import StringIO , text_type
18
+ from flask ._compat import StringIO , text_type , implements_iterator
19
19
20
20
21
21
def has_encoding (name ):
@@ -485,14 +485,15 @@ def test_streaming_with_context_and_custom_close(self):
485
485
app = flask .Flask (__name__ )
486
486
app .testing = True
487
487
called = []
488
+ @implements_iterator
488
489
class Wrapper (object ):
489
490
def __init__ (self , gen ):
490
491
self ._gen = gen
491
492
def __iter__ (self ):
492
493
return self
493
494
def close (self ):
494
495
called .append (42 )
495
- def next (self ):
496
+ def __next__ (self ):
496
497
return next (self ._gen )
497
498
@app .route ('/' )
498
499
def index ():
You can’t perform that action at this time.
0 commit comments