Check connections for closure on LDAP Auth Credential expiration

Author: bigmontz

tests/stub/authorization.py

@@ -655,3 +655,70 @@ def get_vars(self, host=None):
 
     def get_db(self):
         return None
+
+
+class NoRoutingAuthorizationTests(TestkitTestCase):
+    def setUp(self):
+        super().setUp()
+        self._readServer = StubServer(9010)
+        self._uri = "bolt://%s:%d" % (self._readServer.host,
+                                      self._readServer.port)
+        self._auth = types.AuthorizationToken(
+            scheme="basic", principal="p", credentials="c")
+        self._userAgent = "007"
+
+    def tearDown(self):
+        self._readServer.reset()
+        super().tearDown()
+
+    def read_return_1_failure_return_2_succeed_script(self):
+        return """
+        !: BOLT 4
+        !: AUTO HELLO
+        !: AUTO GOODBYE
+        !: AUTO RESET
+        !: ALLOW RESTART
+        !: ALLOW CONCURRENT
+
+        {{
+            C: RUN "RETURN 1 as n" {} {"mode": "r"}
+            C: PULL {"n": 1000}
+            S: SUCCESS {"fields": ["n"]}
+               FAILURE {"code": "Neo.ClientError.Security.AuthorizationExpired", "message": "Authorization expired"}
+        ----
+            C: RUN "RETURN 2 as n" {} {"mode": "r"}
+            C: PULL {"n": 1000}
+            S: SUCCESS {"fields": ["n"]}
+               RECORD [1]
+               SUCCESS {"type": "r"}
+            C: RUN "RETURN 2 as n" {} {"mode": "r"}
+            C: PULL {"n": 1000}
+            S: SUCCESS {"fields": ["n"]}
+               RECORD [1]
+               SUCCESS {"type": "r"}
+        }}
+        """
+
+    def test_should_drop_connection_after_AuthorizationExpired(self):
+        self._readServer.start(
+            script=self.read_return_1_failure_return_2_succeed_script()
+        )
+        driver = Driver(self._backend, self._uri, self._auth,
+                        userAgent=self._userAgent)
+
+        session1 = driver.session('r')
+        session2 = driver.session('r')
+
+        session1.run('RETURN 2 as n').next()
+
+        try:
+            session2.run('RETURN 1 as n').next()
+        except types.DriverError as e:
+            print(e)  # should check exception
+
+        session2.close()
+        session1.close()
+        accept_count = self._readServer.count_responses("<ACCEPT>")
+        hangup_count = self._readServer.count_responses("<HANGUP>")
+        
+        self.assertEqual(accept_count, hangup_count)