[test] Adding some tests