grub: update test code

Change-Id: I48afb0067932e69e0e7bc31d8369f3712bb71db0