[test] Add more tests