Fixes based on review

tjoubert · tjoubert · commit c30a7677ae0d · 2022-11-29T06:06:27.000+04:00
diff --git a/arango/request.py b/arango/request.py
@@ -3,20 +3,21 @@
 from typing import Any, MutableMapping, Optional
 
 from arango.typings import DriverFlags, Fields, Headers, Params
+from arango.version import __version__
 
 
 def normalize_headers(
-    headers: Optional[Headers], driverFlags: Optional[DriverFlags] = None
+    headers: Optional[Headers], driver_flags: Optional[DriverFlags] = None
 ) -> Headers:
     flags = ""
-    if driverFlags is not None:
-        for flag in driverFlags:
+    if driver_flags is not None:
+        for flag in driver_flags:
             flags = flags + flag + ";"
-    driverInfo = "python-arango/7.5.2 (" + flags + ")"
+    driver_header = "python-arango/" + __version__ + " (" + flags + ")"
     normalized_headers: Headers = {
         "charset": "utf-8",
         "content-type": "application/json",
-        "x-arango-driver": driverInfo,
+        "x-arango-driver": driver_header,
     }
     if headers is not None:
         for key, value in headers.items():
diff --git a/tests/test_request.py b/tests/test_request.py
@@ -49,11 +49,9 @@ def test_request_json_data() -> None:
     assert request.method == "post"
     assert request.endpoint == "/_api/test"
     assert request.params == {"bool": "1"}
-    # assert request.headers == {
-    #     "charset": "utf-8",
-    #     "content-type": "application/json",
-    #     "foo": "bar",
-    # }
+    assert request.headers["charset"] == "utf-8"
+    assert request.headers["content-type"] == "application/json"
+    assert request.headers["foo"] == "bar"
     assert request.data == {"baz": "qux"}
 
 
