zebra-scan 0.1.0-alpha.15

Shielded transaction scanner for the Zcash blockchain
Documentation 
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

//! Fixed test vectors for the scanner Storage.

use zebra_chain::{block::Height, parameters::Network};
use zebra_state::TransactionIndex;

use crate::{
    storage::db::tests::{fake_sapling_results, new_test_storage},
    tests::ZECPAGES_SAPLING_VIEWING_KEY,
};

/// Tests that keys are deleted correctly
#[test]
pub fn deletes_keys_and_results_correctly() {
    let mut db = new_test_storage(&Network::Mainnet);

    let zec_pages_sapling_efvk = ZECPAGES_SAPLING_VIEWING_KEY.to_string();

    // Replace the last letter of the zec_pages efvk
    let fake_efvk = format!(
        "{}t",
        &ZECPAGES_SAPLING_VIEWING_KEY[..ZECPAGES_SAPLING_VIEWING_KEY.len() - 1]
    );

    let efvks = [&zec_pages_sapling_efvk, &fake_efvk];
    let fake_heights = [Height::MIN, Height(1), Height::MAX];
    let fake_transaction_indexes = [
        TransactionIndex::MIN,
        TransactionIndex::from_index(40),
        TransactionIndex::MAX,
    ];

    for efvk in efvks {
        for fake_result_height in fake_heights {
            db.insert_sapling_results(
                efvk,
                fake_result_height,
                fake_sapling_results(fake_transaction_indexes),
            );
        }
    }

    let expected_num_entries = fake_heights.len();
    let expected_num_results_per_entry = fake_transaction_indexes.len();

    for efvk in efvks {
        assert_eq!(
            db.sapling_results(efvk).len(),
            expected_num_entries,
            "there should be {expected_num_entries} entries for this key in the db"
        );

        for (_, result) in db.sapling_results(efvk) {
            assert_eq!(
                result.len(),
                expected_num_results_per_entry,
                "there should be {expected_num_results_per_entry} results for this entry in the db"
            );
        }

        db.delete_sapling_keys(vec![efvk.clone()]);

        assert!(
            db.sapling_results(efvk).is_empty(),
            "all results for this key should have been deleted"
        );
    }
}

/// Tests that keys are deleted correctly
#[test]
pub fn clears_results_correctly() {
    let mut db = new_test_storage(&Network::Mainnet);

    let zec_pages_sapling_efvk = ZECPAGES_SAPLING_VIEWING_KEY.to_string();

    // Replace the last letter of the zec_pages efvk
    let fake_efvk = format!(
        "{}t",
        &ZECPAGES_SAPLING_VIEWING_KEY[..ZECPAGES_SAPLING_VIEWING_KEY.len() - 1]
    );

    let efvks = [&zec_pages_sapling_efvk, &fake_efvk];
    let fake_heights = [Height::MIN, Height(1), Height::MAX];
    let fake_transaction_indexes = [
        TransactionIndex::MIN,
        TransactionIndex::from_index(40),
        TransactionIndex::MAX,
    ];

    for efvk in efvks {
        for fake_result_height in fake_heights {
            db.insert_sapling_results(
                efvk,
                fake_result_height,
                fake_sapling_results(fake_transaction_indexes),
            );
        }
    }

    let expected_num_entries = fake_heights.len();
    let expected_num_results_per_entry = fake_transaction_indexes.len();

    for efvk in efvks {
        assert_eq!(
            db.sapling_results(efvk).len(),
            expected_num_entries,
            "there should be {expected_num_entries} entries for this key in the db"
        );

        for (_, result) in db.sapling_results(efvk) {
            assert_eq!(
                result.len(),
                expected_num_results_per_entry,
                "there should be {expected_num_results_per_entry} results for this entry in the db"
            );
        }

        db.delete_sapling_results(vec![efvk.clone()]);

        assert_eq!(
            db.sapling_results(efvk).len(),
            1,
            "all results for this key should have been deleted, one empty entry should remain"
        );

        for (_, result) in db.sapling_results(efvk) {
            assert!(
                result.is_empty(),
                "there should be no results for this entry in the db"
            );
        }
    }
}