[test] Add a missing header in mock_mlagent.h