[test] Adding tests