add some mocking in tests