v/vlib/v/tests/comptime_method_args_test.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
}