serai/processor/src/tests/mod.rs

pub(crate) mod util;

mod key_gen;
pub(crate) use key_gen::test_key_gen;

mod scanner;
pub(crate) use scanner::test_scanner;

mod signer;
pub(crate) use signer::{sign, test_signer};

mod wallet;
pub(crate) use wallet::test_wallet;

mod addresses;
pub(crate) use addresses::test_addresses;

// Effective Once
lazy_static::lazy_static! {
  static ref INIT_LOGGER: () = env_logger::init();
}

#[macro_export]
macro_rules! sequential {
  () => {
    lazy_static::lazy_static! {
      static ref SEQUENTIAL: tokio::sync::Mutex<()> = tokio::sync::Mutex::new(());
    }
  };
}

#[macro_export]
macro_rules! async_sequential {
  ($(async fn $name: ident() $body: block)*) => {
    $(
      #[tokio::test]
      async fn $name() {
        *$crate::tests::INIT_LOGGER;
        let guard = SEQUENTIAL.lock().await;
        let local = tokio::task::LocalSet::new();
        local.run_until(async move {
          if let Err(err) = tokio::task::spawn_local(async move { $body }).await {
            drop(guard);
            Err(err).unwrap()
          }
        }).await;
      }
    )*
  }
}

#[macro_export]
macro_rules! test_coin {
  (
    $C: ident,
    $coin: ident,
    $key_gen: ident,
    $scanner: ident,
    $signer: ident,
    $wallet: ident,
    $addresses: ident,
  ) => {
    use $crate::tests::{test_key_gen, test_scanner, test_signer, test_wallet, test_addresses};

    // This doesn't interact with a node and accordingly doesn't need to be run sequentially
    #[tokio::test]
    async fn $key_gen() {
      test_key_gen::<$C>().await;
    }

    sequential!();

    async_sequential! {
      async fn $scanner() {
        test_scanner($coin().await).await;
      }
    }

    async_sequential! {
      async fn $signer() {
        test_signer($coin().await).await;
      }
    }

    async_sequential! {
      async fn $wallet() {
        test_wallet($coin().await).await;
      }
    }

    async_sequential! {
      async fn $addresses() {
        test_addresses($coin().await).await;
      }
    }
  };
}

mod literal;