Merge pull request #162 from zhenlineo/1.2-auth-examples

Added test for custom auth

Author: zhenlineo

test/examples/basic_auth_example.py

@@ -22,6 +22,7 @@
 from neo4j.v1 import GraphDatabase
 # end::basic-auth-import[]
 
+
 class BasicAuthExample:
     # tag::basic-auth[]
     def __init__(self, uri, user, password):

test/examples/custom_auth_example.py

@@ -19,14 +19,19 @@
 # limitations under the License.
 
 # tag::custom-auth-import[]
-from neo4j.v1 import GraphDatabase;
+from neo4j.v1 import GraphDatabase, custom_auth
 # end::custom-auth-import[]
 
+
 class CustomAuthExample:
     # tag::custom-auth[]
-    def __init__(self, uri, principal, credentials, realm, scheme, parameters):
-        self._driver = GraphDatabase.driver( uri, auth=(principal, credentials, realm, scheme, parameters))
+    def __init__(self, uri, principal, credentials, realm, scheme, **parameters):
+        self._driver = GraphDatabase.driver(uri, auth=custom_auth(principal, credentials, realm, scheme, **parameters))
     # end::custom-auth[]
 
     def close(self):
         self._driver.close()
+
+    def can_connect(self):
+        record_list = list(self._driver.session().run("RETURN 1"))
+        return int(record_list[0][0]) == 1

test/examples/test_examples.py

@@ -61,6 +61,13 @@ def test_basic_auth_example(self):
 
         self.assertTrue(example.can_connect())
 
+    def test_custom_auth_example(self):
+        from test.examples.custom_auth_example import CustomAuthExample
+
+        example = CustomAuthExample(self.bolt_uri, self.user, self.password, None, "basic", **{"key":"value"})
+
+        self.assertTrue(example.can_connect())
+
     def test_config_unencrypted_example(self):
         from test.examples.config_unencrypted_example import ConfigUnencryptedExample