dezswap 1.1.1

Common dezswap types
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

use cosmwasm_std::{DepsMut, StdError, StdResult};
use cw2::{get_contract_version, set_contract_version};

pub fn migrate_version(
    deps: DepsMut,
    target_contract_version: &str,
    name: &str,
    version: &str,
) -> StdResult<()> {
    let prev_version = get_contract_version(deps.as_ref().storage)?;
    if prev_version.contract != name {
        return Err(StdError::generic_err("invalid contract"));
    }

    if prev_version.version != target_contract_version {
        return Err(StdError::generic_err(format!(
            "invalid contract version. target {}, but source is {}",
            target_contract_version, prev_version.version
        )));
    }

    set_contract_version(deps.storage, name, version)?;

    Ok(())
}

#[cfg(test)]
mod test {
    use crate::mock_querier::mock_dependencies;

    use super::*;

    const TARGET_VERSION: &str = "version";
    const NAME: &str = "name";
    const CURRENT_VERSION: &str = "c_version";

    #[test]
    pub fn normal_migration() {
        let mut deps = mock_dependencies(&[]);
        set_contract_version(deps.as_mut().storage, NAME, TARGET_VERSION).unwrap();

        let res = migrate_version(deps.as_mut(), TARGET_VERSION, NAME, CURRENT_VERSION);

        assert_eq!(res, Ok(()));

        let version = get_contract_version(deps.as_ref().storage).unwrap();

        assert_eq!(version.contract, NAME);

        assert_eq!(version.version, CURRENT_VERSION);
    }

    #[test]
    pub fn failed_migration_with_invalid_contract_name() {
        let mut deps = mock_dependencies(&[]);
        set_contract_version(deps.as_mut().storage, NAME, TARGET_VERSION).unwrap();

        let res = migrate_version(
            deps.as_mut(),
            TARGET_VERSION,
            "invalid_name",
            CURRENT_VERSION,
        );

        assert_eq!(res, Err(StdError::generic_err("invalid contract")));

        let version = get_contract_version(deps.as_ref().storage).unwrap();

        assert_eq!(version.contract, NAME);

        assert_eq!(version.version, TARGET_VERSION);
    }

    #[test]
    pub fn failed_migration_with_invalid_target_version() {
        let mut deps = mock_dependencies(&[]);
        set_contract_version(deps.as_mut().storage, NAME, TARGET_VERSION).unwrap();

        let res = migrate_version(deps.as_mut(), "invalide_version", NAME, CURRENT_VERSION);

        assert_eq!(
            res,
            Err(StdError::generic_err(format!(
                "invalid contract version. target {}, but source is {}",
                "invalide_version", TARGET_VERSION
            )))
        );

        let version = get_contract_version(deps.as_ref().storage).unwrap();

        assert_eq!(version.contract, NAME);

        assert_eq!(version.version, TARGET_VERSION);
    }
}