update tests with mocks