Add tests for public APIs