Add tests using the mock interface