use mesher::prelude::*;

mod common;
use common::make_unsigned as make_mesher;

#[test]
fn direct() {
  let (mut sender, sender_pk) = make_mesher("direct_sender");
  let (mut dest1, dest1_pk) = make_mesher("direct_dest1");
  let (mut dest2, dest2_pk) = make_mesher("direct_dest2");

  let mut packet = Packet::unsigned();
  packet.add_hop("inmem:direct_dest1".to_owned(), &sender_pk);
  packet.add_hop("inmem:direct_dest2".to_owned(), &sender_pk);
  packet.add_message(&[1], &dest1_pk);
  packet.add_message(&[2], &dest2_pk);

  sender.launch(packet).expect("failed to launch packet");

  let received1 = dest1.receive().expect("failed to receive at 1");
  assert_eq!(
    vec![vec![1]],
    received1.iter().map(|m| m.contents()).collect::<Vec<_>>()
  );

  let received2 = dest2.receive().expect("failed to receive at 2");
  assert_eq!(
    vec![vec![2]],
    received2.iter().map(|m| m.contents()).collect::<Vec<_>>()
  );
}

#[test]
fn one_hop() {
  let (mut sender, sender_pk) = make_mesher("onehop_sender");
  let (mut im, im_pk) = make_mesher("onehop_im");
  let (mut dest1, dest1_pk) = make_mesher("onehop_dest1");
  let (mut dest2, dest2_pk) = make_mesher("onehop_dest2");

  let mut packet = Packet::unsigned();
  packet.add_hop("inmem:onehop_im".to_owned(), &sender_pk);
  packet.add_hop("inmem:onehop_dest1".to_owned(), &im_pk);
  packet.add_hop("inmem:onehop_dest2".to_owned(), &im_pk);
  packet.add_message(&[1], &dest1_pk);
  packet.add_message(&[2], &dest2_pk);

  sender.launch(packet).expect("failed to launch to dest2");

  // will bounce the message along to dest1 and dest2
  im.receive().expect("failed to receive at im");

  let received1 = dest1.receive().expect("failed to receive at 1");
  assert_eq!(
    vec![vec![1]],
    received1.iter().map(|m| m.contents()).collect::<Vec<_>>()
  );

  let received2 = dest2.receive().expect("failed to receive at 2");
  assert_eq!(
    vec![vec![2]],
    received2.iter().map(|m| m.contents()).collect::<Vec<_>>()
  );
}

#[test]
fn two_hop() {
  let (mut sender, sender_pk) = make_mesher("twohops_sender");
  let (mut im1, im1_pk) = make_mesher("twohops_im1");
  let (mut im2, im2_pk) = make_mesher("twohops_im2");
  let (mut dest1, dest1_pk) = make_mesher("twohops_dest1");
  let (mut dest2, dest2_pk) = make_mesher("twohops_dest2");

  let mut packet = Packet::unsigned();
  packet.add_hop("inmem:twohops_im1".to_owned(), &sender_pk);
  packet.add_hop("inmem:twohops_im2".to_owned(), &im1_pk);
  packet.add_hop("inmem:twohops_dest1".to_owned(), &im2_pk);
  packet.add_hop("inmem:twohops_dest2".to_owned(), &im2_pk);
  packet.add_message(&[1], &dest1_pk);
  packet.add_message(&[2], &dest2_pk);

  sender.launch(packet).expect("failed to launch to dest2");

  // will bounce the message along to im2
  im1.receive().expect("failed to receive at im1");
  // will bounce the message along to dest1 and dest2
  im2.receive().expect("failed to receive at im2");

  let received1 = dest1.receive().expect("failed to receive at 1");
  assert_eq!(
    vec![vec![1]],
    received1.iter().map(|m| m.contents()).collect::<Vec<_>>()
  );

  let received2 = dest2.receive().expect("failed to receive at 2");
  assert_eq!(
    vec![vec![2]],
    received2.iter().map(|m| m.contents()).collect::<Vec<_>>()
  );
}