refactor unit test
添加注释
bug fix & unit test