1.修改代码逻辑.
1.增加测试代码.
1.修改测试代码
Signed-off-by: jinze.yuan