Add functional test for database failing to restart

cnxauthoring/tests/test_functional.py

@@ -2853,6 +2853,31 @@ def test_service_unavailable_response(self):
             expect_errors=True)
         self.assertEqual(response.status, '503 Service Unavailable')
 
+    @mock.patch('cnxauthoring.views.logger')
+    def test_database_restart_failed(self, logger):
+        import psycopg2
+        from ..storage import storage
+
+        self.addCleanup(setattr, storage, 'conn',
+                        psycopg2.connect(storage.conn.dsn))
+
+        storage.conn.close()
+
+        with mock.patch.object(storage, 'restart') as mock_restart:
+            mock_restart.side_effect = storage.Error
+
+            response = self.testapp.post_json(
+                '/users/contents',
+                {'title': 'Test Document'},
+                status=503)
+            self.assertEqual(mock_restart.call_count, 1)
+
+        self.assertEqual(logger.exception.call_count, 3)
+        args1, args2, args3 = logger.exception.call_args_list
+        self.assertEqual(args1[0], ('Storage failure',))
+        self.assertEqual(args2[0], ('Storage failed to abort',))
+        self.assertEqual(args3[0], ('Storage failed to restart',))
+
 
 class PublicationTests(BaseFunctionalTestCase):