add unit test class

add unit test class