[test] add missing header for the test.