diff --git a/gremlin-client/tests/integration_traversal_async.rs b/gremlin-client/tests/integration_traversal_async.rs index 8db61b4..8421311 100644 --- a/gremlin-client/tests/integration_traversal_async.rs +++ b/gremlin-client/tests/integration_traversal_async.rs @@ -7,7 +7,9 @@ mod aio { use serial_test::serial; - use gremlin_client::{aio::GremlinClient, process::traversal::traversal}; + use gremlin_client::process::traversal::traversal; + + use crate::common; use super::common::aio::{connect_serializer, create_vertex_with_label, drop_vertices}; @@ -19,19 +21,12 @@ mod aio { use gremlin_client::{IoProtocol, Vertex}; - #[rstest] - #[case::graphson_v2(connect_serializer(IoProtocol::GraphSONV2))] - #[case::graphson_v3(connect_serializer(IoProtocol::GraphSONV3))] - #[case::graph_binary_v1(connect_serializer(IoProtocol::GraphBinaryV1))] - #[awt] + #[apply(common::serializers)] #[cfg_attr(feature = "async-std-runtime", async_std::test)] #[cfg_attr(feature = "tokio-runtime", tokio::test)] #[serial(test_simple_vertex_traversal_with_multiple_id)] - async fn test_simple_vertex_traversal_with_multiple_id( - #[future] - #[case] - client: GremlinClient, - ) { + async fn test_simple_vertex_traversal_with_multiple_id(protocol: IoProtocol) { + let client = connect_serializer(protocol).await; drop_vertices(&client, "test_simple_vertex_traversal_async") .await .unwrap();