clean-up

Author: robsdedude

tests/stub/session_run/test_session_run.py

@@ -19,7 +19,7 @@ def setUp(self):
     def tearDown(self):
         if self._session is not None:
             self._session.close()
-        self._server.done()
+        self._server.reset()
         super().tearDown()
 
     def test_discard_on_session_close_untouched_result(self):
@@ -89,3 +89,4 @@ def test_raises_error_on_session_run(self):
         with self.assertRaises(types.DriverError) as exc:
             self._session.run("RETURN 1 AS n")
         self.assertEqual(exc.exception.code, "Neo.ClientError.MadeUp.Code")
+        self._server.done()

tests/stub/tx_run/test_tx_run.py

@@ -22,7 +22,6 @@ def tearDown(self):
         self._router.reset()
         self._server1.reset()
         self._server2.reset()
-        self._server1._dump()
         super().tearDown()
 
     def _create_direct_driver(self):