Wait for sync upon graph creation

apetenchea · apetenchea · commit e7bcd4db9e6d · 2025-05-11T19:38:55.000Z
diff --git a/arango/database.py b/arango/database.py
@@ -1699,6 +1699,7 @@ def create_graph(
         replication_factor: Optional[int] = None,
         write_concern: Optional[int] = None,
         satellite_collections: Optional[Sequence[str]] = None,
+        sync: Optional[bool] = None,
     ) -> Result[Graph]:
         """Create a new graph.
 
@@ -1753,6 +1754,8 @@ def create_graph(
             element must be a string and a valid collection name. The
             collection type cannot be modified later.
         :type satellite_collections: [str] | None
+        :param sync: Wait until everything is synced to disk.
+        :type sync: bool | None
         :return: Graph API wrapper.
         :rtype: arango.graph.Graph
         :raise arango.exceptions.GraphCreateError: If create fails.
@@ -1796,7 +1799,16 @@ def create_graph(
         if satellite_collections is not None:  # pragma: no cover
             data["options"]["satellites"] = satellite_collections
 
-        request = Request(method="post", endpoint="/_api/gharial", data=data)
+        params: Params = {}
+        if sync is not None:
+            params["waitForSync"] = sync
+
+        request = Request(
+            method="post",
+            endpoint="/_api/gharial",
+            data=data,
+            params=params,
+        )
 
         def response_handler(resp: Response) -> Graph:
             if resp.is_success:
diff --git a/tests/test_graph.py b/tests/test_graph.py
@@ -51,7 +51,7 @@ def test_graph_properties(graph, bad_graph, db):
         bad_graph.properties()
 
     new_graph_name = generate_graph_name()
-    new_graph = db.create_graph(new_graph_name)
+    new_graph = db.create_graph(new_graph_name, sync=True)
     properties = new_graph.properties()
     assert properties["id"] == f"_graphs/{new_graph_name}"
     assert properties["name"] == new_graph_name
