aboutsummaryrefslogtreecommitdiff
path: root/test/cases/reflection.zig
blob: 3d3af3c889b04b9e45481172e91529800e1d4bd5 (plain)
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
const assert = @import("std").debug.assert;
const mem = @import("std").mem;
const reflection = this;

test "reflection: array, pointer, optional, error union type child" {
    comptime {
        assert(([10]u8).Child == u8);
        assert((*u8).Child == u8);
        assert((error!u8).Payload == u8);
        assert((?u8).Child == u8);
    }
}

test "reflection: function return type, var args, and param types" {
    comptime {
        assert(@typeOf(dummy).ReturnType == i32);
        assert(!@typeOf(dummy).is_var_args);
        assert(@typeOf(dummy_varargs).is_var_args);
        assert(@typeOf(dummy).arg_count == 3);
        assert(@ArgType(@typeOf(dummy), 0) == bool);
        assert(@ArgType(@typeOf(dummy), 1) == i32);
        assert(@ArgType(@typeOf(dummy), 2) == f32);
    }
}

fn dummy(a: bool, b: i32, c: f32) i32 {
    return 1234;
}
fn dummy_varargs(args: ...) void {}

test "reflection: struct member types and names" {
    comptime {
        assert(@memberCount(Foo) == 3);

        assert(@memberType(Foo, 0) == i32);
        assert(@memberType(Foo, 1) == bool);
        assert(@memberType(Foo, 2) == void);

        assert(mem.eql(u8, @memberName(Foo, 0), "one"));
        assert(mem.eql(u8, @memberName(Foo, 1), "two"));
        assert(mem.eql(u8, @memberName(Foo, 2), "three"));
    }
}

test "reflection: enum member types and names" {
    comptime {
        assert(@memberCount(Bar) == 4);

        assert(@memberType(Bar, 0) == void);
        assert(@memberType(Bar, 1) == i32);
        assert(@memberType(Bar, 2) == bool);
        assert(@memberType(Bar, 3) == f64);

        assert(mem.eql(u8, @memberName(Bar, 0), "One"));
        assert(mem.eql(u8, @memberName(Bar, 1), "Two"));
        assert(mem.eql(u8, @memberName(Bar, 2), "Three"));
        assert(mem.eql(u8, @memberName(Bar, 3), "Four"));
    }
}

test "reflection: @field" {
    var f = Foo{
        .one = 42,
        .two = true,
        .three = void{},
    };

    assert(f.one == f.one);
    assert(@field(f, "o" ++ "ne") == f.one);
    assert(@field(f, "t" ++ "wo") == f.two);
    assert(@field(f, "th" ++ "ree") == f.three);
    assert(@field(Foo, "const" ++ "ant") == Foo.constant);
    assert(@field(Bar, "O" ++ "ne") == Bar.One);
    assert(@field(Bar, "T" ++ "wo") == Bar.Two);
    assert(@field(Bar, "Th" ++ "ree") == Bar.Three);
    assert(@field(Bar, "F" ++ "our") == Bar.Four);
    assert(@field(reflection, "dum" ++ "my")(true, 1, 2) == dummy(true, 1, 2));
    @field(f, "o" ++ "ne") = 4;
    assert(f.one == 4);
}

const Foo = struct {
    const constant = 52;

    one: i32,
    two: bool,
    three: void,
};

const Bar = union(enum) {
    One: void,
    Two: i32,
    Three: bool,
    Four: f64,
};