1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

use std::fmt::Debug;
use std::future::Future;
use std::marker::PhantomData;

use async_trait::async_trait;

use crate::AppData;
use crate::AppDataResponse;
use crate::DefensiveCheckBase;
use crate::RaftStorage;
use crate::RaftTypeConfig;
use crate::StorageError;
use crate::StoreExt;

/// The trait to build a [`RaftStorage`] implementation.
#[async_trait]
pub trait StoreBuilder<C, S>: Send + Sync
where
    C: RaftTypeConfig,
    S: RaftStorage<C>,
{
    async fn run_test<Fun, Ret, Res>(&self, t: Fun) -> Result<Ret, StorageError<C::NodeId>>
    where
        Res: Future<Output = Result<Ret, StorageError<C::NodeId>>> + Send,
        Fun: Fn(S) -> Res + Sync + Send;
}

/// A builder for testing [`StoreExt`].
pub struct DefensiveStoreBuilder<C, BaseStore, BaseBuilder>
where
    C: RaftTypeConfig,
    C::D: AppData + Debug,
    C::R: AppDataResponse + Debug,
    BaseStore: RaftStorage<C>,
    BaseBuilder: StoreBuilder<C, BaseStore>,
{
    pub base_builder: BaseBuilder,

    pub s: PhantomData<(C, BaseStore)>,
}

#[async_trait]
impl<C, BaseStore, BaseBuilder> StoreBuilder<C, StoreExt<C, BaseStore>>
    for DefensiveStoreBuilder<C, BaseStore, BaseBuilder>
where
    C: RaftTypeConfig,
    C::D: AppData + Debug,
    C::R: AppDataResponse + Debug,
    BaseStore: RaftStorage<C>,
    BaseBuilder: StoreBuilder<C, BaseStore>,
{
    async fn run_test<Fun, Ret, Res>(&self, t: Fun) -> Result<Ret, StorageError<C::NodeId>>
    where
        Res: Future<Output = Result<Ret, StorageError<C::NodeId>>> + Send,
        Fun: Fn(StoreExt<C, BaseStore>) -> Res + Sync + Send,
    {
        self.base_builder
            .run_test(|base_store| async {
                let sto_ext = StoreExt::new(base_store);
                sto_ext.set_defensive(true);
                assert!(sto_ext.is_defensive(), "must impl defensive check");
                t(sto_ext).await
            })
            .await
    }
}