test(rust): update test function handle logic

test(rust): update test function handle logic #466