[test] Started working on tests