Add rest models tests (WIP)