[test] Add tests