Add missing AQL query tracking attributes

arango/aql.py

@@ -579,6 +579,7 @@ def set_tracking(
         enabled: Optional[bool] = None,
         max_slow_queries: Optional[int] = None,
         slow_query_threshold: Optional[int] = None,
+        slow_streaming_query_threshold: Optional[int] = None,
         max_query_string_length: Optional[int] = None,
         track_bind_vars: Optional[bool] = None,
         track_slow_queries: Optional[bool] = None,
@@ -593,6 +594,9 @@ def set_tracking(
         :param slow_query_threshold: Runtime threshold (in seconds) for treating a
             query as slow.
         :type slow_query_threshold: int
+        :param slow_streaming_query_threshold: Runtime threshold (in seconds) for
+            treating a streaming query as slow.
+        :type slow_query_threshold: int
         :param max_query_string_length: Max query string length (in bytes) tracked.
         :type max_query_string_length: int
         :param track_bind_vars: If set to True, track bind variables used in queries.
@@ -614,6 +618,8 @@ def set_tracking(
             data["maxQueryStringLength"] = max_query_string_length
         if slow_query_threshold is not None:
             data["slowQueryThreshold"] = slow_query_threshold
+        if slow_streaming_query_threshold is not None:
+            data["slowStreamingQueryThreshold"] = slow_streaming_query_threshold
         if track_bind_vars is not None:
             data["trackBindVars"] = track_bind_vars
         if track_slow_queries is not None:

arango/formatter.py

@@ -375,6 +375,10 @@ def format_aql_query(body: Json) -> Json:
         result["modification_query"] = body["modificationQuery"]
     if "warnings" in body:
         result["warnings"] = body["warnings"]
+
+    if "exitCode" in body:
+        result["exit_code"] = body["exitCode"]
+
     return verify_format(body, result)
 
 

tests/test_aql.py

@@ -169,13 +169,15 @@ def test_aql_query_management(db_version, db, sys_db, bad_db, col, docs):
         max_query_string_length=4000,
         max_slow_queries=60,
         slow_query_threshold=15,
+        slow_streaming_query_threshold=20,
         track_bind_vars=not tracking["track_bind_vars"],
         track_slow_queries=not tracking["track_slow_queries"],
     )
     assert new_tracking["enabled"] != tracking["enabled"]
     assert new_tracking["max_query_string_length"] == 4000
     assert new_tracking["max_slow_queries"] == 60
     assert new_tracking["slow_query_threshold"] == 15
+    assert new_tracking["slow_streaming_query_threshold"] == 20
     assert new_tracking["track_bind_vars"] != tracking["track_bind_vars"]
     assert new_tracking["track_slow_queries"] != tracking["track_slow_queries"]