fix test - add stubs and mocks