add test lemock code