Test added for custom method using repository