Update the unit test and add key to fragment.

Bug: 298076922
Change-Id: I8ed880c4f81bd76adc27af9bc3ba66f0b4697f74
2 files changed