Test Code Review

Signed-off-by: Green Wan <greeen.wan@gmail.com>