Fix compiler error for test function was removed.

N/A

Signed-off-by: liminghao <liminghao@xiaomi.com>