add more unit tests.