Add key focus change unit test