Add test for category view-model