Getting the driver up-to-date (#377)

* Adding missing parts

* Bumping driver version

Author: apetenchea

arango/cluster.py

@@ -261,7 +261,7 @@ def endpoints(self) -> Result[List[str]]:
 
         :return: List of endpoints.
         :rtype: [str]
-        :raise arango.exceptions.ServerEndpointsError: If retrieval fails.
+        :raise arango.exceptions.ClusterEndpointsError: If retrieval fails.
         """
         request = Request(method="get", endpoint="/_api/cluster/endpoints")
 

arango/database.py

@@ -45,6 +45,7 @@
     PermissionResetError,
     PermissionUpdateError,
     ServerAvailableOptionsGetError,
+    ServerCheckAvailabilityError,
     ServerCurrentOptionsGetError,
     ServerDetailsError,
     ServerEchoError,
@@ -445,7 +446,7 @@ def set_license(self, license: str, force: bool = False) -> Result[Json]:
         :type force: bool
         :return: Server license.
         :rtype: dict
-        :raise arango.exceptions.ServerLicenseError: If retrieval fails.
+        :raise arango.exceptions.ServerLicenseSetError: If retrieval fails.
         """
         request = Request(
             method="put",
@@ -481,6 +482,25 @@ def response_handler(resp: Response) -> Json:
 
         return self._execute(request, response_handler)
 
+    def check_availability(self) -> Result[str]:
+        """Return ArangoDB server availability mode.
+
+        :return: Server availability mode ("readonly" or "default").
+        :rtype: str
+        :raise arango.exceptions.ServerCheckAvailabilityError: If retrieval fails.
+        """
+        request = Request(
+            method="get",
+            endpoint="/_admin/server/availability",
+        )
+
+        def response_handler(resp: Response) -> str:
+            if not resp.is_success:
+                raise ServerCheckAvailabilityError(resp, request)
+            return str(resp.body["mode"])
+
+        return self._execute(request, response_handler)
+
     def compact(
         self,
         change_level: Optional[bool] = None,
@@ -1069,6 +1089,7 @@ def metrics(self) -> Result[str]:
 
         :return: Server metrics in Prometheus format.
         :rtype: str
+        :raise arango.exceptions.ServerMetricsError: If operation fails.
         """
         request = Request(method="get", endpoint="/_admin/metrics/v2")
 

arango/exceptions.py

@@ -654,6 +654,10 @@ class ServerTimeError(ArangoServerError):
     """Failed to retrieve server system time."""
 
 
+class ServerCheckAvailabilityError(ArangoServerError):
+    """Failed to retrieve server availability mode."""
+
+
 class ServerEchoError(ArangoServerError):
     """Failed to retrieve details on last request."""
 

arango/request.py

@@ -12,7 +12,7 @@ def normalize_headers(
     if driver_flags is not None:
         for flag in driver_flags:
             flags = flags + flag + ";"
-    driver_version = "8.2.1"
+    driver_version = "8.2.2"
     driver_header = "python-arango/" + driver_version + " (" + flags + ")"
     normalized_headers: Headers = {
         "charset": "utf-8",

docs/foxx.rst

@@ -83,9 +83,9 @@ information, refer to `ArangoDB manual`_.
     foxx.readme(service_mount)
     foxx.swagger(service_mount)
     foxx.download(service_mount)
-    foxx.commit(service_mount)
+    foxx.commit()
     foxx.scripts(service_mount)
-    foxx.run_script(service_mount, 'setup', [])
+    foxx.run_script(service_mount, 'setup', {})
     foxx.run_tests(service_mount, reporter='xunit', output_format='xml')
 
     # Delete a service.

tests/test_database.py

@@ -20,6 +20,7 @@
     DatabaseListError,
     DatabasePropertiesError,
     DatabaseSupportInfoError,
+    ServerCheckAvailabilityError,
     ServerDetailsError,
     ServerEchoError,
     ServerEngineError,
@@ -355,6 +356,11 @@ def test_database_misc_methods(client, sys_db, db, bad_db, cluster, secret, db_v
         with pytest.raises(CollectionKeyGeneratorsError):
             bad_db.key_generators()
 
+    with pytest.raises(ServerCheckAvailabilityError):
+        bad_db.check_availability()
+    availability = db.check_availability()
+    assert isinstance(availability, str)
+
 
 def test_database_management(db, sys_db, bad_db):
     # Test list databases