Add test cases for ViewModel

[AkshayChordiya]

It's currently not needed, since our ViewModel doesn't have any logic to test. We will write tests when our ViewModel starts growing and becomes complex