Skip to content

Commit 3884b30

Browse files
committed
Replace futures::stream::iter with our implementation in tests
1 parent 10a164e commit 3884b30

File tree

2 files changed

+9
-10
lines changed

2 files changed

+9
-10
lines changed

src/future.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -243,11 +243,10 @@ mod tests {
243243

244244
#[test]
245245
fn test_flatten_stream() {
246-
use futures::stream;
247-
use crate::stream::collect;
246+
use crate::stream::{collect, iter};
248247
executor::block_on(async {
249248
let stream_items = vec![17, 18, 19];
250-
let future_of_a_stream = ready(stream::iter(stream_items));
249+
let future_of_a_stream = ready(iter(stream_items));
251250

252251
let stream = flatten_stream(future_of_a_stream);
253252
let list: Vec<_> = await!(collect(stream));

src/stream.rs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -94,13 +94,13 @@ pub fn iter<I>(i: I) -> impl Stream<Item = I::Item>
9494

9595
#[cfg(test)]
9696
mod tests {
97-
use futures::{executor, stream};
97+
use futures::executor;
9898
use crate::stream::*;
9999
use crate::future::ready;
100100

101101
#[test]
102102
fn test_next() {
103-
let mut stream = stream::iter(1..=3);
103+
let mut stream = iter(1..=3);
104104

105105
assert_eq!(executor::block_on(next(&mut stream)), Some(1));
106106
assert_eq!(executor::block_on(next(&mut stream)), Some(2));
@@ -110,31 +110,31 @@ mod tests {
110110

111111
#[test]
112112
fn test_collect() {
113-
let stream = stream::iter(1..=5);
113+
let stream = iter(1..=5);
114114

115115
let collection : Vec<i32> = executor::block_on(collect(stream));
116116
assert_eq!(collection, vec![1, 2, 3, 4, 5]);
117117
}
118118

119119
#[test]
120120
fn test_map() {
121-
let stream = stream::iter(1..=3);
121+
let stream = iter(1..=3);
122122
let stream = map(stream, |x| x * 2);
123123

124124
assert_eq!(vec![2, 4, 6], executor::block_on(collect::<_, Vec<_>>(stream)));
125125
}
126126

127127
#[test]
128128
fn test_filter() {
129-
let stream = stream::iter(1..=10);
129+
let stream = iter(1..=10);
130130
let evens = filter(stream, |x| ready(x % 2 == 0));
131131

132132
assert_eq!(vec![2, 4, 6, 8, 10], executor::block_on(collect::<_, Vec<_>>(evens)));
133133
}
134134

135135
#[test]
136136
fn test_filter_map() {
137-
let stream = stream::iter(1..=10);
137+
let stream = iter(1..=10);
138138
let evens = filter_map(stream, |x| {
139139
let ret = if x % 2 == 0 { Some(x + 1) } else { None };
140140
ready(ret)
@@ -145,7 +145,7 @@ mod tests {
145145

146146
#[test]
147147
fn test_into_future() {
148-
let stream = stream::iter(1..=2);
148+
let stream = iter(1..=2);
149149

150150
let (item, stream) = executor::block_on(into_future(stream));
151151
assert_eq!(Some(1), item);

0 commit comments

Comments
 (0)