add unit test
Signed-off-by: 老牛啊 <obullxl@gmail.com>