add mock test, it seems just ok
the mock test the code restruct need