add mock to tests