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

use vit_servicing_station_lib::db::models::{
    api_tokens::ApiTokenData, challenges::Challenge, funds::Fund, proposals::FullProposalInfo,
    voteplans::Voteplan,
};
mod csv_converter;
mod generator;

pub use csv_converter::CsvConverter;
pub use generator::*;

pub fn token() -> (String, ApiTokenData) {
    ArbitrarySnapshotGenerator::default().token()
}

pub fn token_hash() -> String {
    token().0
}

pub fn proposals() -> Vec<FullProposalInfo> {
    let mut gen = ArbitrarySnapshotGenerator::default();
    let funds = gen.funds();
    gen.proposals(&funds)
}

pub fn funds() -> Vec<Fund> {
    ArbitrarySnapshotGenerator::default().funds()
}

pub fn voteplans() -> Vec<Voteplan> {
    let mut gen = ArbitrarySnapshotGenerator::default();
    let funds = gen.funds();
    gen.voteplans(&funds)
}

pub fn challenges() -> Vec<Challenge> {
    let mut gen = ArbitrarySnapshotGenerator::default();
    let funds = gen.funds();
    gen.challenges(&funds)
}

pub fn multivoteplan_snapshot() -> Snapshot {
    let mut template = ArbitraryValidVotingTemplateGenerator::new();
    let mut generator =
        ValidVotePlanGenerator::new(ArbitraryGenerator::default().valid_vote_plan_parameters());
    generator.build(&mut template)
}