[GEM] Update testcases for new API.
3 files changed