[test] Update test case for snapshot