add mock unit tests