add ut

Signed-off-by: wolfhead <wolfhead87@gmail.com>