test_anonymous_traversal_properties_drop

criminosis · Nov 1, 2024 · 35d22ed · 35d22ed
1 parent 54b2457
commit 35d22ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gremlin-client/tests/integration_traversal_omni.rs b/gremlin-client/tests/integration_traversal_omni.rs
@@ -2316,7 +2316,7 @@ fn test_anonymous_traversal_properties_drop(client: GremlinClient) {
     //Make sure the property was assigned
     assert_map_property(&element_map, pre_drop_prop_key, expected_prop_value);
 
-    let created_vertex_id = element_map.get("id").expect("Should have id property");
+    let created_vertex_id = element_map.get("id").or(element_map.get(T::Id)).expect("Should have id property");
     let GValue::Int64(id) = created_vertex_id else {
         panic!("Not expected id type");
     };