#[cfg(test)]
#[allow(clippy::unused_unit)]
mod join_spawn_tests {
use join::{join_spawn, try_join_spawn};
use std::error::Error;
type BoxedError = Box<dyn Error + Send + Sync>;
type Result<T> = std::result::Result<T, BoxedError>;
type _Result<T, E> = std::result::Result<T, E>;
fn three() -> u16 {
3
}
#[allow(clippy::unnecessary_wraps)]
fn ok_four() -> Result<u16> {
Ok(4)
}
#[allow(clippy::unnecessary_wraps)]
fn some_five() -> Option<u16> {
Some(5)
}
fn make_err() -> Result<u16> {
Err("error".into())
}
fn none() -> Option<u16> {
None
}
fn add_one(v: u16) -> u16 {
v + 1
}
#[allow(clippy::unnecessary_wraps)]
fn add_one_ok(v: u16) -> Result<u16> {
Ok(add_one(v))
}
fn to_err(v: u16) -> Result<u16> {
Err(v.to_string().into())
}
fn to_none(_: u16) -> Option<u16> {
None
}
#[test]
fn it_produces_n_branches_with_length_1() {
let product = try_join_spawn! {
Ok(2u16),
Ok(three()),
ok_four(),
some_five().ok_or_else(|| "error".into()),
map => |a, b, c, d| a * b * c * d
};
assert_eq!(product.unwrap(), 120);
let err = try_join_spawn! {
Ok(2),
Ok(three()),
ok_four(),
make_err(),
map => |a, b, c, d| a * b * c * d
};
assert_eq!(
err.unwrap_err().to_string(),
make_err().unwrap_err().to_string()
);
let product = try_join_spawn! {
Some(2),
Some(three()),
some_five(),
ok_four().ok(),
map => |a, b, c, d| a * b * c * d
};
assert_eq!(product, Some(120));
let none = try_join_spawn! {
Some(2),
Some(three()),
some_five(),
none(),
ok_four().ok(),
map => |a, b, c, d, e| a * b * c * d * e
};
assert_eq!(none, None);
}
#[test]
fn it_produces_n_branches_with_any_length() {
let product = try_join_spawn! {
Ok(2u16).map(add_one).and_then(add_one_ok), Ok(three()).and_then(add_one_ok).map(add_one).map(add_one).map(add_one), ok_four().map(add_one), some_five().map(add_one).ok_or_else(|| "error".into()).and_then(to_err).and_then(add_one_ok).or(Ok(5)), map => |a, b, c, d| a * b * c * d
};
assert_eq!(product.unwrap(), 700);
let err = try_join_spawn! {
Ok(2).map(add_one),
Ok(three()).and_then(to_err),
ok_four().and_then(|_| -> Result<u16> { Err("err".into()) }),
make_err(),
map => |a, b, c, d| a * b * c * d
};
assert_eq!(
err.unwrap_err().to_string(),
to_err(three()).unwrap_err().to_string()
);
}
#[test]
fn it_produces_n_branches_with_any_length_using_combinators() {
let product = try_join_spawn! {
Ok(2u16) |> add_one => add_one_ok, Ok(three()) => add_one_ok |> add_one |> add_one |> add_one, ok_four() |> add_one, some_five() |> add_one ..ok_or_else(|| "error".into()) => to_err => add_one_ok <| Ok(5), map => |a, b, c, d| a * b * c * d
};
assert_eq!(product.unwrap(), 700);
let sum = try_join_spawn! {
2u16 -> Ok |> add_one => add_one_ok, three() -> Ok => add_one_ok |> add_one |> add_one |> add_one, ok_four() |> add_one, some_five() |> add_one ..ok_or_else(|| "error".into()) => to_err => add_one_ok <| Ok(5), and_then => |a, b, c, d| Ok(a + b + c + d)
};
assert_eq!(sum.unwrap(), 21);
let err = try_join_spawn! {
Ok(2) |> add_one,
Ok(three()) => to_err,
ok_four() => |_| -> Result<u16> { Err("error".into()) },
make_err(),
map => |a, b, c, d| a * b * c * d
};
assert_eq!(
err.unwrap_err().to_string(),
to_err(three()).unwrap_err().to_string()
);
let none = try_join_spawn! {
2 -> Some |> add_one,
Some(three()) => to_none,
ok_four() => |_| -> Result<u16> { Ok(10) } ..ok(),
none(),
map => |a, b, c, d| a * b * c * d
};
assert_eq!(none, None);
}
#[test]
fn it_tests_handler_behaviour() {
let ok = try_join_spawn! {
Ok(2),
Ok(3),
ok_four(),
and_then => |_a, _b, _c| Ok::<Option<u16>, _>(None)
};
assert_eq!(ok.unwrap(), None);
let err = try_join_spawn! {
Ok(2),
Ok(3),
ok_four(),
and_then => |a: u8, _b, _c| Err::<Option<u16>, _>(a.to_string().into())
};
assert_eq!(
err.unwrap_err().to_string(),
Err::<u8, BoxedError>("2".into()).unwrap_err().to_string()
);
let some = try_join_spawn! {
Some(2),
Some(3),
some_five(),
and_then => |a, _b, _c| Some(a)
};
assert_eq!(some, Some(2));
let none = try_join_spawn! {
Some(2),
Some(3),
some_five(),
and_then => |_a, _b, _c| None::<u16>
};
assert_eq!(none, None);
let some = try_join_spawn! {
Some(2u16),
Some(3u16),
some_five(),
map => |a, b, c| a + b + c
};
assert_eq!(some, Some(10));
let none: Option<u16> = try_join_spawn! {
Some(2u16),
None,
some_five(),
and_then => |a: u16, b: u16, c: u16| -> Option<u16> { Some(a + b + c) }
};
assert_eq!(none, None);
let ok = join_spawn! {
Ok(2),
Ok(3),
ok_four(),
then => |a: Result<u16>, b: Result<u16>, c: Result<u16>| Ok::<_, u8>(a.unwrap() + b.unwrap() + c.unwrap())
};
assert_eq!(ok, Ok(9));
let err = join_spawn! {
Ok(2),
Ok(3),
ok_four(),
then => |a: Result<u16>, b: Result<u16>, c: Result<u16>| Err::<u16, _>(a.unwrap() + b.unwrap() + c.unwrap())
};
assert_eq!(err, Err(9));
}
#[test]
fn it_tests_nested_macro_combinations() {
use join::*;
let value = try_join_spawn! {
try_join_spawn! {
Ok::<_,u8>(2u32),
Ok::<_,u8>(3u32),
Ok::<_,u8>(4u32),
try_join_spawn! {
Ok::<_,u8>(6u32),
join_spawn! {
Ok::<_,u8>(8u32),
Ok::<_,u8>(9) ~|> |v| v + 1
}.1,
map => |a, b| b - a },
map => |a, b, c, d| a + b + c + d },
try_join_spawn!{
try_join! {
Ok::<_,u8>(21u32),
Ok::<_,u8>(22u32),
Ok::<_,u8>(23u32),
map => |a, b, c| a * b * c },
Ok(2u32),
and_then => |a, b| Ok(a * b) },
map => |a, b| a + b }
.unwrap();
assert_eq!(value, 21265);
}
#[test]
fn it_tests_steps() {
let product = try_join_spawn! {
let branch_0 = Ok(2u16) ~|> {
let branch_0 = branch_0.as_ref().ok().cloned();
let branch_1 = branch_1.as_ref().ok().cloned();
let branch_2 = branch_2.as_ref().ok().cloned();
let branch_3 = branch_3.as_ref().ok().cloned();
move |value| {
assert_eq!(branch_0.unwrap(), value);
assert_eq!(branch_1.unwrap(), 3);
assert_eq!(branch_2.unwrap(), 4);
assert_eq!(branch_3.unwrap(), 6);
add_one(value)
}
} ~=> add_one_ok, let branch_1 = Ok(three()) ~=> add_one_ok ~|> {
let branch_0 = branch_0.as_ref().ok().cloned();
let branch_1 = branch_1.as_ref().ok().cloned();
let branch_2 = branch_2.as_ref().ok().cloned();
let branch_3 = branch_3.as_ref().ok().cloned();
move |value| {
assert_eq!(branch_0.unwrap(), 3);
assert_eq!(branch_1.unwrap(), value);
assert_eq!(branch_2.unwrap(), 5);
assert_eq!(branch_3.unwrap(), 5);
add_one(value)
}
} ~|> add_one ~|> add_one, let branch_2 = ok_four() ~|> add_one, let branch_3 = some_five() |> add_one ..ok_or_else(|| "error".into()) ~=> to_err <| Ok(5) ~=> add_one_ok, map => |a, b, c, d| a * b * c * d
};
assert_eq!(product.unwrap(), 840);
}
#[test]
fn it_checks_multi_threading() {
use std::sync::{Arc, Mutex};
use std::thread;
use std::time::Duration;
let values = Arc::new(Mutex::new(Vec::new()));
let (values0, values1, values2) = (values.clone(), values.clone(), values.clone());
let _ = join_spawn! {
Ok::<_,u8>((values0, 1)) |> |(values, value)| {
values.lock().unwrap().push(value);
thread::sleep(Duration::from_secs(1));
{
let mut values = values.lock().unwrap();
values.sort_unstable();
assert_eq!(values[..], [1, 2, 3]);
values.pop();
}
(values, value + 1)
} ~|> |(values, value)| {
values.lock().unwrap().push(value);
thread::sleep(Duration::from_secs(1));
{
let mut values = values.lock().unwrap();
values.sort_unstable();
assert_eq!(values[..], [2, 3, 4]);
}
},
Ok::<_,u8>((values1, 2)) |> |(values, value)| {
values.lock().unwrap().push(value);
thread::sleep(Duration::from_secs(2));
{
let mut values = values.lock().unwrap();
values.sort_unstable();
assert_eq!(values[..], [1, 2]);
values.pop();
}
(values, value + 1)
} ~|> |(values, value)| {
values.lock().unwrap().push(value);
thread::sleep(Duration::from_secs(2));
{
let mut values = values.lock().unwrap();
values.sort_unstable();
assert_eq!(values[..], [2, 3, 4]);
}
},
Ok::<_,u8>((values2, 3)) |> |(values, value)| {
values.lock().unwrap().push(value);
thread::sleep(Duration::from_secs(3));
{
let mut values = values.lock().unwrap();
values.sort_unstable();
assert_eq!(values[..], [1]);
values.pop();
}
(values, value + 1)
} ~|> |(values, value)| {
values.lock().unwrap().push(value);
thread::sleep(Duration::from_secs(3));
{
let mut values = values.lock().unwrap();
values.sort_unstable();
assert_eq!(values[..], [2, 3, 4]);
}
},
then => move |_, _, _| {
assert_eq!(values.lock().unwrap()[..], [2, 3, 4]);
Ok::<_,u8>(())
}
}
.unwrap();
}
#[test]
fn it_checks_multi_threading_steps() {
let product = try_join_spawn! {
let branch_0 = Ok::<_, BoxedError>(2u16) ~|> {
let branch_0 = branch_0.as_ref().ok().cloned();
let branch_1 = branch_1.as_ref().ok().cloned();
let branch_2 = branch_2.as_ref().ok().cloned();
let branch_3 = branch_3.as_ref().ok().cloned();
move |value| {
assert_eq!(branch_0, Some(value));
assert_eq!(branch_1, Some(3));
assert_eq!(branch_2, Some(4));
assert_eq!(branch_3, Some(6));
add_one(value)
}
} ~=> add_one_ok, let branch_1 = Ok::<_, BoxedError>(three()) ~=> add_one_ok ~|> |value| value |> {
let branch_0 = branch_0.as_ref().ok().cloned();
let branch_1 = branch_1.as_ref().ok().cloned();
let branch_2 = branch_2.as_ref().ok().cloned();
let branch_3 = branch_3.as_ref().ok().cloned();
move |value| {
assert_eq!(branch_0, Some(3));
assert_eq!(branch_1, Some(value));
assert_eq!(branch_2, Some(5));
assert_eq!(branch_3, Some(5));
add_one(value)
}
} ~|> add_one ~|> add_one, let branch_2 = Ok::<_, BoxedError>(4u16) ~|> add_one, let branch_3 = some_five() |> add_one ..ok_or_else(|| "error".into()) ~=> |_| Err("".into()) <| Ok(5) ~=> add_one_ok, map => |a, b, c, d| a * b * c * d
};
assert_eq!(product.unwrap(), 840);
}
#[test]
fn it_produces_tuple() {
let values = try_join_spawn! { Ok::<_,u8>(2), Ok::<_,u8>(3) };
assert_eq!(values.unwrap(), (2, 3));
}
#[test]
fn it_produces_single_value() {
let value = try_join_spawn! { Some(1) };
assert_eq!(value.unwrap(), 1);
}
#[test]
fn it_checks_evalutation_in_case_of_error() {
let error = try_join_spawn! { Err::<u8,_>(2) ~|> |_| unreachable!(), Ok::<_,u8>(3) };
assert_eq!(error, Err(2));
}
#[test]
fn it_tests_multi_step_single_branch() {
let values = try_join_spawn! { vec![1,2,3,4,5,6,7,8,9].into_iter() -> Some ~|> >>> ?> |v| v % 3 != 0 =>[] Vec<_> }.unwrap();
assert_eq!(values, vec![1, 2, 4, 5, 7, 8]);
}
#[test]
fn it_tests_iter_combinators() {
let mut some_vec = Some(vec![0u8]);
let values: (Vec<_>, Vec<_>) = join_spawn! {
vec![2u8, 3, 4, 5, 6, 7, 8, 9, 10, 11].into_iter() |> |v| { some_vec = None; v + 1 } ?|> |v| if v % 2 == 0 { Some(v) } else { None } |n> ^@ { some_vec.clone() }, |mut acc, (index, v)| { acc.as_mut().unwrap().push(v + (index as u8)); acc } ..unwrap().into_iter() =>[] Vec<_> ..into_iter() ?&!> |&n| (n as f64).cos().abs() > ::std::f64::consts::PI / 3f64
};
assert_eq!(values, (vec![], vec![0, 4, 7, 10, 13, 16]));
let values = vec![0, 1u8, 2, 3, 4, 5, 6];
let other_values = vec![4u8, 5, 6, 7, 8, 9, 10];
assert_eq!(
join_spawn! {
{ let values = values.clone(); values.into_iter() } >^> { let other_values = other_values.clone(); other_values.into_iter() } ?&!> |(v, v1)| v % 2 == 0 && v1 % 2 == 0,
},
(
vec![(0u8, 4u8), (2, 6), (4, 8), (6, 10)],
vec![(1u8, 5u8), (3, 7), (5, 9)]
)
);
assert_eq!(
join_spawn! {
{ let values = values.clone(); values.into_iter() } >^> { let other_values = other_values.clone(); other_values.into_iter() } <-> _, _, Vec<_>, Vec<_>
},
(values, other_values)
);
assert_eq!(
join_spawn! { vec![1u8, 2, 3, 4, 5].into_iter() ?> |v| v % 2 != 0 =>[] Vec<_> },
vec![1u8, 3, 5]
);
assert_eq!(
try_join_spawn! { let v = vec![1, 2, 3, 4, 5] -> Some ~=> >>> ..into_iter() ?|>@ |v| if v % 2 == 0 { Some(v) } else { None } },
Some(2)
);
assert_eq!(
join_spawn! { vec![vec![1, 2, 3], vec![2]].into_iter() ^^> =>[] Vec<_> },
vec![1, 2, 3, 2]
);
assert!(
try_join_spawn! { vec![Ok(5), Err(4)].into_iter() ?^@ 0, |acc, v| v.map(|v| acc + v) }
.is_err()
);
}
#[test]
fn it_tests_initial_block_capture() {
use std::sync::Arc;
let out = Arc::new(5);
let value = try_join_spawn! {
let pat_1 = { let out = out.clone(); Some(*out) },
let pat_2 = { Some(*out) },
map => |a, _| a
}
.unwrap();
assert_eq!(value, 5);
}
}