[GEM] Update testcases for new API.