AddViewModel extends WatchLaterViewModel implements AccountRepository.TokenCallback, YoutubeRepository.AddVideoCallback,
        YoutubeRepository.VideoInfoCallback { public LiveData<Account> getAccount() { return account; } @SuppressWarnings("WeakerAccess")  AddViewModel(WatchLaterApplication application); LiveData<Account> getAccount(); void setAccount(Account account); void setVideoUri(Uri uri); LiveData<VideoInfo> getVideoInfo(); LiveData<VideoAdd> getVideoAdd(); LiveData<Boolean> getPermissionNeeded(); void setPermissionNeeded(boolean needsPermission); void watchLater(); @Override void onToken(boolean hasError, String token, Intent intent); @Override void onAddResult(YoutubeRepository.ErrorType errorType, String token); @Override void onVideoInfoResult(YoutubeRepository.ErrorType errorType, YoutubeRepository.Videos.Item item);  }
@SuppressWarnings("unchecked") @Test public void should_get_account() { LiveData<Account> liveData = mock(LiveData.class); when(accountRepository.get()).thenReturn(liveData); AddViewModel viewModel = new AddViewModel(application); LiveData<Account> defaultAccount = viewModel.getAccount(); assertThat(defaultAccount).isEqualTo(liveData); }