Reduce the number of cases in GetEntitiesTest

Bug: 68919
Change-Id: I2ca21d0632d42c88bec88e299be6b958a2faf71d