tried add unit test and integration test