Added mock tests for ViewModel