Implement the service unavailable test.

Author: srbaker

test/examples/service_unavailable_example.py

@@ -19,33 +19,20 @@
 # limitations under the License.
 
 # tag::service-unavailable-import[]
-from neo4j.v1 import GraphDatabase
+from neo4j.v1 import GraphDatabase, ServiceUnavailable
+from base_application import BaseApplication
 # end::service-unavailable-import[]
 
-class ServiceUnavailableExample:
+class ServiceUnavailableExample(BaseApplication):
     def __init__(self, uri, user, password):
-        self._driver = GraphDatabase.driver( uri, auth=(user, password), Config.build().withMaxTransactionRetryTime( 3, SECONDS ).withLogging( DEV_NULL_LOGGING ).toConfig() )
-
-    def close(self):
-        self._driver.close();
+        super().__init__(uri, user, password)
 
     # tag::service-unavailable[]
     def addItem(self):
-        try ( Session session = driver.session() )
-        {
-            return session.writeTransaction( new TransactionWork<Boolean>()
-            {
-                @Override
-                public Boolean execute( Transaction tx )
-                {
-                    tx.run( "CREATE (a:Item)" );
-                    return true;
-                }
-            } );
-        }
-        catch ( ServiceUnavailableException ex )
-        {
-            return false;
-        }
-    }
+        try:
+            with self._driver.session() as session:
+                session.write_transaction(lambda tx: tx.run("CREATE (a:Item)"))
+                return True
+        except ServiceUnavailable as e:
+            return False
     # end::service-unavailable[]

test/examples/test_examples.py

@@ -106,9 +106,15 @@ def test_result_retain_example(self):
 
         self.assertEqual(employee_count, 2)
 
-    # def test_service_unavailable_example(self):
-    #     from service_unavailable_example import ServiceUnavailableExample
-    #     pass
+    def test_service_unavailable_example(self):
+        from service_unavailable_example import ServiceUnavailableExample
+
+        example = ServiceUnavailableExample(self.bolt_uri, self.user, self.password)
+        ExamplesTest._stop_server()
+
+        self.assertFalse(example.addItem())
+
+        ExamplesTest._start_server()
 
     def test_session_example(self):
         from session_example import SessionExample