[gem] Add more tests