add more unit test cases