Skip to content

Commit 14fe5e2

Browse files
committed
Fix close_on_teardown option.
1 parent 5d78dd8 commit 14fe5e2

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

flask_pymemcache.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def init_app(self, app, conf_key=None):
8989
if close_on_teardown:
9090
@app.teardown_appcontext
9191
def close_connection(exc=None):
92-
client.disconnect_all()
92+
client.close()
9393

9494
@property
9595
def client(self):

test_flask_pymemcache.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,14 @@ def test_simple(self):
1414
app = flask.Flask(__name__)
1515
app.config['PYMEMCACHE'] = {
1616
'server': ('localhost', 11211),
17-
'key_prefix': b'px'}
17+
'key_prefix': b'px',
18+
'close_on_teardown': True}
1819
memcache.init_app(app)
1920

2021
with app.app_context():
2122
memcache.client.set(b'foo', b'bar')
23+
24+
with app.app_context():
2225
assert memcache.client.get(b'foo') == b'bar'
2326

2427
assert pymc.get(b'pxfoo') == b'bar'

0 commit comments

Comments
 (0)