v/vlib/v/tests/comptime_method_args_test.v

55 lines
1.2 KiB
V

struct TestStruct {
mut:
one_arg_called bool
two_args_called bool
three_args_called bool
}
fn (mut t TestStruct) one_arg(a1 string) {
t.one_arg_called = true
}
fn (mut t TestStruct) two_args(a2 string, b2 int) {
t.two_args_called = true
}
fn (mut t TestStruct) three_args(a3 string, b3 int, c3 []string) {
t.three_args_called = true
}
fn test_comptime_method_names() {
mut num_methods := 0
$for method in TestStruct.methods {
if method.name == 'one_arg' {
assert method.args[0].name == 'a1'
num_methods++
} else if method.name == 'two_args' {
assert method.args[0].name == 'a2'
assert method.args[1].name == 'b2'
num_methods++
} else if method.name == 'three_args' {
assert method.args[0].name == 'a3'
assert method.args[1].name == 'b3'
assert method.args[2].name == 'c3'
num_methods++
}
}
assert num_methods == 3
}
fn test_comptime_call_method() {
mut t := TestStruct{}
$for method in TestStruct.methods {
if method.name == 'one_arg' {
t.$method('one')
} else if method.name == 'two_args' {
t.$method('two', 2)
} else if method.name == 'three_args' {
t.$method('three', 3, ['th', 'ree'])
}
}
assert t.one_arg_called
assert t.two_args_called
assert t.three_args_called
}