Add test for LatteExtension API