The AOT tests are relying on mock implementations that are duplicated in the codebase. We should rationalize those in test fixtures instread as it makes it hard to change an interface that those duplicated mocks implement.