async-recorder 0.3.1

use std::{mem::swap, ops::Range, sync::Arc};

use async_recorder::{Recorder, Storage};
use async_trait::async_trait;
use tokio::sync::Mutex;

#[tokio::test]
async fn record_event_to_persistence_backend() {
    let record = "Hello, World!";
    let storage = Vec::new();

    let recorder = Recorder::new(storage);
    recorder.save(record);
    let storage = recorder.close().await;

    assert_eq!(["Hello, World!"].as_slice(), storage);
}

#[tokio::test]
async fn persist_events_in_bulk() {
    let bulks = Arc::new(Mutex::new(Vec::new()));
    let storage = BlockableStorageSpy::new(bulks.clone());

    let recorder = Recorder::new(storage);
    {
        // Keep guard to bulks, so spy can not persist until it is cleared
        let _guard = bulks.lock().await;
        recorder.save("first");
        recorder.save("second");
    }
    // Now the guard has been freed. Storage can now persist immediatly. Wait for this to be
    // fininshed.
    let _ = recorder.close().await;

    // Verify that there is one bulk with two entries.
    let bulks = bulks.lock().await;
    assert_eq!(1, bulks.len());
    assert_eq!(["first", "second"].as_slice(), bulks[0])
}

#[tokio::test]
async fn records_are_filtered_using_query() {
    let storage = Vec::new();

    let recorder = Recorder::new(storage);
    // We write three records
    recorder.save("first");
    recorder.save("second");
    recorder.save("third");
    // Using this query we will only see the first two
    let records = recorder.records(0..2).await;
    let _ = recorder.close().await;

    assert_eq!(["first", "second"].as_slice(), records);
}

/// When we boot up our application we may want to boot it up really fast without waiting for it to
/// have catched up with the past, by restoring all the data from the persistence backend. Luckily
/// we can accept save calls while still intitializing the storage backend.
#[tokio::test]
async fn recorder_instantiation_does_not_need_to_wait_for_persistence_backend() {
    let lazy_storage = async { vec!["first"] };

    let recorder = Recorder::from_delayed_storage(lazy_storage);
    recorder.save("second");
    let records = recorder.records(0..2).await;
    let _ = recorder.close().await;

    assert_eq!(["first", "second"].as_slice(), records)
}

#[tokio::test]
async fn recorder_should_be_able_to_work_with_a_storage_decided_at_runtime() {
    let dynamic_storage: Box<
        dyn Storage<Record = &'static str, Query = Range<usize>> + Send + 'static,
    > = Box::<Vec<&str>>::default();

    let recoder = Recorder::new(dynamic_storage);
    recoder.save("Hello, World!");
    let first_record = recoder.records(0..1).await;

    assert_eq!(["Hello, World!"].as_slice(), first_record);
}

/// Makes a copy of each received bulk.
struct BlockableStorageSpy<T> {
    /// Make this Arc Mutex, so we can block saving and observe bulk behaviour
    bulks: Arc<Mutex<Vec<Vec<T>>>>,
}

impl<T> BlockableStorageSpy<T> {
    fn new(bulks: Arc<Mutex<Vec<Vec<T>>>>) -> Self {
        BlockableStorageSpy { bulks }
    }
}

#[async_trait]
impl<T> Storage for BlockableStorageSpy<T>
where
    T: Send,
{
    type Record = T;
    type Query = ();

    async fn save(&mut self, records: &mut Vec<T>) {
        let mut tmp = Vec::new();
        swap(&mut tmp, records);
        self.bulks.lock().await.push(tmp);
    }

    async fn load(&mut self, _query: ()) -> Vec<T> {
        // Not called yet. Dummy implementation
        Vec::new()
    }
}