Implemented iterator() method on returned remote stream to consume st…

…ream for only Null terminated traversals

gremlin-client/src/aio/process/traversal/mod.rs

@@ -6,6 +6,7 @@ use crate::GremlinResult;
 use core::task::Context;
 use core::task::Poll;
 use futures::Stream;
+use futures::StreamExt;
 use std::marker::PhantomData;
 use std::pin::Pin;
 
@@ -29,6 +30,17 @@ impl<T> RemoteTraversalStream<T> {
         }
     }
 }
+
+impl RemoteTraversalStream<crate::structure::Null> {
+    pub async fn iterate(&mut self) -> GremlinResult<()> {
+        while let Some(response) = self.next().await {
+            //consume the entire stream, returning any errors
+            response?;
+        }
+        Ok(())
+    }
+}
+
 impl<T: FromGValue> Stream for RemoteTraversalStream<T> {
     type Item = GremlinResult<T>;
 

gremlin-client/src/process/traversal/mod.rs

@@ -47,6 +47,16 @@ impl<T: FromGValue> RemoteTraversalIterator<T> {
     }
 }
 
+impl RemoteTraversalIterator<crate::structure::Null> {
+    pub fn iterate(&mut self) -> GremlinResult<()> {
+        while let Some(response) = self.next() {
+            //consume the entire iterator, returning any errors
+            response?;
+        }
+        Ok(())
+    }
+}
+
 impl<T: FromGValue> Iterator for RemoteTraversalIterator<T> {
     type Item = GremlinResult<T>;
 

gremlin-client/tests/integration_traversal.rs

@@ -2612,15 +2612,12 @@ fn test_none_step() {
     let g = traversal().with_remote(client);
 
     //The addition of a None step however should not IO a vertex back
-    let mut iter = g
-        .add_v("test_none_step")
+    g.add_v("test_none_step")
         .none()
         .iter()
-        .expect("Should get a iter back");
-    assert!(
-        iter.next().is_none(),
-        "But the iter should have no elements because of the None step"
-    );
+        .expect("Should get a iter back")
+        .iterate()
+        .expect("Shouldn't error consuming iterator");
 
     //Make sure the vertex is present in the graph
     let vertex_count = g