add some unit test