@Test
public void testGetCachedViewModelFromCallback() {

    when(callback.getLastCustomNonConfigurationInstance()).thenReturn(viewModel);
    assertSame(activityDelegate.getCachedViewModel(), viewModel);
}