增加单元测试提升单元测试覆盖率

Signed-off-by: jialechan <jiale.chan@gmail.com>