Use mock for test hook called
Remove relative import
Merge pull request #1 from imom0/use-mock

Use mock for test hook called