sui_field_count/
lib.rs

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
// Copyright (c) Mysten Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

pub use sui_field_count_derive::*;

pub trait FieldCount {
    const FIELD_COUNT: usize;
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_basic_field_count() {
        #[derive(FieldCount)]
        struct EmptyStruct {}
        #[derive(FieldCount)]
        struct BasicStruct {
            _field1: String,
            _field2: i32,
            _field3: bool,
        }

        assert_eq!(BasicStruct::FIELD_COUNT, 3);
        assert_eq!(EmptyStruct::FIELD_COUNT, 0);
    }

    #[test]
    fn test_lifetime_field_count() {
        #[derive(FieldCount)]
        struct LifetimeStruct<'a> {
            _field1: &'a str,
            _field2: &'a [u8],
            _field3: &'a Vec<String>,
        }

        assert_eq!(LifetimeStruct::FIELD_COUNT, 3);
    }

    #[test]
    fn test_generic_type_field_count() {
        #[derive(FieldCount)]
        struct GenericStruct<T> {
            _field1: T,
            _field2: Vec<T>,
            _field3: Option<T>,
        }

        assert_eq!(GenericStruct::<String>::FIELD_COUNT, 3);
        assert_eq!(GenericStruct::<i32>::FIELD_COUNT, 3);
    }

    #[test]
    fn test_where_clause_field_count() {
        #[derive(FieldCount)]
        struct WhereStruct<T>
        where
            T: Clone,
        {
            _field1: T,
            _field2: Vec<T>,
        }

        assert_eq!(WhereStruct::<String>::FIELD_COUNT, 2);
        assert_eq!(WhereStruct::<i32>::FIELD_COUNT, 2);
    }
}