Correcting Custom Test Client class docs (#2004)

pallets · Sep 5, 2016 · ccd02bf · ccd02bf
1 parent 92ce20e
commit ccd02bf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/flask/app.py b/flask/app.py
@@ -878,9 +878,9 @@ def test_client(self, use_cookies=True, **kwargs):
             from flask.testing import FlaskClient
 
             class CustomClient(FlaskClient):
-                def __init__(self, authentication=None, *args, **kwargs):
-                    FlaskClient.__init__(*args, **kwargs)
-                    self._authentication = authentication
+                def __init__(self, *args, **kwargs):
+                    self._authentication = kwargs.pop("authentication")
+                    super(CustomClient,self).__init__( *args, **kwargs)
 
             app.test_client_class = CustomClient
             client = app.test_client(authentication='Basic ....')