Fixing grid migration test and adding extra tests to cover one grid updates

Bug: 364711064
Flag: com.android.launcher3.one_grid_specs
Test: GridMigrationTest
Change-Id: Ib162b2f802da610a4d2802af4c890425cd657add
6 files changed