add test for mocking